def test_to_json(self):
api_key = '1234'
platform = '1'
message = '2'
intent = '3'
version = '4'
user_id = '5'
time_stamp = int(round(time.time() * 1e3))
i = Message(api_key=api_key,
platform=platform,
message=message,
intent=intent,
version=version,
user_id=user_id,
type=MessageTypes.USER,
not_handled=True,
time_stamp=time_stamp)
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': time_stamp,
'type':
MessageTypes.USER, # since we did not set as type agent
'not_handled': True,
'feedback': True
})
def test_message_set_append_message(self):
api_key = '1234'
platform = '1'
message = '2'
intent = '3'
version = '4'
user_id = '5'
time_stamp = int(round(time.time() * 1e3))
msg1 = Message(api_key=api_key,
platform=platform,
message=message,
intent=intent,
version=version,
user_id=user_id,
type=MessageTypes.USER,
not_handled=True,
time_stamp=time_stamp)
msg1.set_as_feedback()
msg2 = Message(api_key=api_key,
platform=platform,
message=message,
version=version,
user_id=user_id,
type=MessageTypes.AGENT)
message_set = MessageSet(api_key=api_key,
platform=platform,
version=version,
user_id=user_id)
message_set.append_message(msg1)
message_set.append_message(msg2)
msg1 = message_set.messages[0]
self.assertEqual(
json.loads(msg1.to_json()),
{
'api_key': api_key,
'platform': platform,
'message': message,
'intent': intent,
'version': version,
'user_id': user_id,
'time_stamp': time_stamp,
'type':
MessageTypes.USER, # since we did not set as type agent
'not_handled': True,
'feedback': True
})
msg2 = message_set.messages[1]
self.assertEqual(
json.loads(msg2.to_json()),
{
'api_key': api_key,
'platform': platform,
'message': message,
'intent': msg2.intent,
'version': version,
'user_id': user_id,
'time_stamp': msg2.time_stamp,
'type': MessageTypes.AGENT, # since we did set as type agent
'not_handled': False,
'feedback': False
})
