def serialize_db_to_string(self): from boardinghouse.schema import activate_template_schema, deactivate_schema activate_template_schema() result = super(DatabaseCreation, self).serialize_db_to_string() deactivate_schema() return result
def inner(self, *args, **kwargs): activate_template_schema() test(self, *args, schema='__template__', **kwargs) for schema in Schema.objects.all(): schema.activate() test(self, *args, schema=schema.schema, **kwargs) deactivate_schema()
def test_schema_table_is_in_public(self): deactivate_schema() cursor = connection.cursor() table_name = Schema._meta.db_table cursor.execute(TABLE_QUERY, ['public', table_name]) data = cursor.fetchone() self.assertEqual((table_name,), data) cursor.close()
def test_schema_table_is_in_public(self): deactivate_schema() cursor = connection.cursor() table_name = Schema._meta.db_table cursor.execute(TABLE_QUERY, ["public", table_name]) data = cursor.fetchone() self.assertEqual((table_name,), data) cursor.close()
def test_activate_schema_function(self): self.assertRaises(TemplateSchemaActivation, activate_schema, '__template__') Schema.objects.mass_create('a', 'b') activate_schema('a') self.assertEqual('a', get_active_schema_name()) activate_schema('b') self.assertEqual('b', get_active_schema_name()) deactivate_schema() self.assertEqual(None, get_active_schema_name())
def test_activate_schema_function(self): self.assertRaises(TemplateSchemaActivation, activate_schema, "__template__") Schema.objects.mass_create("a", "b") activate_schema("a") self.assertEqual("a", get_active_schema_name()) activate_schema("b") self.assertEqual("b", get_active_schema_name()) deactivate_schema() self.assertEqual(None, get_active_schema_name())
def test_deserialize_from_string_activates_template_schema(self): with patch('boardinghouse.schema.activate_template_schema') as activate_template_schema: deactivate_schema() connection.creation.deserialize_db_from_string('[]') activate_template_schema.assert_called_once_with()
def deactivate(self): deactivate_schema()
def test_deserialize_from_string_activates_template_schema(self): with patch('boardinghouse.schema.activate_template_schema' ) as activate_template_schema: deactivate_schema() connection.creation.deserialize_db_from_string('[]') activate_template_schema.assert_called_once_with()
def deserialize_db_from_string(self, data): from boardinghouse.schema import activate_template_schema, deactivate_schema activate_template_schema() super(DatabaseCreation, self).deserialize_db_from_string(data) deactivate_schema()