feat: New features in VR4 and assistant2

feat: New features in VR4 and assistant2

Author: Mikemosca

test/integration/test_assistant_v2.rb

@@ -52,6 +52,13 @@ def test_create_delete_session_and_message
       )
       assert((200..299).cover?(service_response.status))
 
+      service_response = service.message_stateless(
+        assistant_id: ENV["ASSISTANT_ASSISTANT_ID"],
+        input: { "text" => "Turn on the lights" },
+        context: nil
+      )
+      assert((200..299).cover?(service_response.status))
+
       service.delete_session(
         assistant_id: ENV["ASSISTANT_ASSISTANT_ID"],
         session_id: session_id

test/integration/test_visual_recognition_v4.rb

@@ -68,6 +68,15 @@ def test_list_images
       ).result
       refute(result.nil?)
     end
+
+    def test_get_model_file
+      result = @service.get_model_file(
+        collection_id: @collection_id,
+        feature: "objects",
+        model_format: "rscnn"
+      ).result
+      refute(result.nil?)
+    end
   end
 else
   class VisualRecognitionV4Test < Minitest::Test

test/unit/test_assistant_v2.rb

@@ -128,4 +128,70 @@ def test_delete_session
     )
     assert_nil(service_response)
   end
+
+  def test_message_stateless
+    # service.set_default_headers("x-watson-learning-opt-out" => true)
+    assistant_id = "f8fdbc65-e0bd-4e43-b9f8-2975a366d4ec"
+    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" => {},
+      "output" => {
+        "text" => "okay",
+        "log_messages" => []
+      }
+    }
+    headers = {
+      "Content-Type" => "application/json"
+    }
+    stub_request(:post, "https://gateway.watsonplatform.net/assistant/api/v2/assistants/f8fdbc65-e0bd-4e43-b9f8-2975a366d4ec/message?version=2018-02-16")
+      .with(
+        body: "{\"input\":{\"text\":\"Turn on the lights\"}}",
+        headers: {
+          "Accept" => "application/json",
+          "Content-Type" => "application/json",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: message_response.to_json, headers: headers)
+    service_response = service.message_stateless(
+      assistant_id: assistant_id,
+      input: { "text" => "Turn on the lights" },
+      context: nil
+    )
+    assert_equal(message_response, service_response.result)
+
+    message_ctx = {
+      "context" => {
+        "conversation_id" => "1b7b67c0-90ed-45dc-8508-9488bc483d5b",
+        "system" => {
+          "dialog_stack" => ["root"],
+          "dialog_turn_counter" => 2,
+          "dialog_request_counter" => 1
+        }
+      }
+    }
+    stub_request(:post, "https://gateway.watsonplatform.net/assistant/api/v2/assistants/f8fdbc65-e0bd-4e43-b9f8-2975a366d4ec/message?version=2018-02-16")
+      .with(
+        body: "{\"input\":{\"text\":\"Turn on the lights\"},\"context\":\"{\\\"conversation_id\\\":\\\"1b7b67c0-90ed-45dc-8508-9488bc483d5b\\\",\\\"system\\\":{\\\"dialog_stack\\\":[\\\"root\\\"],\\\"dialog_turn_counter\\\":2,\\\"dialog_request_counter\\\":1}}\"}",
+        headers: {
+          "Accept" => "application/json",
+          "Content-Type" => "application/json",
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: message_response.to_json, headers: headers)
+    service_response = service.message_stateless(
+      assistant_id: assistant_id,
+      input: { "text" => "Turn on the lights" },
+      context: message_ctx["context"].to_json
+    )
+    assert_equal(message_response, service_response.result)
+  end
 end

test/unit/test_visual_recognition_v4.rb

@@ -401,4 +401,22 @@ def test_delete_object
     )
     assert_nil(service_response)
   end
+
+  def test_get_model_file
+    response = {
+      "binary" => []
+    }
+    stub_request(:get, "https://gateway.watsonplatform.net/visual-recognition/api/v4/collections/collid/model?feature=objects&model_format=rscnn_ready&version=2018-03-19")
+      .with(
+        headers: {
+          "Host" => "gateway.watsonplatform.net"
+        }
+      ).to_return(status: 200, body: response.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_model_file(
+      collection_id: "collid",
+      feature: "objects",
+      model_format: "rscnn_ready"
+    )
+    assert_equal(response, service_response.result)
+  end
 end