repair some more tests

Author: H0R5E

tests/pulumi.index.test.ts

@@ -0,0 +1,96 @@
+import * as pulumi from '@pulumi/pulumi'
+
+import { MyMocks, promiseOf } from './utils'
+import * as resources from '../stacks/main/resources'
+
+vi.mock('../stacks/main/resources')
+
+describe('stacks/main/index.ts', () => {
+  let envOrig: string
+  let mocks: MyMocks
+  let infra: typeof import('../stacks/main')
+
+  beforeEach(async () => {
+    vi.resetModules()
+    envOrig = JSON.stringify(process.env)
+    mocks = new MyMocks()
+    pulumi.runtime.setMocks(mocks)
+  })
+
+  afterEach(() => {
+    process.env = JSON.parse(envOrig)
+  })
+
+  it('Without FQDN', async () => {
+    ;(resources.buildRouter as any).mockImplementation(() => {
+      return 'mock'
+    })
+    ;(resources.buildCDN as any).mockImplementation(() => {
+      return {
+        domainName: 'example.com',
+      }
+    })
+    // @ts-ignore
+    pulumi.Config = vi.fn(() => {
+      return {
+        get: vi.fn((x) => {
+          if (x === 'serverHeaders') {
+            return '{"mock": "mock"}'
+          }
+          return ''
+        })
+      }
+    });
+
+    infra = await import('../stacks/main')
+
+    expect(resources.getLambdaRole).toHaveBeenCalledTimes(1)
+    expect(resources.validateCertificate).toHaveBeenCalledTimes(0)
+    expect(resources.buildStatic).toHaveBeenCalledTimes(1)
+    expect(resources.buildCDN).toHaveBeenCalledTimes(1)
+    expect(resources.createAliasRecord).toHaveBeenCalledTimes(0)
+    expect(resources.buildInvalidator).toHaveBeenCalledTimes(1)
+    
+    const allowedOrigin = await promiseOf(infra.allowedOrigins[0] as pulumi.Output<string>)
+    const appUrl = await promiseOf(infra.appUrl as pulumi.Output<string>)
+    expect(allowedOrigin).toMatch('https://example.com')
+    expect(appUrl).toMatch('https://example.com')
+  })
+
+  it('With FQDN', async () => {
+    const fqdn = 'mock.application.net'
+    ;(resources.buildRouter as any).mockImplementation(() => {
+      return 'mock'
+    })
+    ;(resources.buildCDN as any).mockImplementation(() => {
+      return {
+        domainName: 'example.com',
+      }
+    })
+    // @ts-ignore
+    pulumi.Config = vi.fn(() => {
+      return {
+        get: vi.fn((x) => {
+          if (x === 'serverHeaders') {
+            return '{"mock": "mock"}'
+          } else if (x === 'FQDN') {
+            return fqdn
+          }
+          return ''
+        })
+      }
+    });
+
+    infra = await import('../stacks/main')
+
+    expect(resources.validateCertificate).toHaveBeenCalledTimes(1)
+    expect(resources.createAliasRecord).toHaveBeenCalledTimes(1)
+    
+    const distOrigin = await promiseOf(infra.allowedOrigins[0] as pulumi.Output<string>)
+    const fqdnOrigin = infra.allowedOrigins[1]
+
+    expect(distOrigin).toMatch('https://example.com')
+    expect(fqdnOrigin).toMatch(`https://${fqdn}`)
+    expect(infra.appUrl).toMatch(`https://${fqdn}`)
+  })
+})

tests/stacks.main.index.test.ts

@@ -3,14 +3,14 @@ import * as path from 'path'
 
 import { getTempDir } from './utils'
 
-describe('pulumi/utils.ts', () => {
+describe('stacks/utils.ts', () => {
   let envOrig: string
-  let utils: typeof import('../pulumi/utils')
+  let utils: typeof import('../stacks/utils')
 
   beforeEach(async () => {
     vi.resetModules()
     envOrig = JSON.stringify(process.env)
-    utils = await import('../pulumi/utils')
+    utils = await import('../stacks/utils')
   })
 
   afterEach(() => {