Skip to content

Commit

Permalink
chore: add pagerduty migrator tests
Browse files Browse the repository at this point in the history
  • Loading branch information
joeyorlando committed Dec 18, 2024
1 parent 54ff63a commit 45775fd
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions tools/migrators/lib/tests/pagerduty/test_migrate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
from unittest.mock import call, patch

from lib.pagerduty.migrate import migrate


@patch("lib.pagerduty.migrate.MIGRATE_USERS", False)
@patch("lib.pagerduty.migrate.APISession")
@patch("lib.pagerduty.migrate.OnCallAPIClient")
def test_users_are_skipped_when_migrate_users_is_false(MockOnCallAPIClient, MockAPISession):
mock_session = MockAPISession.return_value
mock_session.list_all.return_value = []
mock_oncall_client = MockOnCallAPIClient.return_value

migrate()

# Assert no user-related fetching or migration occurs
assert mock_session.list_all.call_args_list == [
call("schedules", params={"include[]": "schedule_layers", "time_zone": "UTC"}),
call("escalation_policies"),
call("services", params={"include[]": "integrations"}),
call("vendors"),
# no user notification rules fetching
]

mock_oncall_client.list_users_with_notification_rules.assert_not_called()

0 comments on commit 45775fd

Please sign in to comment.