def test_can_migrate(self): self.create_table() self.schema.append({'name': 'new_field', 'type': 'TEXT'}) migrated = schema.migrate(self.tablename, self.schema) _schema = schema.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 = schema.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'} sql.add_field(TEST_PGA_PATH, self.tablename, field) _schema = schema.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(schema.get_schema(self.tablename)) self.assertFalse(schema.get_schema('notatable'))