|
1 | 1 | import { expectTypeOf, test } from 'vitest';
|
2 | 2 |
|
| 3 | +import type { RedirectFun } from '../../createRedirect'; |
3 | 4 | import type { AuthObject, InvalidTokenAuthObject } from '../authObjects';
|
4 |
| -import type { GetAuthFn, MachineAuthObject, SessionAuthObject } from '../types'; |
5 |
| - |
6 |
| -// Across our SDKs, we have a getAuth() function |
7 |
| -const getAuth: GetAuthFn<Request> = (_request: any, _options: any) => { |
8 |
| - return {} as any; |
9 |
| -}; |
10 |
| - |
11 |
| -test('infers the correct AuthObject type for each accepted token type', () => { |
12 |
| - const request = new Request('https://example.com'); |
13 |
| - |
14 |
| - // Session token by default |
15 |
| - expectTypeOf(getAuth(request)).toMatchTypeOf<SessionAuthObject>(); |
16 |
| - |
17 |
| - // Individual token types |
18 |
| - expectTypeOf(getAuth(request, { acceptsToken: 'session_token' })).toMatchTypeOf<SessionAuthObject>(); |
19 |
| - expectTypeOf(getAuth(request, { acceptsToken: 'api_key' })).toMatchTypeOf<MachineAuthObject<'api_key'>>(); |
20 |
| - expectTypeOf(getAuth(request, { acceptsToken: 'm2m_token' })).toMatchTypeOf<MachineAuthObject<'m2m_token'>>(); |
21 |
| - expectTypeOf(getAuth(request, { acceptsToken: 'oauth_token' })).toMatchTypeOf<MachineAuthObject<'oauth_token'>>(); |
22 |
| - |
23 |
| - // Array of token types |
24 |
| - expectTypeOf(getAuth(request, { acceptsToken: ['session_token', 'm2m_token'] })).toMatchTypeOf< |
25 |
| - SessionAuthObject | MachineAuthObject<'m2m_token'> | InvalidTokenAuthObject |
26 |
| - >(); |
27 |
| - expectTypeOf(getAuth(request, { acceptsToken: ['m2m_token', 'oauth_token'] })).toMatchTypeOf< |
28 |
| - MachineAuthObject<'m2m_token' | 'oauth_token'> | InvalidTokenAuthObject |
29 |
| - >(); |
30 |
| - |
31 |
| - // Any token type |
32 |
| - expectTypeOf(getAuth(request, { acceptsToken: 'any' })).toMatchTypeOf<AuthObject>(); |
| 5 | +import type { GetAuthFn, GetAuthFnNoRequest, MachineAuthObject, SessionAuthObject } from '../types'; |
| 6 | + |
| 7 | +describe('getAuth() or auth() with request parameter', () => { |
| 8 | + const getAuth: GetAuthFn<Request> = (_request: any, _options: any) => { |
| 9 | + return {} as any; |
| 10 | + }; |
| 11 | + |
| 12 | + test('infers the correct AuthObject type for each accepted token type', () => { |
| 13 | + const request = new Request('https://example.com'); |
| 14 | + |
| 15 | + // Session token by default |
| 16 | + expectTypeOf(getAuth(request)).toExtend<SessionAuthObject>(); |
| 17 | + |
| 18 | + // Individual token types |
| 19 | + expectTypeOf(getAuth(request, { acceptsToken: 'session_token' })).toExtend<SessionAuthObject>(); |
| 20 | + expectTypeOf(getAuth(request, { acceptsToken: 'api_key' })).toExtend<MachineAuthObject<'api_key'>>(); |
| 21 | + expectTypeOf(getAuth(request, { acceptsToken: 'm2m_token' })).toExtend<MachineAuthObject<'m2m_token'>>(); |
| 22 | + expectTypeOf(getAuth(request, { acceptsToken: 'oauth_token' })).toExtend<MachineAuthObject<'oauth_token'>>(); |
| 23 | + |
| 24 | + // Array of token types |
| 25 | + expectTypeOf(getAuth(request, { acceptsToken: ['session_token', 'm2m_token'] })).toExtend< |
| 26 | + SessionAuthObject | MachineAuthObject<'m2m_token'> | InvalidTokenAuthObject |
| 27 | + >(); |
| 28 | + expectTypeOf(getAuth(request, { acceptsToken: ['m2m_token', 'oauth_token'] })).toExtend< |
| 29 | + MachineAuthObject<'m2m_token' | 'oauth_token'> | InvalidTokenAuthObject |
| 30 | + >(); |
| 31 | + |
| 32 | + // Any token type |
| 33 | + expectTypeOf(getAuth(request, { acceptsToken: 'any' })).toExtend<AuthObject>(); |
| 34 | + }); |
| 35 | + |
| 36 | + test('verifies discriminated union works correctly with acceptsToken: any', () => { |
| 37 | + const request = new Request('https://example.com'); |
| 38 | + |
| 39 | + const auth = getAuth(request, { acceptsToken: 'any' }); |
| 40 | + |
| 41 | + if (auth.tokenType === 'session_token') { |
| 42 | + expectTypeOf(auth).toExtend<SessionAuthObject>(); |
| 43 | + } else if (auth.tokenType === 'api_key') { |
| 44 | + expectTypeOf(auth).toExtend<MachineAuthObject<'api_key'>>(); |
| 45 | + } else if (auth.tokenType === 'm2m_token') { |
| 46 | + expectTypeOf(auth).toExtend<MachineAuthObject<'m2m_token'>>(); |
| 47 | + } else if (auth.tokenType === 'oauth_token') { |
| 48 | + expectTypeOf(auth).toExtend<MachineAuthObject<'oauth_token'>>(); |
| 49 | + } |
| 50 | + }); |
33 | 51 | });
|
34 | 52 |
|
35 |
| -test('verifies discriminated union works correctly with acceptsToken: any', () => { |
36 |
| - const request = new Request('https://example.com'); |
| 53 | +describe('getAuth() or auth() without request parameter', () => { |
| 54 | + type SessionAuthWithRedirect = SessionAuthObject & { |
| 55 | + redirectToSignIn: RedirectFun<Response>; |
| 56 | + redirectToSignUp: RedirectFun<Response>; |
| 57 | + }; |
| 58 | + |
| 59 | + // Mimic Next.js auth() helper |
| 60 | + const auth: GetAuthFnNoRequest<SessionAuthWithRedirect, true> = (_options: any) => { |
| 61 | + return {} as any; |
| 62 | + }; |
| 63 | + |
| 64 | + test('infers the correct AuthObject type for each accepted token type', async () => { |
| 65 | + // Session token by default |
| 66 | + expectTypeOf(await auth()).toExtend<SessionAuthWithRedirect>(); |
| 67 | + |
| 68 | + // Individual token types |
| 69 | + expectTypeOf(await auth({ acceptsToken: 'session_token' })).toExtend<SessionAuthWithRedirect>(); |
| 70 | + expectTypeOf(await auth({ acceptsToken: 'api_key' })).toExtend<MachineAuthObject<'api_key'>>(); |
| 71 | + expectTypeOf(await auth({ acceptsToken: 'm2m_token' })).toExtend<MachineAuthObject<'m2m_token'>>(); |
| 72 | + expectTypeOf(await auth({ acceptsToken: 'oauth_token' })).toExtend<MachineAuthObject<'oauth_token'>>(); |
| 73 | + |
| 74 | + // Array of token types |
| 75 | + expectTypeOf(await auth({ acceptsToken: ['session_token', 'm2m_token'] })).toExtend< |
| 76 | + SessionAuthWithRedirect | MachineAuthObject<'m2m_token'> | InvalidTokenAuthObject |
| 77 | + >(); |
| 78 | + expectTypeOf(await auth({ acceptsToken: ['m2m_token', 'oauth_token'] })).toExtend< |
| 79 | + MachineAuthObject<'m2m_token' | 'oauth_token'> | InvalidTokenAuthObject |
| 80 | + >(); |
| 81 | + |
| 82 | + // Any token type |
| 83 | + expectTypeOf(await auth({ acceptsToken: 'any' })).toExtend<AuthObject>(); |
| 84 | + }); |
37 | 85 |
|
38 |
| - const auth = getAuth(request, { acceptsToken: 'any' }); |
| 86 | + test('verifies discriminated union works correctly with acceptsToken: any', async () => { |
| 87 | + const authObject = await auth({ acceptsToken: 'any' }); |
39 | 88 |
|
40 |
| - if (auth.tokenType === 'session_token') { |
41 |
| - expectTypeOf(auth).toMatchTypeOf<SessionAuthObject>(); |
42 |
| - } else if (auth.tokenType === 'api_key') { |
43 |
| - expectTypeOf(auth).toMatchTypeOf<MachineAuthObject<'api_key'>>(); |
44 |
| - } else if (auth.tokenType === 'm2m_token') { |
45 |
| - expectTypeOf(auth).toMatchTypeOf<MachineAuthObject<'m2m_token'>>(); |
46 |
| - } else if (auth.tokenType === 'oauth_token') { |
47 |
| - expectTypeOf(auth).toMatchTypeOf<MachineAuthObject<'oauth_token'>>(); |
48 |
| - } |
| 89 | + if (authObject.tokenType === 'session_token') { |
| 90 | + expectTypeOf(authObject).toExtend<SessionAuthWithRedirect>(); |
| 91 | + } else if (authObject.tokenType === 'api_key') { |
| 92 | + expectTypeOf(authObject).toExtend<MachineAuthObject<'api_key'>>(); |
| 93 | + } else if (authObject.tokenType === 'm2m_token') { |
| 94 | + expectTypeOf(authObject).toExtend<MachineAuthObject<'m2m_token'>>(); |
| 95 | + } else if (authObject.tokenType === 'oauth_token') { |
| 96 | + expectTypeOf(authObject).toExtend<MachineAuthObject<'oauth_token'>>(); |
| 97 | + } |
| 98 | + }); |
49 | 99 | });
|
0 commit comments