def testGetSubclass(self):
sub_upper_msg = Message('12345', "SUBSCRIBE 0 0")
subscription = sub_upper_msg.get_subclass()
self.assertEqual(subscription.__class__, Subscription)
self.assertNotEqual(subscription.__class__, Alert)
non_sub_msg = Message('12345', "hello world")
alert = non_sub_msg.get_subclass()
self.assertEqual(alert.__class__, Alert)
self.assertNotEqual(alert.__class__, Subscription)
def sms():
number = request.form.get('From', None)
body = request.form.get('Body', None)
#try:
if number and body:
message = Message(number, body)
return message.get_subclass().handle() # Handle the subscription or alert
#except:
else:
return render_template('subscription_failed.twiml')
def testIsSubscription(self):
sub_upper_msg = Message('12345', "SUBSCRIBE 0 0")
self.assertTrue(sub_upper_msg.is_subscription())
sub_lower_msg = Message('12345', "subscribe 0 0")
self.assertTrue(sub_lower_msg.is_subscription())
sub_varied_msg = Message('12345', "sUbScRiBe 0 0")
self.assertTrue(sub_varied_msg.is_subscription())
non_sub_msg = Message('12345', "hello world")
self.assertFalse(non_sub_msg.is_subscription())
def testConstructor(self):
message = Message('123456', 'hello world')
self.assertEquals('123456', message.get_number())
