def test_empty_signature(self):
operation = custom_migration_operations.operations.TestOperation()
buff, imports = OperationWriter(operation, indentation=0).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)
buff, imports = OperationWriter(operation, indentation=0).serialize()
self.assertEqual(imports,
{'import custom_migration_operations.operations'})
self.assertEqual(
buff, 'custom_migration_operations.operations.KwargsOperation(\n'
' kwarg1=1,\n'
'),')
def test_multiline_args_signature(self):
operation = custom_migration_operations.operations.ArgsOperation(
"test\n arg1", "test\narg2")
buff, imports = OperationWriter(operation, indentation=0).serialize()
self.assertEqual(imports,
{'import custom_migration_operations.operations'})
self.assertEqual(
buff, "custom_migration_operations.operations.ArgsOperation(\n"
" arg1='test\\n arg1',\n"
" arg2='test\\narg2',\n"
"),")
def test_expand_args_signature(self):
operation = custom_migration_operations.operations.ExpandArgsOperation(
[1, 2])
buff, imports = OperationWriter(operation, indentation=0).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_nested_args_signature(self):
operation = custom_migration_operations.operations.ArgsOperation(
custom_migration_operations.operations.ArgsOperation(1, 2),
custom_migration_operations.operations.KwargsOperation(kwarg1=3,
kwarg2=4))
buff, imports = OperationWriter(operation, indentation=0).serialize()
self.assertEqual(imports,
{'import custom_migration_operations.operations'})
self.assertEqual(
buff, 'custom_migration_operations.operations.ArgsOperation(\n'
' arg1=custom_migration_operations.operations.ArgsOperation(\n'
' arg1=1,\n'
' arg2=2,\n'
' ),\n'
' arg2=custom_migration_operations.operations.KwargsOperation(\n'
' kwarg1=3,\n'
' kwarg2=4,\n'
' ),\n'
'),')
def test_nested_operation_expand_args_signature(self):
operation = custom_migration_operations.operations.ExpandArgsOperation(
arg=[
custom_migration_operations.operations.KwargsOperation(
kwarg1=1,
kwarg2=2,
),
])
buff, imports = OperationWriter(operation, indentation=0).serialize()
self.assertEqual(imports,
{'import custom_migration_operations.operations'})
self.assertEqual(
buff,
'custom_migration_operations.operations.ExpandArgsOperation(\n'
' arg=[\n'
' custom_migration_operations.operations.KwargsOperation(\n'
' kwarg1=1,\n'
' kwarg2=2,\n'
' ),\n'
' ],\n'
'),')
def serialize(self):
from djmodels.db.migrations.writer import OperationWriter
string, imports = OperationWriter(self.value,
indentation=0).serialize()
# Nested operation, trailing comma is handled in upper OperationWriter._write()
return string.rstrip(','), imports