cleanup

Author: abretonc7s

app/components/UI/Perps/Views/PerpsClosePositionView/PerpsClosePositionView.test.tsx

@@ -13,7 +13,7 @@ import {
   defaultPerpsClosePositionValidationMock,
   defaultPerpsEventTrackingMock,
   defaultPerpsLivePricesMock,
-  defaultPerpsOrderBookMock,
+  defaultPerpsTopOfBookMock,
   defaultPerpsOrderFeesMock,
   defaultPerpsPositionMock,
   defaultPerpsRewardsMock,
@@ -52,7 +52,7 @@ jest.mock('../../hooks', () => ({
 
 jest.mock('../../hooks/stream', () => ({
   usePerpsLivePrices: jest.fn(),
-  usePerpsOrderBook: jest.fn(),
+  usePerpsTopOfBook: jest.fn(),
 }));
 
 jest.mock('../../hooks/usePerpsEventTracking', () => ({
@@ -122,8 +122,8 @@ describe('PerpsClosePositionView', () => {
   const usePerpsLivePricesMock = jest.mocked(
     jest.requireMock('../../hooks/stream').usePerpsLivePrices,
   );
-  const usePerpsOrderBookMock = jest.mocked(
-    jest.requireMock('../../hooks/stream').usePerpsOrderBook,
+  const usePerpsTopOfBookMock = jest.mocked(
+    jest.requireMock('../../hooks/stream').usePerpsTopOfBook,
   );
   const usePerpsOrderFeesMock = jest.mocked(
     jest.requireMock('../../hooks').usePerpsOrderFees,
@@ -167,7 +167,7 @@ describe('PerpsClosePositionView', () => {
 
     // Setup hook mocks with default values
     usePerpsLivePricesMock.mockReturnValue(defaultPerpsLivePricesMock);
-    usePerpsOrderBookMock.mockReturnValue(defaultPerpsOrderBookMock);
+    usePerpsTopOfBookMock.mockReturnValue(defaultPerpsTopOfBookMock);
     usePerpsOrderFeesMock.mockReturnValue(defaultPerpsOrderFeesMock);
     usePerpsClosePositionValidationMock.mockReturnValue(
       defaultPerpsClosePositionValidationMock,

app/components/UI/Perps/Views/PerpsOrderView/PerpsOrderView.test.tsx

@@ -571,26 +571,23 @@ const createMockStreamManager = () => {
     account: {
       subscribe: jest.fn(() => jest.fn()),
     },
-    orderBooks: {
-      subscribeToSymbols: ({
-        symbols,
+    topOfBook: {
+      subscribeToSymbol: ({
+        symbol: _symbol,
         callback,
       }: {
-        symbols: string[];
+        symbol: string;
         callback: (data: unknown) => void;
       }) => {
         const id = Math.random().toString();
         subscribers.set(id, callback);
-        // Immediately provide mock order book data
-        const mockOrderBooks: Record<string, unknown> = {};
-        symbols.forEach((symbol: string) => {
-          mockOrderBooks[symbol] = {
-            bestBid: '2999',
-            bestAsk: '3001',
-            spread: '2',
-          };
-        });
-        callback(mockOrderBooks);
+        // Immediately provide mock top of book data
+        const mockTopOfBook = {
+          bestBid: '2999',
+          bestAsk: '3001',
+          spread: '2',
+        };
+        callback(mockTopOfBook);
         return () => {
           subscribers.delete(id);
         };

app/components/UI/Perps/providers/PerpsStreamManager.test.tsx

@@ -1414,12 +1414,12 @@ describe('PerpsStreamManager', () => {
     });
   });
 
-  describe('OrderBookStreamChannel', () => {
-    it('subscribes to order books with includeOrderBook flag', () => {
+  describe('TopOfBookStreamChannel', () => {
+    it('subscribes to top of book with includeOrderBook flag', () => {
       const callback = jest.fn();
 
-      testStreamManager.orderBooks.subscribeToSymbols({
-        symbols: ['BTC'],
+      testStreamManager.topOfBook.subscribeToSymbol({
+        symbol: 'BTC',
         callback,
       });
 
@@ -1430,11 +1430,11 @@ describe('PerpsStreamManager', () => {
       });
     });
 
-    it('filters order books to requested symbols', () => {
+    it('provides top of book data for subscribed symbol', () => {
       const callback = jest.fn();
 
-      testStreamManager.orderBooks.subscribeToSymbols({
-        symbols: ['BTC'],
+      testStreamManager.topOfBook.subscribeToSymbol({
+        symbol: 'BTC',
         callback,
       });
 
@@ -1445,35 +1445,37 @@ describe('PerpsStreamManager', () => {
       ]);
 
       expect(callback).toHaveBeenCalledWith({
-        BTC: { bestBid: '50000', bestAsk: '50001', spread: undefined },
+        bestBid: '50000',
+        bestAsk: '50001',
+        spread: undefined,
       });
     });
 
-    it('does not subscribe when symbols array is empty', () => {
+    it('does not subscribe when symbol is empty', () => {
       const callback = jest.fn();
 
-      testStreamManager.orderBooks.subscribeToSymbols({
-        symbols: [],
+      testStreamManager.topOfBook.subscribeToSymbol({
+        symbol: '',
         callback,
       });
 
       expect(mockSubscribeToPrices).not.toHaveBeenCalled();
     });
 
-    it('clears order book cache when clearCache called', () => {
+    it('clears top of book cache when clearCache called', () => {
       const callback = jest.fn();
 
-      testStreamManager.orderBooks.subscribeToSymbols({
-        symbols: ['BTC'],
+      testStreamManager.topOfBook.subscribeToSymbol({
+        symbol: 'BTC',
         callback,
       });
 
       const priceCallback = mockSubscribeToPrices.mock.calls[0][0].callback;
       priceCallback([{ coin: 'BTC', bestBid: '50000', bestAsk: '50001' }]);
 
-      testStreamManager.orderBooks.clearCache();
+      testStreamManager.topOfBook.clearCache();
 
-      expect(callback).toHaveBeenCalledWith({});
+      expect(callback).toHaveBeenCalledWith(undefined);
     });
   });
 });