feat(unified-routing-api): support uniswap-extension request source (#…

…432)
Uniswap · Jun 27, 2024 · 03d5925 · 03d5925
1 parent 0c8b6b6
commit 03d5925
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/entities/request/index.ts b/lib/entities/request/index.ts
@@ -62,6 +62,7 @@ export enum RequestSource {
   UNISWAP_WEB = 'uniswap-web',
   EXTERNAL_API = 'external-api',
   EXTERNAL_API_MOBILE = 'external-api:mobile',
+  UNISWAP_EXTENSION = 'uniswap-extension',
 }
 
 export interface QuoteRequest {

diff --git a/test/unit/lib/handlers/quote/handler.test.ts b/test/unit/lib/handlers/quote/handler.test.ts
@@ -332,6 +332,10 @@ describe('QuoteHandler', () => {
           requestSource = quoteHandler.getQuoteRequestSource(headers);
           expect(requestSource).toBe(RequestSource.UNKNOWN);
 
+          headers = {'x-request-source': 'uniswap-extension'};
+          requestSource = quoteHandler.getQuoteRequestSource(headers);
+          expect(requestSource).toBe(RequestSource.UNISWAP_EXTENSION);
+
           headers = { 'x-request-source': '' };
           requestSource = quoteHandler.getQuoteRequestSource(headers);
           expect(requestSource).toBe(RequestSource.UNKNOWN);