ibm_watson 2.0.0.rc1 → 2.1.1

data/lib/ibm_watson/tone_analyzer_v3.rb

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.17.0-8d569e8f-20201030-142059
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
 # The IBM Watson™ Tone Analyzer service uses linguistic analysis to detect
 # emotional and language tones in written text. The service can analyze tone at both the
@@ -34,7 +34,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Tone Analyzer V3 service.

data/lib/ibm_watson/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module IBMWatson
-  VERSION = "2.0.0.rc1"
+  VERSION = "2.1.1"
 end

data/lib/ibm_watson/visual_recognition_v3.rb

@@ -14,11 +14,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.17.0-8d569e8f-20201030-142059
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
-# The IBM Watson™ Visual Recognition service uses deep learning algorithms to
-# identify scenes and objects in images that you upload to the service. You can create and
-# train a custom classifier to identify subjects that suit your needs.
+# IBM Watson™ Visual Recognition is discontinued. Existing instances are supported
+# until 1 December 2021, but as of 7 January 2021, you can't create instances. Any
+# instance that is provisioned on 1 December 2021 will be deleted.
+# {: deprecated}
+#
+# The IBM Watson Visual Recognition service uses deep learning algorithms to identify
+# scenes and objects in images that you upload to the service. You can create and train a
+# custom classifier to identify subjects that suit your needs.
 
 require "concurrent"
 require "erb"
@@ -26,7 +31,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Visual Recognition V3 service.
@@ -48,6 +52,7 @@ module IBMWatson
     # @option args service_name [String] The name of the service to configure. Will be used as the key to load
     #   any external configuration, if applicable.
     def initialize(args = {})
+      warn "On 1 December 2021, Visual Recognition will no longer be available. For more information, see https://github.com/watson-developer-cloud/ruby-sdk/tree/master#visual-recognition-deprecation."
       @__async_initialized__ = false
       defaults = {}
       defaults[:service_url] = DEFAULT_SERVICE_URL

data/lib/ibm_watson/visual_recognition_v4.rb

@@ -14,10 +14,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-# IBM OpenAPI SDK Code Generator Version: 3.17.0-8d569e8f-20201030-142059
+# IBM OpenAPI SDK Code Generator Version: 3.31.0-902c9336-20210504-161156
 #
-# Provide images to the IBM Watson™ Visual Recognition service for analysis. The
-# service detects objects based on a set of images with training data.
+# IBM Watson™ Visual Recognition is discontinued. Existing instances are supported
+# until 1 December 2021, but as of 7 January 2021, you can't create instances. Any
+# instance that is provisioned on 1 December 2021 will be deleted.
+# {: deprecated}
+#
+# Provide images to the IBM Watson Visual Recognition service for analysis. The service
+# detects objects based on a set of images with training data.
 
 require "concurrent"
 require "erb"
@@ -25,7 +30,6 @@ require "json"
 require "ibm_cloud_sdk_core"
 require_relative "./common.rb"
 
-# Module for the Watson APIs
 module IBMWatson
   ##
   # The Visual Recognition V4 service.
@@ -47,6 +51,7 @@ module IBMWatson
     # @option args service_name [String] The name of the service to configure. Will be used as the key to load
     #   any external configuration, if applicable.
     def initialize(args = {})
+      warn "On 1 December 2021, Visual Recognition will no longer be available. For more information, see https://github.com/watson-developer-cloud/ruby-sdk/tree/master#visual-recognition-deprecation."
       @__async_initialized__ = false
       defaults = {}
       defaults[:service_url] = DEFAULT_SERVICE_URL

data/test/integration/test_discovery_v2.rb

@@ -120,6 +120,7 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
       )
       query_id = service_response.result["query_id"]
       refute(service_response.nil?)
+      # puts JSON.pretty_generate(service_response.result)
 
       service_response = service.get_training_query(
         project_id: @project_id,
@@ -134,6 +135,12 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
         examples: []
       )
       refute(service_response.nil?)
+
+      service_response = service.delete_training_query(
+        project_id: @project_id,
+        query_id: query_id
+      )
+      assert(service_response.nil?)
     end
 
     def test_create_get_update_delete_collection
@@ -238,6 +245,14 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
       assert(service_response.nil?)
     end
 
+    def test_query_collection_notices
+      service_response = service.query_collection_notices(
+        project_id: @project_id,
+        collection_id: @collection_id
+      )
+      refute(service_response.result.nil?)
+    end
+
     def test_delete_user_data
       skip "Covered with the unit test.  No need to run it here"
 
