diff options
Diffstat (limited to 'tests/test_core.py')
| -rw-r--r-- | tests/test_core.py | 225 |
1 files changed, 107 insertions, 118 deletions
diff --git a/tests/test_core.py b/tests/test_core.py index 51c0426..c363950 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -24,48 +24,38 @@ import warlock fixture = { - 'name': 'Country', - 'properties': { - 'name': {'type': 'string'}, - 'population': {'type': 'integer'}, - }, - 'additionalProperties': False, + "name": "Country", + "properties": {"name": {"type": "string"}, "population": {"type": "integer"}}, + "additionalProperties": False, } complex_fixture = { - 'name': 'Mixmaster', - 'properties': { - 'sub': {'type': 'object', - 'properties': {'foo': {'type': 'string'}}} + "name": "Mixmaster", + "properties": { + "sub": {"type": "object", "properties": {"foo": {"type": "string"}}} }, } parent_fixture = { - 'name': 'Parent', - 'properties': { - 'name': {'type': 'string'}, - 'children': {'type': 'array', 'items': [{'type': 'object'}]} + "name": "Parent", + "properties": { + "name": {"type": "string"}, + "children": {"type": "array", "items": [{"type": "object"}]}, }, - 'required': ['name', 'children'] + "required": ["name", "children"], } child_fixture = { - 'name': 'Child', - 'properties': { - 'age': {'type': 'integer'}, - 'mother': {'type': 'object'} - }, - 'required': ['age', 'mother'] + "name": "Child", + "properties": {"age": {"type": "integer"}, "mother": {"type": "object"}}, + "required": ["age", "mother"], } nameless_fixture = { - 'properties': { - 'name': {'type': 'string'}, - 'population': {'type': 'integer'}, - }, - 'additionalProperties': False, + "properties": {"name": {"type": "string"}, "population": {"type": "integer"}}, + "additionalProperties": False, } @@ -76,98 +66,101 @@ class TestCore(unittest.TestCase): def test_class_name_from_unicode_schema_name(self): fixture_copy = copy.deepcopy(fixture) - fixture_copy['name'] = six.text_type(fixture_copy['name']) + fixture_copy["name"] = six.text_type(fixture_copy["name"]) # Can't set class.__name__ to a unicode object, ensure warlock # does some magic to make it possible warlock.model_factory(fixture_copy) def test_invalid_operations(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) # Ensure a valid object was created - self.assertEqual(sweden.name, 'Sweden') + self.assertEqual(sweden.name, "Sweden") self.assertEqual(sweden.population, 9379116) # Specific exceptions should be raised for invalid operations - self.assertRaises(AttributeError, getattr, sweden, 'overlord') + self.assertRaises(AttributeError, getattr, sweden, "overlord") exc = warlock.InvalidOperation - self.assertRaises(exc, setattr, sweden, 'overlord', 'Bears') - self.assertRaises(exc, setattr, sweden, 'name', 5) - self.assertRaises(exc, setattr, sweden, 'population', 'N/A') + self.assertRaises(exc, setattr, sweden, "overlord", "Bears") + self.assertRaises(exc, setattr, sweden, "name", 5) + self.assertRaises(exc, setattr, sweden, "population", "N/A") def test_no_mask_arbitrary_properties(self): fixture_copy = copy.deepcopy(fixture) - fixture_copy['additionalProperties'] = {'type': 'string'} + fixture_copy["additionalProperties"] = {"type": "string"} Country = warlock.model_factory(fixture_copy) # We should still depend on the schema for validation self.assertRaises(ValueError, Country, GDP=56956) # But arbitrary properties should be allowed if they check out - sweden = Country(overlord='Waldon') - sweden.abbreviation = 'SE' + sweden = Country(overlord="Waldon") + sweden.abbreviation = "SE" exc = warlock.InvalidOperation - self.assertRaises(exc, setattr, sweden, 'abbreviation', 0) + self.assertRaises(exc, setattr, sweden, "abbreviation", 0) def test_items(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) - self.assertEqual(set(list(six.iteritems(sweden))), - set([('name', 'Sweden'), ('population', 9379116)])) - self.assertEqual(set(sweden.items()), - set([('name', 'Sweden'), ('population', 9379116)])) + sweden = Country(name="Sweden", population=9379116) + self.assertEqual( + set(list(six.iteritems(sweden))), + set([("name", "Sweden"), ("population", 9379116)]), + ) + self.assertEqual( + set(sweden.items()), set([("name", "Sweden"), ("population", 9379116)]) + ) def test_update(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) exc = warlock.InvalidOperation - self.assertRaises(exc, sweden.update, {'population': 'N/A'}) - self.assertRaises(exc, sweden.update, {'overloard': 'Bears'}) + self.assertRaises(exc, sweden.update, {"population": "N/A"}) + self.assertRaises(exc, sweden.update, {"overloard": "Bears"}) def test_naming(self): Country = warlock.model_factory(fixture) - self.assertEqual('Country', Country.__name__) + self.assertEqual("Country", Country.__name__) - Country2 = warlock.model_factory(fixture, name='Country2') - self.assertEqual('Country2', Country2.__name__) + Country2 = warlock.model_factory(fixture, name="Country2") + self.assertEqual("Country2", Country2.__name__) nameless = warlock.model_factory(nameless_fixture) - self.assertEqual('Model', nameless.__name__) + self.assertEqual("Model", nameless.__name__) - nameless2 = warlock.model_factory(nameless_fixture, name='Country3') - self.assertEqual('Country3', nameless2.__name__) + nameless2 = warlock.model_factory(nameless_fixture, name="Country3") + self.assertEqual("Country3", nameless2.__name__) def test_deepcopy(self): """Make sure we aren't leaking references.""" Mixmaster = warlock.model_factory(complex_fixture) - mike = Mixmaster(sub={'foo': 'mike'}) + mike = Mixmaster(sub={"foo": "mike"}) - self.assertEquals('mike', mike.sub['foo']) + self.assertEquals("mike", mike.sub["foo"]) mike_1 = mike.copy() - mike_1['sub']['foo'] = 'james' - self.assertEquals('mike', mike.sub['foo']) + mike_1["sub"]["foo"] = "james" + self.assertEquals("mike", mike.sub["foo"]) mike_2 = dict(six.iteritems(mike)) - mike_2['sub']['foo'] = 'james' - self.assertEquals('mike', mike.sub['foo']) + mike_2["sub"]["foo"] = "james" + self.assertEquals("mike", mike.sub["foo"]) mike_2 = dict(mike.items()) - mike_2['sub']['foo'] = 'james' - self.assertEquals('mike', mike.sub['foo']) + mike_2["sub"]["foo"] = "james" + self.assertEquals("mike", mike.sub["foo"]) mike_3_sub = list(six.itervalues(mike))[0] - mike_3_sub['foo'] = 'james' - self.assertEquals('mike', mike.sub['foo']) + mike_3_sub["foo"] = "james" + self.assertEquals("mike", mike.sub["foo"]) mike_3_sub = list(mike.values())[0] - mike_3_sub['foo'] = 'james' - self.assertEquals('mike', mike.sub['foo']) + mike_3_sub["foo"] = "james" + self.assertEquals("mike", mike.sub["foo"]) def test_forbidden_methods(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) exc = warlock.InvalidOperation self.assertRaises(exc, sweden.clear) self.assertRaises(exc, sweden.pop, 0) @@ -175,97 +168,97 @@ class TestCore(unittest.TestCase): def test_dict_syntax(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) - sweden['name'] = 'Finland' - self.assertEqual('Finland', sweden['name']) + sweden["name"] = "Finland" + self.assertEqual("Finland", sweden["name"]) - del sweden['name'] - self.assertRaises(AttributeError, getattr, sweden, 'name') + del sweden["name"] + self.assertRaises(AttributeError, getattr, sweden, "name") def test_attr_syntax(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) - sweden.name = 'Finland' - self.assertEqual('Finland', sweden.name) + sweden.name = "Finland" + self.assertEqual("Finland", sweden.name) - delattr(sweden, 'name') - self.assertRaises(AttributeError, getattr, sweden, 'name') + delattr(sweden, "name") + self.assertRaises(AttributeError, getattr, sweden, "name") def test_changes(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) self.assertEqual(sweden.changes, {}) - sweden['name'] = 'Finland' - self.assertEqual(sweden.changes, {'name': 'Finland'}) - sweden['name'] = 'Norway' - self.assertEqual(sweden.changes, {'name': 'Norway'}) + sweden["name"] = "Finland" + self.assertEqual(sweden.changes, {"name": "Finland"}) + sweden["name"] = "Norway" + self.assertEqual(sweden.changes, {"name": "Norway"}) def test_patch_no_changes(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) - self.assertEqual(sweden.patch, '[]') + sweden = Country(name="Sweden", population=9379116) + self.assertEqual(sweden.patch, "[]") def test_patch_alter_value(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) - sweden['name'] = 'Finland' + sweden = Country(name="Sweden", population=9379116) + sweden["name"] = "Finland" self.assertEqual( json.loads(sweden.patch), - json.loads( - '[{"path": "/name", "value": "Finland", "op": "replace"}]')) + json.loads('[{"path": "/name", "value": "Finland", "op": "replace"}]'), + ) def test_patch_drop_attribute(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) - del sweden['name'] + sweden = Country(name="Sweden", population=9379116) + del sweden["name"] self.assertEqual( - json.loads(sweden.patch), - json.loads('[{"path": "/name", "op": "remove"}]')) + json.loads(sweden.patch), json.loads('[{"path": "/name", "op": "remove"}]') + ) def test_patch_reduce_operations(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) - sweden['name'] = 'Finland' + sweden["name"] = "Finland" self.assertEqual( json.loads(sweden.patch), - json.loads( - '[{"path": "/name", "value": "Finland", "op": "replace"}]')) + json.loads('[{"path": "/name", "value": "Finland", "op": "replace"}]'), + ) - sweden['name'] = 'Norway' + sweden["name"] = "Norway" self.assertEqual( json.loads(sweden.patch), - json.loads( - '[{"path": "/name", "value": "Norway", "op": "replace"}]')) + json.loads('[{"path": "/name", "value": "Norway", "op": "replace"}]'), + ) def test_patch_multiple_operations(self): Country = warlock.model_factory(fixture) - sweden = Country(name='Sweden', population=9379116) + sweden = Country(name="Sweden", population=9379116) - sweden['name'] = 'Finland' - sweden['population'] = 5387000 + sweden["name"] = "Finland" + sweden["population"] = 5387000 self.assertEqual(len(json.loads(sweden.patch)), 2) patches = json.loads( '[{"path": "/name", "value": "Finland", "op": "replace"}, ' - '{"path": "/population", "value": 5387000, "op": "replace"}]') + '{"path": "/population", "value": 5387000, "op": "replace"}]' + ) for patch in json.loads(sweden.patch): self.assertTrue(patch in patches) def test_resolver(self): from jsonschema import RefResolver + dirname = os.path.dirname(__file__) - schemas_path = 'file://' + os.path.join(dirname, 'schemas/') + schemas_path = "file://" + os.path.join(dirname, "schemas/") resolver = RefResolver(schemas_path, None) - country_schema_file = \ - open(os.path.join(dirname, 'schemas/') + 'country.json') - person_schema_file = \ - open(os.path.join(dirname, 'schemas/') + 'person.json') + country_schema_file = open(os.path.join(dirname, "schemas/") + "country.json") + person_schema_file = open(os.path.join(dirname, "schemas/") + "person.json") country_schema = json.load(country_schema_file) person_schema = json.load(person_schema_file) @@ -275,20 +268,16 @@ class TestCore(unittest.TestCase): england = Country( name="England", population=53865800, - overlord=Person( - title="Queen", - firstname="Elizabeth", - lastname="Windsor" - ) + overlord=Person(title="Queen", firstname="Elizabeth", lastname="Windsor"), ) expected = { - 'name': 'England', - 'population': 53865800, - 'overlord': { - 'title': 'Queen', - 'lastname': 'Windsor', - 'firstname': 'Elizabeth' - } + "name": "England", + "population": 53865800, + "overlord": { + "title": "Queen", + "lastname": "Windsor", + "firstname": "Elizabeth", + }, } self.assertEqual(england, expected) @@ -296,7 +285,7 @@ class TestCore(unittest.TestCase): Parent = warlock.model_factory(parent_fixture) Child = warlock.model_factory(child_fixture) - mom = Parent(name='Abby', children=[]) + mom = Parent(name="Abby", children=[]) teenager = Child(age=15, mother=mom) toddler = Child(age=3, mother=mom) |