def setUp(self):
# Set up Google App Engine testbed
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.testbed.init_app_identity_stub()
self.contactNumber = "+16135551234"
# Set up some handy constants
self._APP_ID = app_identity.get_application_id()
self.MAIL_SUFFIX = "@" + self._APP_ID + ".appspotmail.com"
self.XMPP_SUFFIX = "@" + self._APP_ID + ".appspotchat.com"
self.ownerPhoneNumber = "+16135554444"
self.ownerJid = "*****@*****.**"
self.ownerEmailAddress = "*****@*****.**"
self.owner = Owner(self.ownerPhoneNumber, self.ownerJid, self.ownerEmailAddress)
self.xmppvoicemail = XmppVoiceMail(self.owner)
self.communications = self.xmppvoicemail._communications = CommunicationsFixture()
# Subscribe the default sender.
defaultSender = Contact.getDefaultSender()
defaultSender.subscribed = True
Contact.update(defaultSender)
class XmppVoiceMailTestCases(unittest.TestCase):
def setUp(self):
# Set up Google App Engine testbed
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.testbed.init_app_identity_stub()
self.contactNumber = "+16135551234"
# Set up some handy constants
self._APP_ID = app_identity.get_application_id()
self.MAIL_SUFFIX = "@" + self._APP_ID + ".appspotmail.com"
self.XMPP_SUFFIX = "@" + self._APP_ID + ".appspotchat.com"
self.ownerPhoneNumber = "+16135554444"
self.ownerJid = "*****@*****.**"
self.ownerEmailAddress = "*****@*****.**"
self.owner = Owner(self.ownerPhoneNumber, self.ownerJid, self.ownerEmailAddress)
self.xmppvoicemail = XmppVoiceMail(self.owner)
self.communications = self.xmppvoicemail._communications = CommunicationsFixture()
# Subscribe the default sender.
defaultSender = Contact.getDefaultSender()
defaultSender.subscribed = True
Contact.update(defaultSender)
def tearDown(self):
self.testbed.deactivate()
def createContact(self, subscribed):
# Create a known contact
c = Contact(
name="mrtest",
phoneNumber=self.contactNumber,
normalizedPhoneNumber=phonenumberutils.toNormalizedNumber(self.contactNumber),
subscribed=subscribed,
)
Contact.update(c)
def test_incomingSmsMessageFromUnknownUser(self):
"""
Tests an incoming SMS message from an unknown user with xmppVoiceMail unsubscribed.
"""
# Unsubscribe the default sender.
defaultSender = Contact.getDefaultSender()
defaultSender.subscribed = False
Contact.update(defaultSender)
# Send the SMS
self.xmppvoicemail.handleIncomingSms("+16135551234", self.ownerPhoneNumber, "Hello")
# Should get an email, since user is not subscribed to xmppVoiceMail via XMPP.
self.assertEqual(1, len(self.communications.mails), "Should have sent an email.")
message = self.communications.mails[0]
self.assertEqual('"' + defaultSender.name + '" <16135551234' + self.MAIL_SUFFIX + ">", message["sender"])
self.assertEqual(self.ownerEmailAddress, message["to"])
self.assertEqual("Hello", message["subject"])
def test_incomingSmsMessageFromUnknownUserSubscribed(self):
"""
Tests an incoming SMS message from an unknown user with xmppVoiceMail subscribed.
"""
defaultSender = Contact.getDefaultSender()
# Send the SMS
self.xmppvoicemail.handleIncomingSms("+16135551234", self.ownerPhoneNumber, "Hello")
# Should get an XMPP from the default sender.
self.assertEqual(1, len(self.communications.xmppMessages), "Should have sent an XMPP message")
message = self.communications.xmppMessages[0]
self.assertEqual(defaultSender.name + self.XMPP_SUFFIX, message["fromJid"])
self.assertEqual(self.ownerJid, message["toJid"])
self.assertEqual("(613)555-1234: Hello", message["message"])
def test_incomingSmsMessageFromKnownUserUnsubscribed(self):
"""
Tests an incoming SMS message from a known user with user unsubscribed.
"""
defaultSender = Contact.getDefaultSender()
# Create a known contact
self.createContact(subscribed=False)
# Send the SMS
self.xmppvoicemail.handleIncomingSms(self.contactNumber, self.ownerPhoneNumber, "Hello")
# Should get an XMPP from the default sender.
self.assertEqual(1, len(self.communications.xmppMessages), "Should have sent an XMPP message")
message = self.communications.xmppMessages[0]
self.assertEqual(defaultSender.name + self.XMPP_SUFFIX, message["fromJid"])
self.assertEqual(self.ownerJid, message["toJid"])
self.assertEqual("(613)555-1234: Hello", message["message"])
def test_incomingSmsMessageFromKnownUserSubscribed(self):
"""
Tests an incoming SMS message from a known user with user subscribed.
"""
# Create a known contact
self.createContact(subscribed=True)
# Send the SMS
self.xmppvoicemail.handleIncomingSms(self.contactNumber, self.ownerPhoneNumber, "Hello")
# Should get an XMPP from the contact.
self.assertEqual(1, len(self.communications.xmppMessages), "Should have sent an XMPP message")
message = self.communications.xmppMessages[0]
self.assertEqual("mrtest" + self.XMPP_SUFFIX, message["fromJid"])
self.assertEqual(self.ownerJid, message["toJid"])
self.assertEqual("Hello", message["message"])
def test_incomingSmsMessageFromUnknownUserSubscribedOwnerOffline(self):
"""
Tests an incoming SMS message from an unknown user with xmppVoiceMail subscribed.
"""
# Set the owner as offline
self.communications.ownerOnline = False
# Create a known contact
self.createContact(subscribed=True)
# Send the SMS
self.xmppvoicemail.handleIncomingSms(self.contactNumber, self.ownerPhoneNumber, "Hello")
# Should get an email, since owner is offline.
self.assertEqual(1, len(self.communications.mails), "Should have sent an email.")
message = self.communications.mails[0]
self.assertEqual('"mrtest" <16135551234' + self.MAIL_SUFFIX + ">", message["sender"])
self.assertEqual(self.ownerEmailAddress, message["to"])
self.assertEqual("Hello", message["subject"])
def test_incomingXmppForNumber(self):
"""
Test an incoming XMPP to the default sender, with a number in the body.
"""
defaultSender = Contact.getDefaultSender()
self.xmppvoicemail.handleIncomingXmpp(
sender=self.ownerJid, to=defaultSender.name + self.XMPP_SUFFIX, messageBody="(613)555-1234: Hello"
)
# Should send an SMS to 555-1234
self.assertEqual(1, len(self.communications.sms), "Should have sent an SMS")
sms = self.communications.sms[0]
self.assertEqual("+16135551234", sms["toNumber"])
self.assertEqual("Hello", sms["body"])
def test_incomingXmppForContact(self):
"""
Test an incoming XMPP to a contact.
"""
self.createContact(subscribed=False)
self.createContact(subscribed=True)
self.xmppvoicemail.handleIncomingXmpp(sender=self.ownerJid, to="mrtest" + self.XMPP_SUFFIX, messageBody="Hello")
# Should send an SMS to 555-1234
self.assertEqual(1, len(self.communications.sms), "Should have sent an SMS")
sms = self.communications.sms[0]
self.assertEqual("+16135551234", sms["toNumber"])
self.assertEqual("Hello", sms["body"])
def test_incomingXmppForInvalidNumber(self):
"""
Test an incoming XMPP to the default sender, with an invalid number
in the body.
"""
with self.assertRaises(InvalidParametersException):
defaultSender = Contact.getDefaultSender()
self.xmppvoicemail.handleIncomingXmpp(
sender=self.ownerJid, to=defaultSender.name + self.XMPP_SUFFIX, messageBody="(613)555-123a: Hello"
)
def test_incomingXmppFromBadUser(self):
"""
Test an incoming email to the default sender with a number in the message body.
"""
with self.assertRaises(PermissionException):
defaultSender = Contact.getDefaultSender()
self.xmppvoicemail.handleIncomingXmpp(
sender="*****@*****.**", to=defaultSender.name + self.XMPP_SUFFIX, messageBody="(613)555-1235: Hello"
)
def test_incomingEmailForContact(self):
"""
Test an incoming email to a contact.
"""
self.createContact(subscribed=True)
self.xmppvoicemail.handleIncomingEmail(
sender=self.ownerEmailAddress, to="mrtest" + self.XMPP_SUFFIX, subject="Foo", messageBody="Hello"
)
# Should send an SMS to 555-1234
self.assertEqual(1, len(self.communications.sms), "Should have sent an SMS")
sms = self.communications.sms[0]
self.assertEqual("+16135551234", sms["toNumber"])
self.assertEqual("Hello", sms["body"])
def test_incomingEmailForNumber(self):
"""
Test an incoming email to the default sender with a number in the message body.
"""
defaultSender = Contact.getDefaultSender()
self.xmppvoicemail.handleIncomingEmail(
sender=self.ownerEmailAddress,
to=defaultSender.name + self.XMPP_SUFFIX,
subject="Foo",
messageBody="16135551234: Hello",
)
# Should send an SMS to 555-1234
self.assertEqual(1, len(self.communications.sms), "Should have sent an SMS")
sms = self.communications.sms[0]
self.assertEqual("+16135551234", sms["toNumber"])
self.assertEqual("Hello", sms["body"])
def test_incomingEmailFromBadUser(self):
"""
Test an incoming email to the default sender with a number in the message body.
"""
with self.assertRaises(PermissionException):
defaultSender = Contact.getDefaultSender()
self.xmppvoicemail.handleIncomingEmail(
sender="*****@*****.**",
to=defaultSender.name + self.XMPP_SUFFIX,
subject="Foo",
messageBody="16135551234: Hello",
)
# We don't deal with terrorists:
self.assertEqual(0, len(self.communications.mails))
