Skip to content

Commit 7b965f6

Browse files
🎨 improving retrieve and rank tests
1 parent 295429a commit 7b965f6

File tree

2 files changed

+216
-160
lines changed

2 files changed

+216
-160
lines changed

test/test_conversation_v1.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22
import watson_developer_cloud
33

44
@responses.activate
5-
def test_success():
6-
5+
def test_message():
6+
77
# Ranker endpoints
88
conversation = watson_developer_cloud.ConversationV1(username="username",
99
password="password",
@@ -13,23 +13,26 @@ def test_success():
1313
message_url = 'https://gateway.watsonplatform.net/conversation/api/v1/workspaces/%s/message' % workspace_id
1414
message_url1 = 'https://gateway.watsonplatform.net/conversation/api/v1/workspaces/%s/message?version=2016-07-11' % workspace_id
1515
message_response = '{"context":{"conversation_id":"1b7b67c0-90ed-45dc-8508-9488bc483d5b","system":{"dialog_stack":["root"],"dialog_turn_counter":1,"dialog_request_counter":1}},"intents":[],"entities":[],"input":{}}'
16-
16+
1717
responses.add(responses.POST, message_url,
1818
body=message_response, status=200,
1919
content_type='application/json')
2020

21-
conversation.message(workspace_id=workspace_id, message_input={'text': 'Turn on the lights'}, context=None)
22-
21+
message = conversation.message(workspace_id=workspace_id, message_input={'text': 'Turn on the lights'}, context=None)
22+
23+
assert message is not None
2324
assert responses.calls[0].request.url == message_url1
2425
assert responses.calls[0].response.text == message_response
2526

27+
2628
# test context
2729
responses.add(responses.POST, message_url,
2830
body=message_response, status=200,
2931
content_type='application/json')
30-
31-
conversation.message(workspace_id=workspace_id, message_input={'text': 'Turn on the lights'}, context={'context': {'conversation_id':'1b7b67c0-90ed-45dc-8508-9488bc483d5b', 'system': {'dialog_stack':['root'], 'dialog_turn_counter':2, 'dialog_request_counter':1}}})
3232

33+
message = conversation.message(workspace_id=workspace_id, message_input={'text': 'Turn on the lights'}, context={'context': {'conversation_id':'1b7b67c0-90ed-45dc-8508-9488bc483d5b', 'system': {'dialog_stack':['root'], 'dialog_turn_counter':2, 'dialog_request_counter':1}}})
34+
35+
assert message is not None
3336
assert responses.calls[1].request.url == message_url1
3437
assert responses.calls[1].response.text == message_response
3538

0 commit comments

Comments
 (0)