test: stop pook mocks leaking between REST test classes

test_rest/base.py

@@ -3,6 +3,7 @@
 import os
 import re
 import unittest
+from typing import Set
 import pook  # type: ignore
 
 # mocks are stored in file tree
@@ -32,6 +33,10 @@
 
 
 class BaseTest(unittest.TestCase):
+    # mocks that answer more than once, for tests that repeat a request;
+    # others answer once, which the pagination tests rely on
+    persistent_mocks: Set[str] = set()
+
     @classmethod
     def setUpClass(cls):
         cls.maxDiff = None
@@ -40,4 +45,14 @@ def setUpClass(cls):
         for m in mocks:
             url = cls.c.BASE + m[0]
             # print("register", url)
-            pook.get(url, reply=200, response_body=m[1])
+            pook.get(
+                url,
+                reply=200,
+                response_body=m[1],
+                persist=m[0] in cls.persistent_mocks,
+            )
+
+    @classmethod
+    def tearDownClass(cls):
+        # stop pook and drop the mocks, so they don't leak into later classes
+        pook.off()

test_rest/test_dividends.py

@@ -3,6 +3,9 @@
 
 
 class DividendsTest(BaseTest):
+    # test_list_stocks_dividends_ticker_any_of requests this twice
+    persistent_mocks = {"/stocks/v1/dividends?ticker.any_of=AAPL%2CVZ"}
+
     def test_list_dividends(self):
         dividends = [d for d in self.c.list_dividends()]
         expected = [

test_rest/test_snapshots.py

@@ -26,6 +26,9 @@
 
 
 class SnapshotsTest(BaseTest):
+    # test_get_snapshot_indices requests this twice
+    persistent_mocks = {"/v3/snapshot/indices?ticker.any_of=SPX%2CAPx%2CAPy"}
+
     def test_list_universal_snapshots(self):
         expected = [
             UniversalSnapshot(