def test_register_for_deletion_one_arg(self):
"""
Tests the register_for_deletion_function with one argument.
"""
initial_data = BaseInitialData()
account = G(Account)
initial_data.register_for_deletion(account)
self.assertEquals(initial_data.get_model_objs_registered_for_deletion(), [account])
def test_register_for_deletion_multiple_args(self):
"""
Tests the register_for_deletion_function with multiple arguments.
"""
initial_data = BaseInitialData()
account1 = G(Account)
account2 = G(Account)
initial_data.register_for_deletion(account1, account2)
self.assertEquals(initial_data.get_model_objs_registered_for_deletion(), [account1, account2])
def test_register_for_deletion_one_arg(self):
"""
Tests the register_for_deletion_function with one argument.
"""
initial_data = BaseInitialData()
account = G(Account)
initial_data.register_for_deletion(account)
self.assertEquals(
initial_data.get_model_objs_registered_for_deletion(), [account])
def test_register_for_deletion_multiple_args(self):
"""
Tests the register_for_deletion_function with multiple arguments.
"""
initial_data = BaseInitialData()
account1 = G(Account)
account2 = G(Account)
initial_data.register_for_deletion(account1, account2)
self.assertEquals(
initial_data.get_model_objs_registered_for_deletion(),
[account1, account2])
