update tests

Author: patinthehat

tests/lib/Repository.test.ts

@@ -25,27 +25,25 @@ it('sets properties correctly on create', async () => {
     expect(repository).toMatchSnapshot();
 });
 
-if (typeof process.env.CI === 'undefined') {
-    it('gets the current branch names', async () => {
-        const repository = new Repository('permafrost-dev/codeboost', tempPath);
-        repository.path = `${__dirname}/../..`;
+it('gets a list of local branches', async () => {
+    const repository = new Repository('permafrost-dev/codeboost', tempPath);
+    repository.path = `${__dirname}/../..`;
 
-        const mainBranchName = 'main';
-        const branch = await repository.currentBranch();
+    const branches = await repository.localBranches();
 
-        expect(branch).toBe(mainBranchName);
-    });
+    expect(branches.all.length).toBeGreaterThan(0);
+});
 
-    it('gets a list of local branches', async () => {
-        const repository = new Repository('permafrost-dev/codeboost', tempPath);
-        repository.path = `${__dirname}/../..`;
+it('gets the current branch names', async () => {
+    const repository = new Repository('permafrost-dev/codeboost', tempPath);
+    repository.path = `${__dirname}/../..`;
 
-        const mainBranchName = 'main';
-        const branches = await repository.localBranches();
+    const branch = await repository.currentBranch();
 
-        expect(branches.all).toContain(mainBranchName);
-    });
+    expect(branch.match(/main|pull.+/)).toBeTruthy();
+});
 
+if (typeof process.env.CI === 'undefined') {
     it('checks if it is on a branch', async () => {
         const repository = new Repository('permafrost-dev/codeboost', tempPath);
         repository.path = `${__dirname}/../..`;