def test_register_resource_two_resources(): context = Context() resource = Mock() resource.get_resource_keys.return_value = [890, 981] context.resources = {123: "hoho", 567: "bebe"} # in real, "hoho" and "bebe" will be resource instances context.current_resource_set = Mock() context.register_resource(resource) assert context.current_resource_set.mock_calls == [call.add_item(resource), call.add_item(resource)]
def test_register_resource_conflicts(): context = Context() resource = Mock() resource2 = Mock() resource.get_resource_keys.return_value = [123] context.resources = {123: resource2} # in real, "hoho" and "bebe" will be resource instances context.current_resource_set = Mock() context.register_resource(resource) assert resource2.mock_calls == [call.resolve_conflict(resource)] assert context.current_resource_set.mock_calls == [call.add_item(resource2)]