def setUp(self): super(PostponedUpdateAccountTaskTests, self).setUp() create_test_map() self.account = self.accounts_factory.create_account() self.postponed_task = UpdateAccount(account_id=self.account.id, method=AccountPrototype.prolong_premium, data={'days': 17})
def setUp(self): super(PostponedUpdateAccountTaskTests, self).setUp() create_test_map() result, account_id, bundle_id = register_user('test_user', '*****@*****.**', '111111') self.account = AccountPrototype.get_by_id(account_id) self.postponed_task = UpdateAccount( account_id=self.account.id, method=AccountPrototype.prolong_premium, data={'days': 17})
class PostponedUpdateAccountTaskTests(testcase.TestCase): def setUp(self): super(PostponedUpdateAccountTaskTests, self).setUp() create_test_map() self.account = self.accounts_factory.create_account() self.postponed_task = UpdateAccount(account_id=self.account.id, method=AccountPrototype.prolong_premium, data={'days': 17}) def test_create(self): self.assertEqual(self.postponed_task.account_id, self.account.id) self.assertEqual(self.postponed_task.method, 'prolong_premium') self.assertEqual(self.postponed_task.data, {'days': 17}) self.assertTrue(self.postponed_task.state.is_UNPROCESSED) def test_serialization(self): self.assertEqual(self.postponed_task.serialize(), UpdateAccount.deserialize(self.postponed_task.serialize()).serialize()) def test_processed__success(self): self.assertFalse(self.account.is_premium) self.postponed_task.process(main_task=mock.Mock()) self.account.reload() self.assertTrue(self.account.is_premium) def test_processed__wrong_state(self): for state in UPDATE_ACCOUNT_STATE.records: if state.is_UNPROCESSED: continue self.postponed_task.state = state self.postponed_task.process(main_task=mock.Mock()) self.account.reload() self.assertFalse(self.account.is_premium)
def test_serialization(self): self.assertEqual( self.postponed_task.serialize(), UpdateAccount.deserialize( self.postponed_task.serialize()).serialize())
def test_serialization(self): self.assertEqual(self.postponed_task.serialize(), UpdateAccount.deserialize(self.postponed_task.serialize()).serialize())