fix: testing around current branch logic

packages/core/src/__tests__/get-current-branch.test.ts

@@ -0,0 +1,65 @@
+import { execSync } from "child_process";
+import { CiEnv } from "env-ci";
+import { getCurrentBranch } from "../auto";
+
+jest.mock('child_process')
+
+describe('getCurrentBranch', () => {
+
+    beforeEach(() => {
+        jest.clearAllMocks()
+    })
+
+    describe('when isPr', () => {
+        it('returns pr branch from env ci when valid', () => {
+            const env: Partial<CiEnv> = {
+                isPr: true,
+                prBranch: 'my-pr-branch'
+            }
+            expect(getCurrentBranch(env)).toBe('my-pr-branch')
+        });
+
+        it('tries git command when PR is invalid', () => {
+            const env: Partial<CiEnv> = {
+                isPr: true,
+                prBranch: 'undefined'
+            }
+
+            getCurrentBranch(env);
+
+            expect(execSync).toHaveBeenCalledWith("git symbolic-ref --short HEAD", {
+                encoding: "utf8",
+                stdio: "ignore",
+            })
+        });
+    })
+
+    describe('when not isPr', () => {
+
+        it('returns pr branch from env ci when valid', () => {
+            const env: Partial<CiEnv> = {
+                isPr: false,
+                prBranch: 'my-pr-branch',
+                branch: 'my-release-branch'
+            }
+
+            expect(getCurrentBranch(env)).toBe('my-release-branch');
+
+            expect(execSync).not.toHaveBeenCalled()
+        });
+
+        it('tries git command when branch name is invalid', () => {
+            const env: Partial<CiEnv> = {
+                isPr: false,
+                prBranch: 'my-pr-branch',
+                branch: undefined
+            }
+            getCurrentBranch(env);
+
+            expect(execSync).toHaveBeenCalledWith("git symbolic-ref --short HEAD", {
+                encoding: "utf8",
+                stdio: "ignore",
+            })
+        });
+    })
+})

packages/core/src/utils/get-current-branch.ts

@@ -1,15 +1,15 @@
-import envCi from "env-ci";
+import envCi, { CiEnv } from "env-ci";
 import { execSync } from "child_process";
 
-const env = envCi();
+const defaultCiEnvironment = envCi();
 
 /**
  * Validates that the given branch name should be returned by environment context
  */
 const isValidBranch = (branch: string | undefined) => typeof branch === "string" && branch !== "undefined"
 
 /** Get the current branch the git repo is set to */
-export function getCurrentBranch() {
+export function getCurrentBranch(env: Partial<CiEnv> = defaultCiEnvironment) {
   const isPR = "isPr" in env && env.isPr;
   let branch: string | undefined;
   // env-ci sets branch to target branch (ex: main) in some CI services.