def test_can_migrate(self):
self.create_table()
self.schema.append({'name': 'new_field', 'type': 'TEXT'})
migrated = pga.migrate(self.tablename, self.schema)
schema = pga.get_schema(self.tablename)
self.assertEqual(schema[2]['name'], 'new_field')
self.assertEqual(migrated, ['new_field'])
def test_get_schema(self):
self.create_table()
schema = pga.get_schema(self.tablename)
self.assertEqual(schema[0]['name'], 'id')
self.assertEqual(schema[0]['type'], 'INTEGER')
self.assertEqual(schema[1]['name'], 'name')
self.assertEqual(schema[1]['type'], 'TEXT')
def test_add_field(self):
self.create_table()
field = {'name': 'counter', 'type': 'INTEGER'}
pga.add_field(self.tablename, field)
schema = pga.get_schema(self.tablename)
self.assertEqual(schema[2]['name'], 'counter')
self.assertEqual(schema[2]['type'], 'INTEGER')
def test_add_field(self):
self.create_table()
field = {
'name': 'counter',
'type': 'INTEGER'
}
sql.add_field(TEST_PGA_PATH, self.tablename, field)
schema = pga.get_schema(self.tablename)
self.assertEqual(schema[2]['name'], 'counter')
self.assertEqual(schema[2]['type'], 'INTEGER')
def test_can_know_if_table_exists(self):
self.create_table()
self.assertTrue(pga.get_schema(self.tablename))
self.assertFalse(pga.get_schema('notatable'))
