chatGPTBox/tests/unit/config/user-config.test.mjs at 5862984933ef4b6fe1f822a42e9efd4ada63fc90 · PeterDaveHello/chatGPTBox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import assert from 'node:assert/strict'
import { beforeEach, test } from 'node:test'
import { clearOldAccessToken, getUserConfig, setAccessToken } from '../../../src/config/index.mjs'

const THIRTY_DAYS_MS = 30 * 24 * 3600 * 1000

beforeEach(() => {
  globalThis.__TEST_BROWSER_SHIM__.clearStorage()
})

test('getUserConfig migrates legacy chat.openai.com URL to chatgpt.com', async () => {
  globalThis.__TEST_BROWSER_SHIM__.replaceStorage({
    customChatGptWebApiUrl: 'https://chat.openai.com',
  })

  const config = await getUserConfig()

  assert.equal(config.customChatGptWebApiUrl, 'https://chatgpt.com')
})

test('getUserConfig keeps modern chatgpt.com URL unchanged', async () => {
  globalThis.__TEST_BROWSER_SHIM__.replaceStorage({
    customChatGptWebApiUrl: 'https://chatgpt.com',
  })

  const config = await getUserConfig()

  assert.equal(config.customChatGptWebApiUrl, 'https://chatgpt.com')
})

test('clearOldAccessToken clears expired token older than 30 days', async (t) => {
  const now = 1_700_000_000_000
  t.mock.method(Date, 'now', () => now)

  globalThis.__TEST_BROWSER_SHIM__.replaceStorage({
    accessToken: 'stale-token',
    tokenSavedOn: now - THIRTY_DAYS_MS - 1_000,
  })

  await clearOldAccessToken()

  const storage = globalThis.__TEST_BROWSER_SHIM__.getStorage()
  assert.equal(storage.accessToken, '')
  // tokenSavedOn write behavior is covered in the dedicated setAccessToken test below.
})

test('setAccessToken updates tokenSavedOn to Date.now', async (t) => {
  const now = 1_700_000_000_000
  t.mock.method(Date, 'now', () => now)

  await setAccessToken('new-token')

  const storage = globalThis.__TEST_BROWSER_SHIM__.getStorage()
  assert.equal(storage.accessToken, 'new-token')
  assert.equal(storage.tokenSavedOn, now)
})

test('clearOldAccessToken keeps recent token within 30 days', async (t) => {
  const now = 1_700_000_000_000
  t.mock.method(Date, 'now', () => now)
  const recentSavedOn = now - THIRTY_DAYS_MS + 1_000

  globalThis.__TEST_BROWSER_SHIM__.replaceStorage({
    accessToken: 'fresh-token',
    tokenSavedOn: recentSavedOn,
  })

  await clearOldAccessToken()

  const storage = globalThis.__TEST_BROWSER_SHIM__.getStorage()
  assert.equal(storage.accessToken, 'fresh-token')
  assert.equal(storage.tokenSavedOn, recentSavedOn)
})

test('clearOldAccessToken keeps token when exactly 30 days old', async (t) => {
  const now = 1_700_000_000_000
  t.mock.method(Date, 'now', () => now)
  const savedOn = now - THIRTY_DAYS_MS

  globalThis.__TEST_BROWSER_SHIM__.replaceStorage({
    accessToken: 'boundary-token',
    tokenSavedOn: savedOn,
  })

  await clearOldAccessToken()

  const storage = globalThis.__TEST_BROWSER_SHIM__.getStorage()
  assert.equal(storage.accessToken, 'boundary-token')
  assert.equal(storage.tokenSavedOn, savedOn)
})