def test_enqueue_messages_with_messages_mocker (self): dummy_AccountGroup = self.mocker.replace('singularmsd.AccountGroup') dummy_AccountGroup.account.id self.mocker.result (1) self.mocker.count (6) dummy_AccountGroup(ANY) self.mocker.result (dummy_AccountGroup) self.mocker.count (6) dummy_AccountGroup.start () self.mocker.result (True) self.mocker.count (6) dummy_AccountGroup.event.set () self.mocker.result (True) self.mocker.count (20) self.mocker.replay () ms = MessageScheduler ({}) ms.create_account_groups() processor = SMSProcessor () for i in range (0, 10): processor.sendSMS (WebServiceTests.username, WebServiceTests.password, 'AccessWithPurchase', 'foo', 'bar', datetime.datetime.now()) for i in range (0, 10): processor.sendSMSToChannel (WebServiceTests.username, WebServiceTests.password, 'AccessWithPurchase', 'TestChannel', 'bar', datetime.datetime.now()) ms.enqueue_messages ()
def test_create_account_groups_without_accounts (self): dummy_Account = self.mocker.replace('accounting.models.Account') dummy_Account.objects.all () self.mocker.result ([]) self.mocker.count (2) self.mocker.replay () ms = MessageScheduler ({}) ms.create_account_groups()
def test_create_account_groups_with_accounts (self): dummy_AccountGroup = self.mocker.replace('singularmsd.AccountGroup') dummy_AccountGroup(ANY) self.mocker.result (dummy_AccountGroup) self.mocker.count (3) dummy_AccountGroup.account.id self.mocker.result (1) self.mocker.count (3) dummy_AccountGroup.start () self.mocker.result (True) self.mocker.count (3) self.mocker.replay () ms = MessageScheduler ({}) ms.create_account_groups()