class TestViewsetBindedSchemas(unittest.TestCase):
def setUp(self):
self.viewset = ViewSet()
self.viewset.responses = mock.MagicMock()
self.resource = mock.MagicMock()
def test_request_schemas_have_header_and_querystring(self):
self.viewset = ViewSet(
default_get_arguments={},
default_record_arguments={},
record_get_arguments={}
)
arguments = self.viewset.record_arguments(self.resource, 'GET')
schema = arguments['schema']
self.assertIn('querystring', schema)
self.assertIn('header', schema)
def test_payload_request_schemas_have_a_body(self):
arguments = self.viewset.record_arguments(self.resource, 'PUT')
schema = arguments['schema']
self.assertIn('body', schema)
def test_collection_deserialize_sort(self):
arguments = self.viewset.collection_arguments(self.resource, 'DELETE')
schema = arguments['schema']
value = {'querystring': {'_sort': 'foo,-bar'}}
deserialized = schema.deserialize(value)
expected = {'querystring': {'_sort': ['foo', '-bar']}}
self.assertEquals(deserialized, expected)
def test_get_collection_deserialize_fields(self):
arguments = self.viewset.collection_arguments(self.resource, 'GET')
schema = arguments['schema']
value = {'querystring': {'_fields': 'foo,bar'}}
deserialized = schema.deserialize(value)
expected = {'querystring': {'_fields': ['foo', 'bar']}}
self.assertEquals(deserialized, expected)
def test_get_record_deserialize_fields(self):
arguments = self.viewset.record_arguments(self.resource, 'GET')
schema = arguments['schema']
value = {'querystring': {'_fields': 'foo,bar'}}
deserialized = schema.deserialize(value)
expected = {'querystring': {'_fields': ['foo', 'bar']}}
self.assertEquals(deserialized, expected)
def test_patch_record_validate_response_behavior(self):
arguments = self.viewset.collection_arguments(self.resource, 'PATCH')
schema = arguments['schema']
invalid = {'header': {'Response-Behavior': 'impolite'}}
self.assertRaises(colander.Invalid, schema.deserialize, invalid)
class TestViewsetBindedSchemas(unittest.TestCase):
def setUp(self):
self.viewset = ViewSet()
self.viewset.responses = mock.MagicMock()
self.resource = mock.MagicMock()
def test_request_schemas_have_header_and_querystring(self):
self.viewset = ViewSet(default_get_arguments={},
default_record_arguments={},
record_get_arguments={})
arguments = self.viewset.record_arguments(self.resource, "GET")
schema = arguments["schema"]
self.assertIn("querystring", schema)
self.assertIn("header", schema)
def test_payload_request_schemas_have_a_body(self):
arguments = self.viewset.record_arguments(self.resource, "PUT")
schema = arguments["schema"]
self.assertIn("body", schema)
def test_collection_deserialize_sort(self):
arguments = self.viewset.collection_arguments(self.resource, "DELETE")
schema = arguments["schema"]
value = {"querystring": {"_sort": "foo,-bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_sort": ["foo", "-bar"]}}
self.assertEqual(deserialized, expected)
def test_get_collection_deserialize_fields(self):
arguments = self.viewset.collection_arguments(self.resource, "GET")
schema = arguments["schema"]
value = {"querystring": {"_fields": "foo,bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_fields": ["foo", "bar"]}}
self.assertEqual(deserialized, expected)
def test_get_record_deserialize_fields(self):
arguments = self.viewset.record_arguments(self.resource, "GET")
schema = arguments["schema"]
value = {"querystring": {"_fields": "foo,bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_fields": ["foo", "bar"]}}
self.assertEqual(deserialized, expected)
def test_patch_record_validate_response_behavior(self):
arguments = self.viewset.collection_arguments(self.resource, "PATCH")
schema = arguments["schema"]
invalid = {"header": {"Response-Behavior": "impolite"}}
self.assertRaises(colander.Invalid, schema.deserialize, invalid)
class TestViewsetBindedSchemas(unittest.TestCase):
def setUp(self):
self.viewset = ViewSet()
self.viewset.responses = mock.MagicMock()
self.resource = mock.MagicMock()
def test_request_schemas_have_header_and_querystring(self):
self.viewset = ViewSet(
default_get_arguments={}, default_record_arguments={}, record_get_arguments={}
)
arguments = self.viewset.record_arguments(self.resource, "GET")
schema = arguments["schema"]
self.assertIn("querystring", schema)
self.assertIn("header", schema)
def test_payload_request_schemas_have_a_body(self):
arguments = self.viewset.record_arguments(self.resource, "PUT")
schema = arguments["schema"]
self.assertIn("body", schema)
def test_collection_deserialize_sort(self):
arguments = self.viewset.collection_arguments(self.resource, "DELETE")
schema = arguments["schema"]
value = {"querystring": {"_sort": "foo,-bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_sort": ["foo", "-bar"]}}
self.assertEqual(deserialized, expected)
def test_get_collection_deserialize_fields(self):
arguments = self.viewset.collection_arguments(self.resource, "GET")
schema = arguments["schema"]
value = {"querystring": {"_fields": "foo,bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_fields": ["foo", "bar"]}}
self.assertEqual(deserialized, expected)
def test_get_record_deserialize_fields(self):
arguments = self.viewset.record_arguments(self.resource, "GET")
schema = arguments["schema"]
value = {"querystring": {"_fields": "foo,bar"}}
deserialized = schema.deserialize(value)
expected = {"querystring": {"_fields": ["foo", "bar"]}}
self.assertEqual(deserialized, expected)
def test_patch_record_validate_response_behavior(self):
arguments = self.viewset.collection_arguments(self.resource, "PATCH")
schema = arguments["schema"]
invalid = {"header": {"Response-Behavior": "impolite"}}
self.assertRaises(colander.Invalid, schema.deserialize, invalid)
def test_class_parameters_are_used_for_collection_arguments(self):
default_arguments = {
'cors_headers': mock.sentinel.cors_headers,
}
default_get_arguments = {
'accept': mock.sentinel.accept,
}
default_collection_arguments = {
'cors_origins': mock.sentinel.cors_origins,
}
collection_get_arguments = {
'error_handler': mock.sentinel.error_handler
}
viewset = ViewSet(
default_arguments=default_arguments,
default_get_arguments=default_get_arguments,
default_collection_arguments=default_collection_arguments,
collection_get_arguments=collection_get_arguments
)
arguments = viewset.collection_arguments(mock.MagicMock, 'get')
arguments.pop('schema')
self.assertDictEqual(
arguments,
{
'accept': mock.sentinel.accept,
'cors_headers': mock.sentinel.cors_headers,
'cors_origins': mock.sentinel.cors_origins,
'error_handler': mock.sentinel.error_handler,
}
)
def test_class_parameters_are_used_for_collection_arguments(self, mocked):
default_arguments = {"cors_headers": mock.sentinel.cors_headers}
default_get_arguments = {"accept": mock.sentinel.accept}
default_collection_arguments = {"cors_origins": mock.sentinel.cors_origins}
collection_get_arguments = {"error_handler": mock.sentinel.error_handler}
viewset = ViewSet(
default_arguments=default_arguments,
default_get_arguments=default_get_arguments,
default_collection_arguments=default_collection_arguments,
collection_get_arguments=collection_get_arguments,
)
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), "get")
self.assertDictEqual(
arguments,
{
"schema": mocked().bind(),
"accept": mock.sentinel.accept,
"cors_headers": mock.sentinel.cors_headers,
"cors_origins": mock.sentinel.cors_origins,
"error_handler": mock.sentinel.error_handler,
"validators": [colander_validator],
"response_schemas": viewset.responses.get_and_bind(),
},
)
def test_default_arguments_are_copied_before_being_returned(self):
original_arguments = {}
viewset = ViewSet(collection_get_arguments=original_arguments)
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), 'GET')
self.assertEqual(original_arguments, {})
self.assertNotEquals(original_arguments, arguments)
def test_class_parameters_are_used_for_collection_arguments(self, mocked):
default_arguments = {"cors_headers": mock.sentinel.cors_headers}
default_get_arguments = {"accept": mock.sentinel.accept}
default_collection_arguments = {
"cors_origins": mock.sentinel.cors_origins
}
collection_get_arguments = {
"error_handler": mock.sentinel.error_handler
}
viewset = ViewSet(
default_arguments=default_arguments,
default_get_arguments=default_get_arguments,
default_collection_arguments=default_collection_arguments,
collection_get_arguments=collection_get_arguments,
)
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), "get")
self.assertDictEqual(
arguments,
{
"schema": mocked().bind(),
"accept": mock.sentinel.accept,
"cors_headers": mock.sentinel.cors_headers,
"cors_origins": mock.sentinel.cors_origins,
"error_handler": mock.sentinel.error_handler,
"validators": [colander_validator],
"response_schemas": viewset.responses.get_and_bind(),
},
)
def test_default_arguments_are_copied_before_being_returned(self):
original_arguments = {}
viewset = ViewSet(collection_get_arguments=original_arguments,
validate_schema_for=())
arguments = viewset.collection_arguments(mock.sentinel.resource, 'GET')
self.assertEquals(original_arguments, {})
self.assertNotEquals(original_arguments, arguments)
def test_class_parameters_are_used_for_collection_arguments(self, mocked):
default_arguments = {
'cors_headers': mock.sentinel.cors_headers,
}
default_get_arguments = {
'accept': mock.sentinel.accept,
}
default_collection_arguments = {
'cors_origins': mock.sentinel.cors_origins,
}
collection_get_arguments = {
'error_handler': mock.sentinel.error_handler
}
viewset = ViewSet(
default_arguments=default_arguments,
default_get_arguments=default_get_arguments,
default_collection_arguments=default_collection_arguments,
collection_get_arguments=collection_get_arguments)
arguments = viewset.collection_arguments(mock.MagicMock(), 'get')
self.assertDictEqual(
arguments, {
'schema': mocked().bind(),
'accept': mock.sentinel.accept,
'cors_headers': mock.sentinel.cors_headers,
'cors_origins': mock.sentinel.cors_origins,
'error_handler': mock.sentinel.error_handler,
'validators': [colander_validator]
})
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(
collection_methods=('GET', 'DELETE')
)
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), 'get')
self.assertIn('schema', arguments)
def test_class_parameters_are_used_for_collection_arguments(self):
default_arguments = {
'cors_headers': mock.sentinel.cors_headers,
}
default_get_arguments = {
'accept': mock.sentinel.accept,
}
default_collection_arguments = {
'cors_origins': mock.sentinel.cors_origins,
}
collection_get_arguments = {
'error_handler': mock.sentinel.error_handler
}
viewset = ViewSet(
default_arguments=default_arguments,
default_get_arguments=default_get_arguments,
default_collection_arguments=default_collection_arguments,
collection_get_arguments=collection_get_arguments)
arguments = viewset.collection_arguments(mock.MagicMock, 'get')
arguments.pop('schema')
self.assertDictEqual(
arguments, {
'accept': mock.sentinel.accept,
'cors_headers': mock.sentinel.cors_headers,
'cors_origins': mock.sentinel.cors_origins,
'error_handler': mock.sentinel.error_handler,
})
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(
validate_schema_for=('GET', )
)
resource = mock.MagicMock()
arguments = viewset.collection_arguments(resource, 'get')
self.assertIn('schema', arguments)
def test_default_arguments_are_copied_before_being_returned(self):
original_arguments = {}
viewset = ViewSet(collection_get_arguments=original_arguments)
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), "GET")
self.assertEqual(original_arguments, {})
self.assertNotEquals(original_arguments, arguments)
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(
validate_schema_for=('GET', )
)
resource = mock.MagicMock(mapping=colander.SchemaNode(colander.Int()))
arguments = viewset.collection_arguments(resource, 'get')
schema = arguments['schema']
self.assertEquals(schema.children[0], resource.mapping)
def test_a_default_schema_is_added_when_method_doesnt_match(self, mocked):
viewset = ViewSet()
resource = mock.MagicMock()
mocked.Mapping.return_value = mock.sentinel.default_schema
arguments = viewset.collection_arguments(resource, 'GET')
self.assertNotEqual(arguments['schema'], resource.schema)
mocked.assert_called_with()
def test_default_arguments_are_copied_before_being_returned(self):
original_arguments = {}
viewset = ViewSet(
collection_get_arguments=original_arguments,
validate_schema_for=()
)
arguments = viewset.collection_arguments(mock.sentinel.resource, 'GET')
self.assertEquals(original_arguments, {})
self.assertNotEquals(original_arguments, arguments)
def test_a_default_schema_is_added_when_method_doesnt_match(self, mocked):
viewset = ViewSet()
resource = mock.MagicMock()
viewset.responses = mock.MagicMock()
mocked.Mapping.return_value = mock.sentinel.default_schema
arguments = viewset.collection_arguments(resource, "GET")
self.assertNotEqual(arguments["schema"], resource.schema)
mocked.assert_called_with()
def test_a_default_schema_is_added_when_method_doesnt_match(self, mocked):
viewset = ViewSet(validate_schema_for=('GET', ))
resource = mock.MagicMock()
mocked.MappingSchema.return_value = mock.sentinel.default_schema
arguments = viewset.collection_arguments(resource, 'POST')
self.assertEquals(arguments['schema'], mock.sentinel.default_schema)
self.assertNotEqual(arguments['schema'], resource.mapping)
mocked.MappingSchema.assert_called_with(unknown='preserve')
def test_a_default_schema_is_added_when_method_doesnt_match(self, mocked):
viewset = ViewSet(
validate_schema_for=('GET', )
)
resource = mock.MagicMock()
mocked.MappingSchema.return_value = mock.sentinel.default_schema
arguments = viewset.collection_arguments(resource, 'POST')
self.assertEquals(arguments['schema'], mock.sentinel.default_schema)
self.assertNotEqual(arguments['schema'], resource.mapping)
mocked.MappingSchema.assert_called_with(unknown='preserve')
def test_permission_private_is_set_by_default(self):
viewset = ViewSet()
args = viewset.collection_arguments(mock.sentinel.resource, 'GET')
self.assertEquals(args['permission'], 'private')
def test_schema_is_added_when_method_matches(self):
viewset = ViewSet()
viewset.responses = mock.MagicMock()
resource = mock.MagicMock()
arguments = viewset.collection_arguments(resource, "GET")
self.assertIn("schema", arguments)
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(validate_schema_for=('GET', ))
resource = mock.MagicMock(mapping=colander.SchemaNode(colander.Int()))
arguments = viewset.collection_arguments(resource, 'get')
schema = arguments['schema']
self.assertEquals(schema.children[0], resource.mapping)
def test_permission_private_is_set_by_default(self):
viewset = ViewSet()
args = viewset.collection_arguments(mock.sentinel.resource, 'GET')
self.assertEquals(args['permission'], 'private')
def test_permission_private_is_set_by_default(self):
viewset = ViewSet()
viewset.responses = mock.MagicMock()
args = viewset.collection_arguments(mock.MagicMock(), "GET")
self.assertEqual(args["permission"], "private")
def test_schema_is_added_when_method_matches(self):
viewset = ViewSet()
viewset.responses = mock.MagicMock()
resource = mock.MagicMock()
arguments = viewset.collection_arguments(resource, 'GET')
self.assertIn('schema', arguments)
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(collection_methods=("GET", "DELETE"))
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), "get")
self.assertIn("schema", arguments)
def test_permission_private_is_set_by_default(self):
viewset = ViewSet()
viewset.responses = mock.MagicMock()
args = viewset.collection_arguments(mock.MagicMock(), 'GET')
self.assertEqual(args['permission'], 'private')
def test_schema_is_added_when_uppercase_method_matches(self):
viewset = ViewSet(collection_methods=("GET", "DELETE"))
viewset.responses = mock.MagicMock()
arguments = viewset.collection_arguments(mock.MagicMock(), "get")
self.assertIn("schema", arguments)
