From 7c2c8e4961b04366cebfc28b45f737831a8c71bf Mon Sep 17 00:00:00 2001
From: Tavi Nolan <Tavi.Nolan@fmr.com>
Date: Thu, 4 Apr 2024 17:25:57 +0100
Subject: [PATCH] added dry run to github.ts Signed-off-by: Tavi Nolan
 <Tavi.Nolan@fmr.com>

---
 .../src/actions/github.test.ts                | 46 +++++++++++++++++--
 .../src/actions/github.ts                     | 27 +++++++----
 2 files changed, 59 insertions(+), 14 deletions(-)

diff --git a/plugins/scaffolder-backend-module-github/src/actions/github.test.ts b/plugins/scaffolder-backend-module-github/src/actions/github.test.ts
index 9eda9a2eb0..4802fd9078 100644
--- a/plugins/scaffolder-backend-module-github/src/actions/github.test.ts
+++ b/plugins/scaffolder-backend-module-github/src/actions/github.test.ts
@@ -33,21 +33,23 @@ jest.mock('@backstage/plugin-scaffolder-node', () => {
   };
 });
 
-import { TemplateAction } from '@backstage/plugin-scaffolder-node';
-import { ConfigReader } from '@backstage/config';
-import { createMockActionContext } from '@backstage/plugin-scaffolder-node-test-utils';
 import {
   DefaultGithubCredentialsProvider,
   GithubCredentialsProvider,
   ScmIntegrations,
 } from '@backstage/integration';
-import { createPublishGithubAction } from './github';
-import { initRepoAndPush } from '@backstage/plugin-scaffolder-node';
 import {
   enableBranchProtectionOnDefaultRepoBranch,
   entityRefToName,
 } from './gitHelpers';
 
+import { ConfigReader } from '@backstage/config';
+import { TemplateAction } from '@backstage/plugin-scaffolder-node';
+import { createMockActionContext } from '@backstage/plugin-scaffolder-node-test-utils';
+import { createPublishGithubAction } from './github';
+import { initRepoAndPush } from '@backstage/plugin-scaffolder-node';
+import { when } from 'jest-when';
+
 const publicKey = '2Sg8iYjAxxmI2LvUXpJjkYrMxURPc8r+dB7TJyvvcCU=';
 
 const initRepoAndPushMocked = initRepoAndPush as jest.Mock<
@@ -2064,4 +2066,38 @@ describe('publish:github', () => {
       requiredCommitSigning: false,
     });
   });
+
+  it('should not call createInOrg during dry run', async () => {
+    mockOctokit.rest.users.getByUsername.mockResolvedValue({
+      data: { type: 'Organization' },
+    });
+
+    mockOctokit.rest.teams.getByName.mockResolvedValue({
+      data: {
+        name: 'blam',
+        id: 42,
+      },
+    });
+
+    mockOctokit.rest.repos.createInOrg.mockResolvedValue({ data: {} });
+
+    mockContext.isDryRun = true;
+    await action.handler(mockContext);
+    expect(mockOctokit.rest.repos.createInOrg).not.toHaveBeenCalled();
+  });
+
+  it('should not call createForAuthenticatedUser during dry run', async () => {
+    mockOctokit.rest.users.getByUsername.mockResolvedValue({
+      data: { type: 'User' },
+    });
+
+    mockOctokit.rest.repos.createForAuthenticatedUser.mockResolvedValue({
+      data: {},
+    });
+
+    await action.handler(mockContext);
+    expect(
+      mockOctokit.rest.repos.createForAuthenticatedUser,
+    ).not.toHaveBeenCalled();
+  });
 });
diff --git a/plugins/scaffolder-backend-module-github/src/actions/github.ts b/plugins/scaffolder-backend-module-github/src/actions/github.ts
index f32b65bc9a..923190465a 100644
--- a/plugins/scaffolder-backend-module-github/src/actions/github.ts
+++ b/plugins/scaffolder-backend-module-github/src/actions/github.ts
@@ -14,24 +14,26 @@
  * limitations under the License.
  */
 
-import { Config } from '@backstage/config';
-import { InputError } from '@backstage/errors';
+import * as inputProps from './inputProperties';
+import * as outputProps from './outputProperties';
+
 import {
   GithubCredentialsProvider,
   ScmIntegrationRegistry,
 } from '@backstage/integration';
-import { Octokit } from 'octokit';
-import {
-  createTemplateAction,
-  parseRepoUrl,
-} from '@backstage/plugin-scaffolder-node';
 import {
   createGithubRepoWithCollaboratorsAndTopics,
   getOctokitOptions,
   initRepoPushAndProtect,
 } from './helpers';
-import * as inputProps from './inputProperties';
-import * as outputProps from './outputProperties';
+import {
+  createTemplateAction,
+  parseRepoUrl,
+} from '@backstage/plugin-scaffolder-node';
+
+import { Config } from '@backstage/config';
+import { InputError } from '@backstage/errors';
+import { Octokit } from 'octokit';
 import { examples } from './github.examples';
 
 /**
@@ -227,6 +229,13 @@ export function createPublishGithubAction(options: {
         throw new InputError('Invalid repository owner provided in repoUrl');
       }
 
+      if (ctx.isDryRun) {
+        ctx.logger.info(`Performing dry run of creating repository`);
+        ctx.output('remoteUrl', repoUrl);
+        ctx.logger.info(`Dry run complete`);
+        return;
+      }
+
       const newRepo = await createGithubRepoWithCollaboratorsAndTopics(
         client,
         repo,