diff --git a/tests/unit/cli/test_entrypoint.py b/tests/unit/cli/test_entrypoint.py index bc6cdb68..08ae5b87 100644 --- a/tests/unit/cli/test_entrypoint.py +++ b/tests/unit/cli/test_entrypoint.py @@ -260,15 +260,15 @@ def test_transaction_log_to_logger( def test_output_summary(): - subscription_values: List[Tuple[str, int, int, int, int]] = [ - ("long_name_but_lil_values", 0, 0, 0, 6), - ("john_smith", 1, 0, 0, 52), - ("david_gore", 0, 0, 0, 4), - ("christopher_snoop", 50, 0, 3, 518), - ("beyond funk", 0, 0, 0, 176), + subscription_values: List[Tuple[str, int, int, int, int, bool]] = [ + ("long_name_but_lil_values", 0, 0, 0, 6, False), + ("john_smith", 1, 0, 0, 52, False), + ("david_gore", 0, 0, 0, 4, False), + ("christopher_snoop", 50, 0, 3, 518, False), + ("beyond funk", 0, 0, 0, 176, True), ] - mock_subscriptions: List[Tuple[MagicMock, FileHandlerTransactionLog]] = [] + mock_subscriptions: List[MagicMock] = [] for values in subscription_values: sub = Mock() sub.name = values[0] @@ -276,10 +276,11 @@ def test_output_summary(): sub.num_entries_modified = values[2] sub.num_entries_removed = values[3] sub.num_entries = values[4] + sub.exception = values[5] - mock_subscriptions.append((sub, FileHandlerTransactionLog())) + mock_subscriptions.append(sub) - _ = output_summary(transaction_logs=mock_subscriptions) + _ = output_summary(subscriptions=mock_subscriptions) assert True # Test used for manual inspection - too hard to test ansi color codes