Fix import error in privacy policy unit test

Author: webstackdev

src/integrations/PrivacyPolicyVersion/__tests__/index.spec.ts

@@ -2,17 +2,24 @@ import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest'
 import { execSync } from 'node:child_process'
 import { TestError } from '@test/errors'
 
+vi.mock('../../../lib/config/environmentServer', () => ({
+  getOptionalEnv: vi.fn(() => undefined),
+}))
+
 // Mock child_process
 vi.mock('node:child_process', () => ({
   execSync: vi.fn(),
 }))
 
+import { getOptionalEnv } from '../../../lib/config/environmentServer'
+
 describe('PrivacyPolicyVersion Integration', () => {
   let consoleLogSpy: ReturnType<typeof vi.spyOn>
   let consoleWarnSpy: ReturnType<typeof vi.spyOn>
 
   beforeEach(() => {
-    // Spy on console methods
+    vi.mocked(getOptionalEnv).mockReturnValue(undefined)
+
     consoleLogSpy = vi.spyOn(console, 'log').mockImplementation(() => {})
     consoleWarnSpy = vi.spyOn(console, 'warn').mockImplementation(() => {})
 
@@ -27,6 +34,7 @@ describe('PrivacyPolicyVersion Integration', () => {
     // Restore console methods
     consoleLogSpy.mockRestore()
     consoleWarnSpy.mockRestore()
+    vi.useRealTimers()
   })
 
   describe('getPrivacyPolicyVersionFromGit', () => {
@@ -61,7 +69,7 @@ describe('PrivacyPolicyVersion Integration', () => {
       })
 
       expect(consoleLogSpy).toHaveBeenCalledWith(
-        expect.stringContaining('Privacy policy version set to: 2024-03-15'),
+        expect.stringContaining('Privacy policy version set from git: 2024-03-15'),
       )
     })
 
@@ -89,44 +97,58 @@ describe('PrivacyPolicyVersion Integration', () => {
       })
     })
 
-    it('should throw BuildError when git command fails', async () => {
-      // Mock git command failure
+    it('logs warning and falls back when git command fails', async () => {
       vi.mocked(execSync).mockImplementation(() => {
         throw new TestError('Git command failed')
       })
+      vi.useFakeTimers()
+      vi.setSystemTime(new Date('2025-12-06T00:00:00Z'))
 
       const { privacyPolicyVersion } = await import('../index')
 
       const mockUpdateConfig = vi.fn()
       const integration = privacyPolicyVersion()
 
-      await expect(
-        integration.hooks['astro:config:setup']?.({
-          updateConfig: mockUpdateConfig,
-          // @ts-expect-error - Partial mock
-          config: {},
-        }),
-      ).rejects.toThrow('Could not get privacy policy version from git')
+      await integration.hooks['astro:config:setup']?.({
+        updateConfig: mockUpdateConfig,
+        // @ts-expect-error - Partial mock
+        config: {},
+      })
 
-      expect(consoleWarnSpy).not.toHaveBeenCalled()
+      expect(mockUpdateConfig).toHaveBeenCalledWith({
+        vite: {
+          define: {
+            'import.meta.env.PRIVACY_POLICY_VERSION': '"2025-12-06"',
+          },
+        },
+      })
+      expect(consoleWarnSpy).toHaveBeenCalled()
     })
 
-    it('should throw BuildError when git returns empty string', async () => {
-      // Mock git command returning empty string
+    it('logs warning and falls back when git returns empty string', async () => {
       vi.mocked(execSync).mockReturnValue('')
+      vi.useFakeTimers()
+      vi.setSystemTime(new Date('2025-05-01T00:00:00Z'))
 
       const { privacyPolicyVersion } = await import('../index')
 
       const mockUpdateConfig = vi.fn()
       const integration = privacyPolicyVersion()
 
-      await expect(
-        integration.hooks['astro:config:setup']?.({
-          updateConfig: mockUpdateConfig,
-          // @ts-expect-error - Partial mock
-          config: {},
-        }),
-      ).rejects.toThrow('No git commits found for privacy policy file')
+      await integration.hooks['astro:config:setup']?.({
+        updateConfig: mockUpdateConfig,
+        // @ts-expect-error - Partial mock
+        config: {},
+      })
+
+      expect(mockUpdateConfig).toHaveBeenCalledWith({
+        vite: {
+          define: {
+            'import.meta.env.PRIVACY_POLICY_VERSION': '"2025-05-01"',
+          },
+        },
+      })
+      expect(consoleWarnSpy).toHaveBeenCalled()
     })
   })