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")