@@ -246,6 +261,20 @@ if !ENV["DISCOVERY_V2_APIKEY"].nil?
       )
       assert(service_response.nil?)
     end
+
+    def test_analyze_document
+      skip "CPD only.  Do not run on premium"
+      analyze_data = File.open(Dir.getwd + "/resources/problem.json")
+
+      service_response = service.analyze_document(
+        project_id: @project_id,
+        collection_id: @collection_id,
+        file: analyze_data,
+        file_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+      refute(service_response.nil?)
+    end
   end
 else
   class DiscoveryV2Test < Minitest::Test

data/test/integration/test_natural_language_understanding_v1.rb

@@ -86,7 +86,7 @@ if !ENV["NATURAL_LANGUAGE_UNDERSTANDING_APIKEY"].nil? && !ENV["NATURAL_LANGUAGE_
     end
 
     def test_list_models
-      skip
+      # skip
       service.add_default_headers(
         headers: {
           "X-Watson-Learning-Opt-Out" => "1",
@@ -96,6 +96,139 @@ if !ENV["NATURAL_LANGUAGE_UNDERSTANDING_APIKEY"].nil? && !ENV["NATURAL_LANGUAGE_
       service_response = service.list_models
       assert((200..299).cover?(service_response.status))
     end
+
+    def test_check_orphands
+      skip "Use to help delete old models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      service_response = service.list_sentiment_models
+      puts JSON.pretty_generate(service_response.result)
+      service_response = service.list_categories_models
+      puts JSON.pretty_generate(service_response.result)
+      service_response = service.list_classifications_models
+      puts JSON.pretty_generate(service_response.result)
+
+      # service.delete_sentiment_model(model_id: "model_id1")
+      # service.delete_categories_model(model_id: "0122b971-94c9-4468-a98f-930f4ce28c32")
+      # service.delete_classifications_model(model_id: "0122b971-94c9-4468-a98f-930f4ce28c32")
+    end
+
+    def test_sentiment_models
+      # skip "test_sentiment_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_sentiment_data.csv")
+      service_response = service.create_sentiment_model(
+        language: "en",
+        training_data: training_data
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_sentiment_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.get_sentiment_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_sentiment_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_sentiment_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
+
+    def test_categories_models
+      # skip "test_categories_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_categories_training.json")
+      service_response = service.create_categories_model(
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_categories_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+      # puts JSON.pretty_generate(service_response.result)
+      # puts JSON.pretty_generate(model_id)
+
+      service_response = service.get_categories_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_categories_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_categories_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
+
+    def test_classifications_models
+      # skip "test_classifications_models"
+      service.add_default_headers(
+        headers: {
+          "X-Watson-Learning-Opt-Out" => "1",
+          "X-Watson-Test" => "1"
+        }
+      )
+      training_data = File.open(Dir.getwd + "/resources/nlu_classifications_training.json")
+      service_response = service.create_classifications_model(
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.list_classifications_models
+      model_id = service_response.result["models"][0]["model_id"]
+      assert((200..299).cover?(service_response.status))
+      # puts JSON.pretty_generate(service_response.result)
+      # puts JSON.pretty_generate(model_id)
+
+      service_response = service.get_classifications_model(model_id: model_id)
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.update_classifications_model(
+        model_id: model_id,
+        language: "en",
+        training_data: training_data,
+        training_data_content_type: "application/json"
+      )
+      assert((200..299).cover?(service_response.status))
+
+      service_response = service.delete_classifications_model(
+        model_id: model_id
+      )
+      assert((200..299).cover?(service_response.status))
+    end
   end
 else
   class NaturalLanguageUnderstandingV1Test < Minitest::Test

data/test/integration/test_text_to_speech_v1.rb

@@ -83,6 +83,63 @@ if !ENV["TEXT_TO_SPEECH_APIKEY"].nil? && !ENV["TEXT_TO_SPEECH_URL"].nil?
         customization_id: customization_id
       )
     end
+
+    def test_custom_prompts
+      customization_id = @service.create_custom_model(
+        name: "test_integration_customization",
+        description: "RUBY customization for tests"
+      ).result["customization_id"]
+
+      prompt_metadata = { prompt_text: "Thank you and goodbye" }
+      prompt_json = JSON[prompt_metadata]
+      audio_file = File.open(Dir.getwd + "/resources/speech.wav")
+      prompt = @service.add_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id",
+        metadata: prompt_json,
+        file: audio_file
+      ).result["prompt"]
+      assert(prompt == "Thank you and goodbye")
+
+      prompts = @service.list_custom_prompts(customization_id: customization_id).result["prompts"]
+      refute(prompts.nil?)
+
+      prompt = @service.get_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id"
+      ).result["prompt"]
+      assert(prompt == "Thank you and goodbye")
+
+      @service.delete_custom_prompt(
+        customization_id: customization_id,
+        prompt_id: "rubyprompt_id"
+      )
+
+      @service.delete_custom_model(
+        customization_id: customization_id
+      )
+    end
+
+    def test_speaker_models
+      audio_file = File.open(Dir.getwd + "/resources/speech.wav")
+      @service.create_speaker_model(
+        speaker_name: "RubyMike",
+        audio: audio_file
+      ).result["speaker_model"]
+
+      speakers = @service.list_speaker_models.result["speakers"]
+      speaker_id = speakers[0]["speaker_id"]
+      # puts JSON.pretty_generate(speakers)
+
+      @service.get_speaker_model(
+        speaker_id: speaker_id
+      ).result["models_speaker"]
+
+      # speaker_id = d3d03b69-4035-4420-928d-7ac2e0249829
+      @service.delete_speaker_model(
+        speaker_id: speaker_id
+      )
+    end
   end
 else
   class TextToSpeechV1Test < Minitest::Test

data/test/unit/test_assistant_v1.rb

@@ -8,7 +8,7 @@ SimpleCov.command_name "test:unit"
 
 WebMock.disable_net_connect!(allow_localhost: true)
 
-# Unit tests for the Watson Assistant V1 Service
+# Unit tests for the Watson Assistant V1 Services
 class AssistantV1Test < Minitest::Test
   include Minitest::Hooks
   attr_accessor :service

data/test/unit/test_discovery_v2.rb

@@ -467,4 +467,33 @@ class DiscoveryV2Test < Minitest::Test
     )
     assert_nil(service_response)
   end
+
+  def test_delete_training_query
+    stub_request(:delete, "https://api.us-south.discovery.watson.cloud.ibm.com/discovery/api/v2/projects/project/training_data/queries/queryid?version=2018-03-05")
+      .with(
+        headers: {
+          "Host" => "api.us-south.discovery.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { "body" => "hello" }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_training_query(
+      project_id: "project",
+      query_id: "queryid"
+    )
+    assert_nil(service_response)
+  end
+
+  def test_query_collection_notices
+    stub_request(:get, "https://api.us-south.discovery.watson.cloud.ibm.com/discovery/api/v2/projects/project/collections/collection/notices?version=2018-03-05")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Host" => "api.us-south.discovery.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { "body" => "hello" }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.query_collection_notices(
+      project_id: "project",
+      collection_id: "collection"
+    )
+    assert_equal({ "body" => "hello" }, service_response.result)
+  end
 end

data/test/unit/test_natural_language_understanding_v1.rb

@@ -144,4 +144,235 @@ class NaturalLanguageUnderstandingV1Test < Minitest::Test
     )
     assert_equal({ "deleted" => model_id }, service_response.result)
   end
+
+  def test_sentiment_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_sentiment_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_sentiment_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_sentiment_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_sentiment_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/sentiment/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_sentiment_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
+
+  def test_categories_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_categories_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_categories_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_categories_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_categories_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/categories/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_categories_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
+
+  def test_classifications_models
+    authenticator = IBMWatson::Authenticators::BasicAuthenticator.new(
+      username: "username",
+      password: "password"
+    )
+    service = IBMWatson::NaturalLanguageUnderstandingV1.new(
+      version: "2018-03-16",
+      authenticator: authenticator
+    )
+
+    stub_request(:post, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.create_classifications_model(
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.list_classifications_models
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:get, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.get_classifications_model(model_id: "model_id")
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:put, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    training_data = File.open(Dir.getwd + "/resources/weather_data_train.csv")
+    service_response = service.update_classifications_model(
+      model_id: "model_id",
+      language: "en",
+      training_data: training_data
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+
+    stub_request(:delete, "https://api.us-south.natural-language-understanding.watson.cloud.ibm.com/v1/models/classifications/model_id?version=2018-03-16")
+      .with(
+        headers: {
+          "Accept" => "application/json",
+          "Authorization" => "Basic dXNlcm5hbWU6cGFzc3dvcmQ=",
+          "Host" => "api.us-south.natural-language-understanding.watson.cloud.ibm.com"
+        }
+      ).to_return(status: 200, body: { resulting_key: true }.to_json, headers: { "Content-Type" => "application/json" })
+    service_response = service.delete_classifications_model(
+      model_id: "model_id"
+    )
+    assert_equal({ "resulting_key" => true }, service_response.result)
+  end
 end