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)