shortening the tests - somewhat

Author: rreddy2

tests/test_audit.py

@@ -1,3 +1,5 @@
+from datetime import datetime, timedelta, timezone
+
 import pytest
 from freezegun import freeze_time
 
@@ -150,25 +152,21 @@ async def test_get_audit_parameterized(
                 "2025-08-01",
             ],
         ),
-        (
-            {"months": 2},
-            "action:protected_branch.create created:2025-06-01",
-            ["2025-06-01", "2025-06-02", "2025-06-03"],
-        ),
-        (
-            {"years": 1},
-            "action:protected_branch.create created:2024-08-01",
-            ["2024-08-01", "2024-08-02", "2024-08-03"],
-        ),
         (
             {"days": 15, "months": 1},
             "action:protected_branch.create created:2025-06-16",
-            ["2025-06-16", "2025-06-17", "2025-06-18"],
-        ),
-        (
-            {"days": 10, "months": 1, "years": 1},
-            "action:protected_branch.create created:2024-06-21",
-            ["2024-06-21", "2024-06-22", "2024-06-23"],
+            [
+                (
+                    datetime(2025, 6, 16, tzinfo=timezone.utc) + timedelta(days=i)
+                ).strftime("%Y-%m-%d")
+                for i in range(
+                    (
+                        datetime(2025, 8, 1, tzinfo=timezone.utc)
+                        - datetime(2025, 6, 16, tzinfo=timezone.utc)
+                    ).days
+                    + 1
+                )
+            ],
         ),
     ],
 )
@@ -205,19 +203,9 @@ async def test_get_audit_lookback_periods(
                 json=GITHUB_AUDIT,
             )
 
-    # replacing generate_date_range with test dates
-    # so that we don't iterate through all dates
-    import nodestream_github.audit as client_module
-
-    original_generate = client_module.generate_date_range
-    client_module.generate_date_range = lambda x: expected_dates
-
-    try:
-        all_records = [record async for record in extractor.extract_records()]
-        expected_output = GITHUB_EXPECTED_OUTPUT * len(expected_dates)
-        assert all_records == expected_output
-    finally:
-        client_module.generate_date_range = original_generate
+    all_records = [record async for record in extractor.extract_records()]
+    expected_output = GITHUB_EXPECTED_OUTPUT * len(expected_dates)
+    assert all_records == expected_output
 
 
 # Test generate_date_range