def test_enum_property(self):
model = Person()
model.gender = Gender.male
self.assertEquals(model.gender, Gender.male)
with self.assertRaises(ValueError):
a = Person()
a.gender = "invalid valid"
def test_put(self):
p = Person("philip", "folk", 20)
self.assertIsNone(p.uuid)
p = storage.put(p)
self.assertIsNotNone(p.uuid)
p.name = "petter"
new_p = storage.put(p)
self.assertEqual(new_p.uuid, p.uuid)
def test_property(self):
with self.assertRaises(ValueError):
obj = Person()
obj.name = 3
with self.assertRaises(ValueError):
obj = Person()
obj.age = "a"
obj = Person()
obj.name = "string"
obj.age = 1
self.assertEqual(obj.name, "string")
self.assertEqual(obj.age, 1)
def test_different_object(self):
obj1 = Person()
obj1.name = "string"
obj1.age = 1
self.assertEqual(obj1.name, "string")
self.assertEqual(obj1.age, 1)
obj2 = Person()
obj2.name = "new"
obj2.age = 2
self.assertEqual(obj2.name, "new")
self.assertEqual(obj2.age, 2)
self.assertEqual(obj1.name, "string")
self.assertEqual(obj1.age, 1)
def test_to_json(self):
fake = Person(name='testa', age=10)
self.assertEqual({'active': None, 'age': 10, 'gender': None, 'lastname': None, 'name': 'testa', 'uuid': None, 'weight': 23}, fake.to_json())
def test_columns(self):
self.assertEqual(['active', 'age', 'gender', 'lastname', 'name', 'uuid', 'weight'], Person.columns())
p = Person()
self.assertEqual(['active', 'age', 'gender', 'lastname', 'name', 'uuid', 'weight'], p.columns())
def test_create_table(self):
types = [PostgreSQLTypes.char, PostgreSQLTypes.boolean,
PostgreSQLTypes.real, PostgreSQLTypes.int]
sql = "CREATE TABLE tb_Person (uuid {0}(36) PRIMARY KEY, \
active {1}, age {3}, gender {0}, lastname {0}(140), name {0}(512), weight {2});".format(*types)
self.assertEqual(sql, Person.create_table_sql())
def test_to_instance(self):
model_json = {'name': 'john', 'age': 53}
instance = Person.to_instance(model_json)
self.assertTrue(isinstance(instance, Person))
self.assertEqual(instance.name, 'john')
self.assertEqual(instance.age, 53)
