Skip to content

Commit a22a061

Browse files
maxnussbaumGERMAN ATTANASIO RUIZ
maxnussbaum
authored and
GERMAN ATTANASIO RUIZ
committed
test(discovery): Add more tests for discovery
1 parent c378204 commit a22a061

File tree

1 file changed

+206
-3
lines changed

1 file changed

+206
-3
lines changed

‎test/unit/test_discovery_v1.rb‎

Lines changed: 206 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,15 @@ def test_document
425425
service_response = service.test_configuration_in_environment(
426426
environment_id: "envid",
427427
configuration_id: "bogus",
428-
file: file_info
428+
file: file_info,
429+
filename: "simple.html"
430+
)
431+
refute(service_response.nil?)
432+
433+
service_response = service.test_configuration_in_environment(
434+
environment_id: "envid",
435+
configuration_id: "bogus",
436+
file: "file_info"
429437
)
430438
refute(service_response.nil?)
431439

@@ -494,7 +502,8 @@ def test_document
494502
service_response = service.update_document(
495503
environment_id: "envid",
496504
collection_id: "collid",
497-
document_id: "docid"
505+
document_id: "docid",
506+
file: "file"
498507
)
499508
assert_equal({ "body" => [] }, service_response.body)
500509

@@ -516,7 +525,7 @@ def test_document
516525
service_response = service.add_document(
517526
environment_id: "envid",
518527
collection_id: "collid",
519-
file: StringIO.new("my string of file"),
528+
file: "my string of file",
520529
filename: "file.txt"
521530
)
522531
refute(service_response.nil?)
@@ -904,4 +913,198 @@ def test_delete_user_data
904913
)
905914
assert_nil(service_response)
906915
end
916+
917+
def test_query_notices
918+
service = WatsonAPIs::DiscoveryV1.new(
919+
username: "username",
920+
password: "password",
921+
version: "2018-03-05"
922+
)
923+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/collections/collid/notices?version=2018-03-05")
924+
.with(
925+
headers: {
926+
"Accept" => "application/json",
927+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
928+
"Host" => "gateway.watsonplatform.net"
929+
}
930+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
931+
service_response = service.query_notices(
932+
environment_id: "envid",
933+
collection_id: "collid"
934+
)
935+
assert_equal({ "received" => "true" }, service_response.body)
936+
end
937+
938+
def test_federated_query
939+
service = WatsonAPIs::DiscoveryV1.new(
940+
username: "username",
941+
password: "password",
942+
version: "2018-03-05"
943+
)
944+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/query?collection_ids=collid&version=2018-03-05")
945+
.with(
946+
headers: {
947+
"Accept" => "application/json",
948+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
949+
"Host" => "gateway.watsonplatform.net"
950+
}
951+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
952+
service_response = service.federated_query(
953+
environment_id: "envid",
954+
collection_ids: ["collid"]
955+
)
956+
assert_equal({ "received" => "true" }, service_response.body)
957+
end
958+
959+
def test_federated_query_notices
960+
service = WatsonAPIs::DiscoveryV1.new(
961+
username: "username",
962+
password: "password",
963+
version: "2018-03-05"
964+
)
965+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/notices?collection_ids=collid&version=2018-03-05")
966+
.with(
967+
headers: {
968+
"Accept" => "application/json",
969+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
970+
"Host" => "gateway.watsonplatform.net"
971+
}
972+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
973+
service_response = service.federated_query_notices(
974+
environment_id: "envid",
975+
collection_ids: ["collid"]
976+
)
977+
assert_equal({ "received" => "true" }, service_response.body)
978+
end
979+
980+
def test_list_training_examples
981+
service = WatsonAPIs::DiscoveryV1.new(
982+
username: "username",
983+
password: "password",
984+
version: "2018-03-05"
985+
)
986+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/collections/collid/training_data/queryid/examples?version=2018-03-05")
987+
.with(
988+
headers: {
989+
"Accept" => "application/json",
990+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
991+
"Host" => "gateway.watsonplatform.net"
992+
}
993+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
994+
service_response = service.list_training_examples(
995+
environment_id: "envid",
996+
collection_id: "collid",
997+
query_id: "queryid"
998+
)
999+
assert_equal({ "received" => "true" }, service_response.body)
1000+
end
1001+
1002+
def test_list_credentials
1003+
service = WatsonAPIs::DiscoveryV1.new(
1004+
username: "username",
1005+
password: "password",
1006+
version: "2018-03-05"
1007+
)
1008+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/credentials?version=2018-03-05")
1009+
.with(
1010+
headers: {
1011+
"Accept" => "application/json",
1012+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
1013+
"Host" => "gateway.watsonplatform.net"
1014+
}
1015+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
1016+
service_response = service.list_credentials(
1017+
environment_id: "envid"
1018+
)
1019+
assert_equal({ "received" => "true" }, service_response.body)
1020+
end
1021+
1022+
def test_create_credentials
1023+
service = WatsonAPIs::DiscoveryV1.new(
1024+
username: "username",
1025+
password: "password",
1026+
version: "2018-03-05"
1027+
)
1028+
stub_request(:post, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/credentials?version=2018-03-05")
1029+
.with(
1030+
body: "{\"source_type\":\"salesforce\"}",
1031+
headers: {
1032+
"Accept" => "application/json",
1033+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
1034+
"Content-Type" => "application/json",
1035+
"Host" => "gateway.watsonplatform.net"
1036+
}
1037+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
1038+
service_response = service.create_credentials(
1039+
environment_id: "envid",
1040+
source_type: "salesforce"
1041+
)
1042+
assert_equal({ "received" => "true" }, service_response.body)
1043+
end
1044+
1045+
def test_get_credentials
1046+
service = WatsonAPIs::DiscoveryV1.new(
1047+
username: "username",
1048+
password: "password",
1049+
version: "2018-03-05"
1050+
)
1051+
stub_request(:get, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/credentials/credid?version=2018-03-05")
1052+
.with(
1053+
headers: {
1054+
"Accept" => "application/json",
1055+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
1056+
"Host" => "gateway.watsonplatform.net"
1057+
}
1058+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
1059+
service_response = service.get_credentials(
1060+
environment_id: "envid",
1061+
credential_id: "credid"
1062+
)
1063+
assert_equal({ "received" => "true" }, service_response.body)
1064+
end
1065+
1066+
def test_update_credentials
1067+
service = WatsonAPIs::DiscoveryV1.new(
1068+
username: "username",
1069+
password: "password",
1070+
version: "2018-03-05"
1071+
)
1072+
stub_request(:put, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/credentials/credid?version=2018-03-05")
1073+
.with(
1074+
body: "{\"source_type\":\"salesforce\"}",
1075+
headers: {
1076+
"Accept" => "application/json",
1077+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
1078+
"Content-Type" => "application/json",
1079+
"Host" => "gateway.watsonplatform.net"
1080+
}
1081+
).to_return(status: 200, body: { "received" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
1082+
service_response = service.update_credentials(
1083+
environment_id: "envid",
1084+
credential_id: "credid",
1085+
source_type: "salesforce"
1086+
)
1087+
assert_equal({ "received" => "true" }, service_response.body)
1088+
end
1089+
1090+
def test_delete_credentials
1091+
service = WatsonAPIs::DiscoveryV1.new(
1092+
username: "username",
1093+
password: "password",
1094+
version: "2018-03-05"
1095+
)
1096+
stub_request(:delete, "https://gateway.watsonplatform.net/discovery/api/v1/environments/envid/credentials/credid?version=2018-03-05")
1097+
.with(
1098+
headers: {
1099+
"Accept" => "application/json",
1100+
"Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
1101+
"Host" => "gateway.watsonplatform.net"
1102+
}
1103+
).to_return(status: 200, body: { "deleted" => "true" }.to_json, headers: { "Content-Type" => "application/json" })
1104+
service_response = service.delete_credentials(
1105+
environment_id: "envid",
1106+
credential_id: "credid"
1107+
)
1108+
assert_equal({ "deleted" => "true" }, service_response.body)
1109+
end
9071110
end

0 commit comments

Comments
 (0)