def test_not_found(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): mock_switch_repository.delete_switch = MagicMock( side_effect=SwitchNotFoundError) with raises(SwitchNotFoundError): switch_manager.delete(ctx, TEST_SWITCH_ID)
def test_switch_not_found(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): mock_switch_repository.search_switches_by = MagicMock(return_value=([], 0)) with raises(SwitchNotFoundError): switch_manager.get_by_id(ctx, TEST_SWITCH_ID)
def test_happy_path(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): mock_switch_repository.delete_switch = MagicMock() switch_manager.delete(ctx, TEST_SWITCH_ID) mock_switch_repository.delete_switch.assert_called_once_with( ctx, TEST_SWITCH_ID)
def test_happy_path(self, ctx, mock_switch_repository: SwitchRepository, sample_switch: Switch, switch_manager: SwitchManager): mock_switch_repository.search_switches_by = MagicMock( return_value=([sample_switch], 1)) result = switch_manager.get_by_id(ctx, TEST_SWITCH_ID) assert sample_switch == result mock_switch_repository.search_switches_by.assert_called_once()
def test_switch_not_found(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): req = MutationRequest( description='desc', ip_v4='157.159.123.123', community='ip', ) mock_switch_repository.update_switch = MagicMock( side_effect=SwitchNotFoundError) with raises(SwitchNotFoundError): switch_manager.update(ctx, '2', req)
def test_happy_path(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): req = MutationRequest( description='desc', ip_v4='157.159.123.123', community='ip', ) mock_switch_repository.update_switch = MagicMock() switch_manager.update(ctx, '2', req) mock_switch_repository.update_switch.assert_called_once_with( ctx, switch_id='2', **asdict(req))
def test_happy_path(self, ctx, mock_switch_repository: SwitchRepository, sample_switch: Switch, switch_manager: SwitchManager): mock_switch_repository.search_switches_by = MagicMock( return_value=([sample_switch], 1)) result, count = switch_manager.search(ctx, limit=42, offset=2, terms='abc') assert [sample_switch] == result assert 1 == count mock_switch_repository.search_switches_by.assert_called_once_with( ctx, limit=42, offset=2, terms='abc')
def test_invalid_mutation_request(self, ctx, mock_switch_repository: SwitchRepository, field: str, value, switch_manager: SwitchManager): req = MutationRequest( description='desc', ip_v4='157.159.123.123', community='ip', ) req = MutationRequest(**{**asdict(req), **{field: value}}) mock_switch_repository.update_switch = MagicMock() with raises(ValueError): switch_manager.update(ctx, '2', req)
def test_missing_required_field(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): req = MutationRequest( description='desc', ip_v4='157.159.123.123', community='ip', ) req.community = None mock_switch_repository.update_switch = MagicMock() with raises(MissingRequiredField): switch_manager.update(ctx, '2', req) mock_switch_repository.update_switch.assert_not_called()
def test_happy_path(self, ctx, mock_switch_repository: SwitchRepository, switch_manager: SwitchManager): req = MutationRequest( description='desc', ip_v4='157.159.123.123', community='ip', ) mock_switch_repository.create_switch = MagicMock() switch_manager.create(ctx, req) mock_switch_repository.create_switch.assert_called_once_with( ctx, description=req.description, ip_v4=req.ip_v4, community=req.community)