def test_restart_peer_groups(): frr = MagicMock() c = ConfigMgr(frr) c.restart_peer_groups(["pg_1", "pg_2"]) assert c.peer_groups_to_restart == ["pg_1", "pg_2"] c.restart_peer_groups(["pg_3", "pg_4"]) assert c.peer_groups_to_restart == ["pg_1", "pg_2", "pg_3", "pg_4"]
def commit_changes_common(write_error, restart_error, result): frr = MagicMock() frr.write = MagicMock(return_value=write_error) frr.restart_peer_groups = MagicMock(return_value=restart_error) c = ConfigMgr(frr) c.reset = MagicMock() c.push_list(["change1", "change2"]) c.restart_peer_groups(["pg1", "pg2"]) res = c.commit() assert res == result assert c.reset.called frr.write.assert_called_with('change1\nchange2\n') frr.restart_peer_groups.assert_called_with(["pg1", "pg2"])