def test_add_field_creates_column(self, model_schema):
field_schema = FieldSchema(name="special",
data_type="integer",
model_schema=model_schema)
table_name = model_schema.db_table
column_name = field_schema.db_column
assert not utils.db_table_has_field(table_name, column_name)
field_schema.save()
assert utils.db_table_has_field(table_name, column_name)
def test_add_field_creates_column(self, model_schema):
field_schema = FieldSchema(name="special",
class_name="django.db.models.IntegerField",
model_schema=model_schema)
table_name = model_schema.db_table
column_name = field_schema.db_column
assert not utils.db_table_has_field(table_name, column_name)
field_schema.save()
assert utils.db_table_has_field(table_name, column_name)
def test_alter_column(self, initial_model, changed_field_name_model):
initial_field = initial_model._meta.get_field("integer")
new_field = changed_field_name_model._meta.get_field("changed")
assert db_table_has_field("tests_initialmodel", "integer")
assert not db_table_has_field("tests_initialmodel", "changed")
FieldSchemaEditor(initial_field).alter_column(changed_field_name_model,
new_field)
assert db_table_has_field("tests_initialmodel", "changed")
assert not db_table_has_field("tests_initialmodel", "integer")
def test_alter_column(self, initial_model, changed_field_name_model):
initial_field = initial_model._meta.get_field('integer')
new_field = changed_field_name_model._meta.get_field('changed')
assert db_table_has_field('tests_initialmodel', 'integer')
assert not db_table_has_field('tests_initialmodel', 'changed')
FieldSchemaEditor(initial_field).alter_column(changed_field_name_model,
new_field)
assert db_table_has_field('tests_initialmodel', 'changed')
assert not db_table_has_field('tests_initialmodel', 'integer')
def test_update_column_creates_if_not_exists(self, initial_model):
initial_field = initial_model._meta.get_field("integer")
assert not db_table_has_field("tests_initialmodel", "integer")
FieldSchemaEditor().update_column(initial_model, initial_field)
assert db_table_has_field("tests_initialmodel", "integer")
def test_add_column(self, initial_model):
assert not db_table_has_field("tests_initialmodel", "integer")
new_field = initial_model._meta.get_field("integer")
FieldSchemaEditor().add_column(initial_model, new_field)
assert db_table_has_field("tests_initialmodel", "integer")
def test_drop_column(self, initial_model):
field = initial_model._meta.get_field('integer')
assert db_table_has_field('tests_initialmodel', 'integer')
FieldSchemaEditor().drop_column(initial_model, field)
assert not db_table_has_field('tests_initialmodel', 'integer')
def test_add_field_creates_column(self, model_schema, field_schema):
table_name = model_schema.db_table
column_name = field_schema.db_column
assert not utils.db_table_has_field(table_name, column_name)
model_schema.add_field(field_schema)
assert utils.db_table_has_field(table_name, column_name)
def test_remove_field_drops_column(self, model_schema, field_schema):
table_name = model_schema.db_table
column_name = field_schema.db_column
assert utils.db_table_has_field(table_name, column_name)
model_schema.remove_field(field_schema)
assert not utils.db_table_has_field(table_name, column_name)
def test_deleting_field_drops_column(self, model_schema, field_schema):
table_name = model_schema.db_table
column_name = field_schema.db_column
assert utils.db_table_has_field(table_name, column_name)
field_schema.delete()
assert not utils.db_table_has_field(table_name, column_name)
