def test_send_circle(self, mock_smtplib): gift_circle = GiftCircle("test_unit_3.txt") gift_circle.parse_data() gift_circle.shuffle_data() smtpserver = mock_smtplib.return_value smtpserver.sendmail.return_value={} gift_circle.send_circle() self.assertTrue(smtpserver.sendmail.called)
def test_send_circle(self, mock_twilio): gift_circle = GiftCircle("test_unit_6.txt") gift_circle.parse_data() gift_circle.shuffle_data() import random class message(object): sid = "SM"+"".join(random.choice("abcdef0123456789") for i in range(32)) twilio = mock_twilio.return_value twilio.sms.messages.create.return_value=message() gift_circle.send_circle() self.assertTrue(twilio.sms.messages.create.called)
def test_cannotsend_without_shuffle(self): gift_circle = GiftCircle("test_unit_3.txt") gift_circle.parse_data() with self.assertRaisesRegexp(AttributeError, "'GiftCircle' object has no attribute 'shuffled'") as ex: gift_circle.send_circle("This is a test message")