def test_empty_signature(self): operation = custom_migration_operations.operations.TestOperation() writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.TestOperation(\n' '),')
def test_empty_signature(self): operation = custom_migration_operations.operations.TestOperation() writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.TestOperation(\n' '),' )
def test_kwargs_signature(self): operation = custom_migration_operations.operations.KwargsOperation(kwarg1=1) writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.KwargsOperation(\n' ' kwarg1=1,\n' '),' )
def test_kwargs_signature(self): operation = custom_migration_operations.operations.KwargsOperation(kwarg1=1) writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.KwargsOperation(\n' ' kwarg1=1,\n' '),' )
def test_expand_args_signature(self): operation = custom_migration_operations.operations.ExpandArgsOperation([1, 2]) writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.ExpandArgsOperation(\n' ' arg=[\n' ' 1,\n' ' 2,\n' ' ],\n' '),' )
def test_expand_args_signature(self): operation = custom_migration_operations.operations.ExpandArgsOperation([1, 2]) writer = OperationWriter(operation) writer.indentation = 0 buff, imports = writer.serialize() self.assertEqual(imports, {'import custom_migration_operations.operations'}) self.assertEqual( buff, 'custom_migration_operations.operations.ExpandArgsOperation(\n' ' arg=[\n' ' 1,\n' ' 2,\n' ' ],\n' '),' )