def test_to_json(self): api_key = '1234' platform = '1' message = '2' intent = '3' version = '4' user_id = '5' i = Message(api_key=api_key, platform=platform, message=message, intent=intent, version=version, user_id=user_id) i.set_as_not_handled() i.set_as_feedback() self.assertEqual( json.loads(i.to_json()), { 'api_key': api_key, 'platform': platform, 'message': message, 'intent': intent, 'version': version, 'user_id': user_id, 'time_stamp': i.time_stamp, 'type': MessageTypes.USER, # since we did not set as type agent 'not_handled': True, 'feedback': True })
def test_not_handled_setting(self): i = Message() i.set_as_not_handled() self.assertTrue(i.not_handled) i.set_as_handled() self.assertFalse(i.not_handled)