fmt

Author: rreddy2

nodestream_github/client/githubclient.py

@@ -92,6 +92,7 @@ def validate_positive_int(value: int) -> int:
         exception_msg = "Formatting lookback period failed"
         raise ValueError(exception_msg) from e
 
+
 def generate_date_range(lookback_period: dict[str, int]) -> list[str]:
     """
     Generate a list of date strings in YYYY-MM-DD format for
@@ -105,10 +106,10 @@ def generate_date_range(lookback_period: dict[str, int]) -> list[str]:
 
     delta_days = (end_date - start_date).days + 1
     return [
-        (start_date + timedelta(days=i)).strftime("%Y-%m-%d")
-        for i in range(delta_days)
+        (start_date + timedelta(days=i)).strftime("%Y-%m-%d") for i in range(delta_days)
     ]
 
+
 def build_search_phrase(
     actions: list[str],
     actors: list[str],
@@ -423,6 +424,7 @@ async def fetch_enterprise_audit_log(
             dates = generate_date_range(lookback_period) or [None]
 
             for target_date in dates:
+                print(target_date)
                 search_phrase = build_search_phrase(
                     actions=actions,
                     actors=actors,

tests/client/test_githubclient.py

@@ -135,25 +135,30 @@ def test_all_null_args():
     # noinspection PyTypeChecker
     assert GithubRestApiClient(auth_token=None, github_hostname=None)
 
+
 # Test generate_date_range
 
+
 def test_generate_date_range_empty_lookback_period():
     result = generate_date_range({})
     assert result == []
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_days_only():
     result = generate_date_range({"days": 3})
     expected = ["2025-07-29", "2025-07-30", "2025-07-31", "2025-08-01"]
     assert result == expected
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_zero_days():
     """Test with zero days (same day only)."""
     result = generate_date_range({"days": 0})
     expected = ["2025-08-01"]
     assert result == expected
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_months_only():
     result = generate_date_range({"months": 1})
@@ -162,20 +167,23 @@ def test_generate_date_range_months_only():
     assert result[0] == "2025-07-01"
     assert result[-1] == "2025-08-01"
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_years_only():
     result = generate_date_range({"years": 1})
     assert len(result) == 366
     assert result[0] == "2024-08-01"
     assert result[-1] == "2025-08-01"
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_combined_periods():
     """Test with combined periods."""
     result = generate_date_range({"months": 1, "days": 5})
     assert result[0] == "2025-06-26"
     assert result[-1] == "2025-08-01"
 
+
 @freeze_time("2025-08-01")
 def test_generate_date_range_complex_combination():
     """Test with years, months, and days."""
@@ -186,6 +194,7 @@ def test_generate_date_range_complex_combination():
 
 # Test validate_lookback_period
 
+
 def test_validate_lookback_period_valid_input():
     result = validate_lookback_period({"days": 7, "months": 2, "years": 1})
     expected = {"days": 7, "months": 2, "years": 1}
@@ -207,6 +216,7 @@ def test_validate_lookback_period_string_to_int_conversion():
     expected = {"days": 7, "months": 2}
     assert result == expected
 
+
 def test_validate_lookback_period_zero_value():
     with pytest.raises(ValueError, match="Formatting lookback period failed"):
         validate_lookback_period({"days": 0})
@@ -244,4 +254,4 @@ def test_validate_lookback_period_invalid_type():
 
 def test_validate_lookback_period_none_value():
     with pytest.raises(ValueError, match="Formatting lookback period failed"):
-        validate_lookback_period({"days": None})
+        validate_lookback_period({"days": None})

tests/test_audit.py

@@ -164,7 +164,7 @@ async def test_get_audit_lookback_periods(
     expected_path: str,
 ):
     from nodestream_github.client.githubclient import generate_date_range
-    
+
     extractor = GithubAuditLogExtractor(
         auth_token="test-token",
         github_hostname=DEFAULT_HOSTNAME,
@@ -177,14 +177,14 @@ async def test_get_audit_lookback_periods(
     )
 
     expected_dates = generate_date_range(lookback_period)
-    
+
     # Mock the first date call with the expected_path
     gh_rest_mock.get_enterprise_audit_logs(
         status_code=200,
         search_phrase=expected_path,
         json=GITHUB_AUDIT,
     )
-    
+
     # Mock additional dates if there are more than one
     test_dates = expected_dates[:3] if len(expected_dates) > 3 else expected_dates
     if len(test_dates) > 1:
@@ -198,9 +198,10 @@ async def test_get_audit_lookback_periods(
 
     # replacing generate_date_range with test dates so that we don't iterate through all dates
     import nodestream_github.client.githubclient as client_module
+
     original_generate = client_module.generate_date_range
     client_module.generate_date_range = lambda x: test_dates
-    
+
     try:
         all_records = [record async for record in extractor.extract_records()]
         expected_output = GITHUB_EXPECTED_OUTPUT * len(test_dates)