Merge branch 'main' into feature/webui
This commit is contained in:
Badri Narayanan S
@@ -14,7 +14,9 @@ const tests = [
|
||||
{ name: 'Multi-turn Tools (Streaming)', file: 'test-multiturn-thinking-tools-streaming.cjs' },
|
||||
{ name: 'Interleaved Thinking', file: 'test-interleaved-thinking.cjs' },
|
||||
{ name: 'Image Support', file: 'test-images.cjs' },
|
||||
{ name: 'Prompt Caching', file: 'test-caching-streaming.cjs' }
|
||||
{ name: 'Prompt Caching', file: 'test-caching-streaming.cjs' },
|
||||
{ name: 'Cross-Model Thinking', file: 'test-cross-model-thinking.cjs' },
|
||||
{ name: 'OAuth No-Browser Mode', file: 'test-oauth-no-browser.cjs' }
|
||||
];
|
||||
|
||||
async function runTest(test) {
|
||||
|
||||
217
tests/test-oauth-no-browser.cjs
Normal file
217
tests/test-oauth-no-browser.cjs
Normal file
@@ -0,0 +1,217 @@
|
||||
/**
|
||||
* OAuth No-Browser Mode Unit Tests
|
||||
*
|
||||
* Tests the extractCodeFromInput() function which enables OAuth authentication
|
||||
* on headless servers without a desktop environment.
|
||||
*
|
||||
* ============================================================================
|
||||
* FEATURE: --no-browser OAuth Mode
|
||||
* ============================================================================
|
||||
*
|
||||
* PURPOSE:
|
||||
* Allow users to add Google accounts on remote servers (headless Linux,
|
||||
* Docker containers, SSH sessions) where automatic browser opening is
|
||||
* not possible.
|
||||
*
|
||||
* USAGE:
|
||||
* npm run accounts:add -- --no-browser
|
||||
*
|
||||
* USER FLOW:
|
||||
* 1. User runs command on headless server
|
||||
* 2. System displays Google OAuth URL
|
||||
* 3. User opens URL on another device (phone/laptop) with a browser
|
||||
* 4. User signs in to Google and authorizes the app
|
||||
* 5. Browser redirects to localhost (page won't load - this is expected)
|
||||
* 6. User copies the redirect URL or authorization code from address bar
|
||||
* 7. User pastes into server terminal
|
||||
* 8. System extracts code using extractCodeFromInput() (tested here)
|
||||
* 9. Account is added successfully
|
||||
*
|
||||
* FUNCTION UNDER TEST:
|
||||
* extractCodeFromInput(input: string) => { code: string, state: string|null }
|
||||
*
|
||||
* Accepts either:
|
||||
* - Full callback URL: http://localhost:51121/callback?code=xxx&state=yyy
|
||||
* - Raw authorization code: 4/0AQSTgQG...
|
||||
*
|
||||
* Throws on:
|
||||
* - Empty/null input
|
||||
* - Too short input (< 10 chars)
|
||||
* - URL with OAuth error parameter
|
||||
* - URL without code parameter
|
||||
*
|
||||
* ============================================================================
|
||||
*
|
||||
* Run: node tests/test-oauth-no-browser.cjs
|
||||
*/
|
||||
|
||||
// Note: Using dynamic import because oauth.js is ESM
|
||||
async function runTests() {
|
||||
console.log('='.repeat(60));
|
||||
console.log('OAUTH NO-BROWSER MODE UNIT TESTS');
|
||||
console.log('Testing: extractCodeFromInput()');
|
||||
console.log('='.repeat(60));
|
||||
console.log('');
|
||||
|
||||
// Import the ESM module
|
||||
const { extractCodeFromInput } = await import('../src/auth/oauth.js');
|
||||
|
||||
let allPassed = true;
|
||||
const results = [];
|
||||
|
||||
/**
|
||||
* Helper to run a single test case
|
||||
* @param {string} name - Test name
|
||||
* @param {Function} testFn - Test function that returns { passed, message }
|
||||
*/
|
||||
async function test(name, testFn) {
|
||||
try {
|
||||
const { passed, message } = await testFn();
|
||||
results.push({ name, passed, message });
|
||||
const status = passed ? 'PASS' : 'FAIL';
|
||||
console.log(` [${status}] ${name}`);
|
||||
if (message) console.log(` ${message}`);
|
||||
if (!passed) allPassed = false;
|
||||
} catch (error) {
|
||||
results.push({ name, passed: false, message: error.message });
|
||||
console.log(` [FAIL] ${name}`);
|
||||
console.log(` Error: ${error.message}`);
|
||||
allPassed = false;
|
||||
}
|
||||
}
|
||||
|
||||
// ===== Test Group 1: Valid URL Inputs =====
|
||||
console.log('\n--- Valid URL Inputs ---');
|
||||
|
||||
await test('Parse full callback URL with code and state', () => {
|
||||
const input = 'http://localhost:51121/oauth-callback?code=4/0AQSTg123&state=abc123';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === '4/0AQSTg123' && result.state === 'abc123';
|
||||
return { passed, message: `code=${result.code}, state=${result.state}` };
|
||||
});
|
||||
|
||||
await test('Parse URL with only code (no state)', () => {
|
||||
const input = 'http://localhost:51121/oauth-callback?code=4/0AQSTg456';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === '4/0AQSTg456' && result.state === null;
|
||||
return { passed, message: `code=${result.code}, state=${result.state}` };
|
||||
});
|
||||
|
||||
await test('Parse HTTPS URL', () => {
|
||||
const input = 'https://localhost:51121/callback?code=secureCode123&state=xyz';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === 'secureCode123';
|
||||
return { passed, message: `code=${result.code}` };
|
||||
});
|
||||
|
||||
await test('Parse URL with additional query params', () => {
|
||||
const input = 'http://localhost:51121/?code=myCode&state=myState&scope=email';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === 'myCode' && result.state === 'myState';
|
||||
return { passed, message: `code=${result.code}, state=${result.state}` };
|
||||
});
|
||||
|
||||
// ===== Test Group 2: Raw Code Inputs =====
|
||||
console.log('\n--- Raw Authorization Code Inputs ---');
|
||||
|
||||
await test('Parse raw authorization code (Google format)', () => {
|
||||
const input = '4/0AQSTgQGxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === input && result.state === null;
|
||||
return { passed, message: `code length=${result.code.length}` };
|
||||
});
|
||||
|
||||
await test('Parse raw code with whitespace (should trim)', () => {
|
||||
const input = ' 4/0AQSTgQGcode123 \n';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === '4/0AQSTgQGcode123' && result.state === null;
|
||||
return { passed, message: `trimmed code=${result.code}` };
|
||||
});
|
||||
|
||||
// ===== Test Group 3: Error Cases =====
|
||||
console.log('\n--- Error Handling ---');
|
||||
|
||||
await test('Throw on empty input', () => {
|
||||
try {
|
||||
extractCodeFromInput('');
|
||||
return { passed: false, message: 'Should have thrown' };
|
||||
} catch (e) {
|
||||
return { passed: e.message.includes('No input'), message: e.message };
|
||||
}
|
||||
});
|
||||
|
||||
await test('Throw on null input', () => {
|
||||
try {
|
||||
extractCodeFromInput(null);
|
||||
return { passed: false, message: 'Should have thrown' };
|
||||
} catch (e) {
|
||||
return { passed: e.message.includes('No input'), message: e.message };
|
||||
}
|
||||
});
|
||||
|
||||
await test('Throw on too short code', () => {
|
||||
try {
|
||||
extractCodeFromInput('abc');
|
||||
return { passed: false, message: 'Should have thrown' };
|
||||
} catch (e) {
|
||||
return { passed: e.message.includes('too short'), message: e.message };
|
||||
}
|
||||
});
|
||||
|
||||
await test('Throw on OAuth error in URL', () => {
|
||||
try {
|
||||
const input = 'http://localhost:51121/?error=access_denied&error_description=User%20denied';
|
||||
extractCodeFromInput(input);
|
||||
return { passed: false, message: 'Should have thrown' };
|
||||
} catch (e) {
|
||||
return { passed: e.message.includes('OAuth error'), message: e.message };
|
||||
}
|
||||
});
|
||||
|
||||
await test('Throw on URL without code param', () => {
|
||||
try {
|
||||
extractCodeFromInput('http://localhost:51121/callback?state=onlyState');
|
||||
return { passed: false, message: 'Should have thrown' };
|
||||
} catch (e) {
|
||||
return { passed: e.message.includes('No authorization code'), message: e.message };
|
||||
}
|
||||
});
|
||||
|
||||
// ===== Test Group 4: Edge Cases =====
|
||||
console.log('\n--- Edge Cases ---');
|
||||
|
||||
await test('Handle URL-encoded characters in code', () => {
|
||||
const input = 'http://localhost:51121/?code=4%2F0AQSTg%2B%2B&state=test';
|
||||
const result = extractCodeFromInput(input);
|
||||
// URL class automatically decodes
|
||||
const passed = result.code === '4/0AQSTg++';
|
||||
return { passed, message: `decoded code=${result.code}` };
|
||||
});
|
||||
|
||||
await test('Accept minimum valid code length (10 chars)', () => {
|
||||
const input = '1234567890';
|
||||
const result = extractCodeFromInput(input);
|
||||
const passed = result.code === input;
|
||||
return { passed, message: `code=${result.code}` };
|
||||
});
|
||||
|
||||
// ===== Summary =====
|
||||
console.log('\n' + '='.repeat(60));
|
||||
console.log('SUMMARY');
|
||||
console.log('='.repeat(60));
|
||||
|
||||
const passed = results.filter(r => r.passed).length;
|
||||
const failed = results.filter(r => !r.passed).length;
|
||||
console.log(` Total: ${results.length} | Passed: ${passed} | Failed: ${failed}`);
|
||||
|
||||
console.log('\n' + '='.repeat(60));
|
||||
console.log(allPassed ? 'ALL TESTS PASSED' : 'SOME TESTS FAILED');
|
||||
console.log('='.repeat(60));
|
||||
|
||||
process.exit(allPassed ? 0 : 1);
|
||||
}
|
||||
|
||||
runTests().catch(err => {
|
||||
console.error('Test suite failed:', err);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user