ibm_watson 0.1.0

data/test/integration/test_discovery_v1.rb

@@ -0,0 +1,200 @@
+# frozen_string_literal: true
+
+require_relative("./../test_helper.rb")
+require("minitest/hooks/test")
+
+# Integration tests for the Discovery V1 Service
+class DiscoveryV1Test < Minitest::Test
+  include Minitest::Hooks
+  attr_accessor :service, :environment_id, :collection_id
+  def before_all
+    @service = IBMWatson::DiscoveryV1.new(
+      username: ENV["DISCOVERY_USERNAME"],
+      password: ENV["DISCOVERY_PASSWORD"],
+      version: "2018-03-05"
+    )
+    @environment_id = ENV["DISCOVERY_ENVIRONMENT_ID"]
+    @collection_id = ENV["DISCOVERY_COLLECTION_ID"]
+    @service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+  end
+
+  def test_environments
+    envs = @service.list_environments.result
+    refute(envs.nil?)
+    env = @service.get_environment(
+      environment_id: envs["environments"][0]["environment_id"]
+    )
+    refute(env.nil?)
+    fields = @service.list_fields(
+      environment_id: @environment_id,
+      collection_ids: [@collection_id]
+    )
+    refute(fields.nil?)
+  end
+
+  def test_configurations
+    configs = @service.list_configurations(
+      environment_id: @environment_id
+    ).result
+    refute(configs.nil?)
+
+    name = "test" + ("A".."Z").to_a.sample
+    new_configuration_id = @service.create_configuration(
+      environment_id: @environment_id,
+      name: name,
+      description: "creating new config for ruby sdk"
+    ).result
+    new_configuration_id = new_configuration_id["configuration_id"]
+    refute(new_configuration_id.nil?)
+    @service.get_configuration(
+      environment_id: @environment_id,
+      configuration_id: new_configuration_id
+    )
+
+    updated_config = @service.update_configuration(
+      environment_id: @environment_id,
+      configuration_id: new_configuration_id,
+      name: "lala"
+    ).result
+    updated_config = updated_config
+    assert_equal("lala", updated_config["name"])
+
+    deleted_config = @service.delete_configuration(
+      environment_id: @environment_id,
+      configuration_id: new_configuration_id
+    ).result
+    deleted_config = deleted_config
+    assert_equal("deleted", deleted_config["status"])
+  end
+
+  def test_collections_and_expansions
+    skip "Time consuming"
+    name = "Example collection for ruby" + ("A".."Z").to_a.sample
+    new_collection_id = @service.create_collection(
+      environment_id: @environment_id,
+      name: name,
+      description: "Integration test for ruby sdk"
+    ).result
+    new_collection_id = new_collection_id["collection_id"]
+    refute(new_collection_id.nil?)
+
+    collection_status = { "status" => "pending" }
+    while collection_status["status"] == "pending"
+      sleep(1)
+      collection_status = @service.get_collection(
+        environment_id: @environment_id,
+        collection_id: new_collection_id
+      ).result
+      collection_status = collection_status
+    end
+    updated_collection = @service.update_collection(
+      environment_id: @environment_id,
+      collection_id: new_collection_id,
+      name: name,
+      description: "Updating description"
+    ).result
+    updated_collection = updated_collection
+    assert_equal("Updating description", updated_collection["description"])
+
+    @service.create_expansions(
+      environment_id: @environment_id,
+      collection_id: new_collection_id,
+      expansions: [
+        {
+          "input_terms" => ["a"],
+          "expanded_terms" => ["aa"]
+        }
+      ]
+    )
+    expansions = @service.list_expansions(
+      environment_id: @environment_id,
+      collection_id: new_collection_id
+    ).result
+    expansions = expansions
+    refute(expansions["expansions"].nil?)
+
+    @service.delete_expansions(
+      environment_id: @environment_id,
+      collection_id: new_collection_id
+    )
+    deleted_collection = @service.delete_collection(
+      environment_id: @environment_id,
+      collection_id: new_collection_id
+    ).result
+    deleted_collection = deleted_collection
+    assert_equal("deleted", deleted_collection["status"])
+  end
+
+  def test_documents
+    skip "Time consuming"
+    add_doc = nil
+    File.open(Dir.getwd + "/resources/simple.html") do |file_info|
+      add_doc = @service.add_document(
+        environment_id: @environment_id,
+        collection_id: @collection_id,
+        file: file_info
+      ).result
+    end
+    add_doc = add_doc
+    refute(add_doc["document_id"].nil?)
+
+    doc_status = { "status" => "processing" }
+    while doc_status["status"] == "processing"
+      sleep(1)
+      doc_status = @service.get_document_status(
+        environment_id: @environment_id,
+        collection_id: @collection_id,
+        document_id: add_doc["document_id"]
+      ).result
+      doc_status = doc_status
+    end
+    assert_equal("available", doc_status["status"])
+
+    update_doc = nil
+    File.open(Dir.getwd + "/resources/simple.html") do |file_info|
+      update_doc = @service.update_document(
+        environment_id: @environment_id,
+        collection_id: @collection_id,
+        document_id: add_doc["document_id"],
+        file: file_info,
+        filename: "newname.html"
+      ).result
+    end
+    refute(update_doc.nil?)
+
+    doc_status = { "status" => "processing" }
+    while doc_status["status"] == "processing"
+      sleep(1)
+      doc_status = @service.get_document_status(
+        environment_id: @environment_id,
+        collection_id: @collection_id,
+        document_id: add_doc["document_id"]
+      ).result
+      doc_status = doc_status
+    end
+    assert_equal("available", doc_status["status"])
+
+    delete_doc = @service.delete_document(
+      environment_id: @environment_id,
+      collection_id: @collection_id,
+      document_id: add_doc["document_id"]
+    ).result
+    delete_doc = delete_doc
+    assert_equal("deleted", delete_doc["status"])
+  end
+
+  def test_queries
+    query_results = @service.query(
+      environment_id: @environment_id,
+      collection_id: @collection_id,
+      filter: "extracted_metadata.sha1::9181d244*",
+      return_fields: ["extracted_metadata.sha1"]
+    ).result
+    refute(query_results.nil?)
+  end
+end

data/test/integration/test_iam_assistant_v1.rb

@@ -0,0 +1,707 @@
+# frozen_string_literal: true
+
+require_relative("./../test_helper.rb")
+
+# Integration tests for the Watson Assistant V1 Service
+class IAMAssistantV1Test < Minitest::Test
+  def test_create_update_delete_workspace
+    service = IBMWatson::AssistantV1.new(
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service._iam_api_key(iam_api_key: ENV["ASSISTANT_IAM_APIKEY"])
+    service._iam_api_key(iam_api_key: ENV["ASSISTANT_IAM_APIKEY"])
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_workspace(
+      name: "Pizza app",
+      description: "Pizza app",
+      language: "en",
+      metadata: {}
+    )
+    workspace_id = service_response.result["workspace_id"]
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_workspace(
+      workspace_id: workspace_id,
+      name: "Pizza app",
+      description: "Pizza app",
+      language: "en",
+      metadata: {}
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_workspace(
+      workspace_id: workspace_id
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_workspace
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_workspace(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      export: false
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_workspaces
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_workspaces
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_counterexample
+    service = IBMWatson::AssistantV1.new(
+      version: "2018-02-16",
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"]
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_counterexample(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      text: "I want a large pizza please."
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_counterexample(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      text: "I want a large pizza please.",
+      new_text: "I want a large pizza please."
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_counterexample(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      text: "I want a large pizza please."
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_counterexample
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_counterexample(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      text: "what is your nickname"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_counterexamples
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_counterexamples(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"]
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_entity
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_entity(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "pizza_toppings",
+      description: "Tasty pizza toppings",
+      metadata: { "property" => "value" },
+      values: nil,
+      fuzzy_match: nil
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_entity(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "pizza_toppings",
+      new_entity: "pizza_toppings"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_entity(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "pizza_toppings"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_entity
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_entity(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      export: true
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_entities
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_entities(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      export: true
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_example
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_example(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel",
+      text: "Gimme a pizza with pepperoni"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_example(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel",
+      text: "Gimme a pizza with pepperoni",
+      new_text: "Gimme a pizza with pepperoni"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_example(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel",
+      text: "Gimme a pizza with pepperoni"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_example
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_example(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel",
+      text: "forget it"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_examples
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_examples(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_intent
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_intent(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "pizza_order",
+      description: "User wants to start a new pizza order"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_intent(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "pizza_order",
+      new_intent: "pizza_order",
+      new_description: "User wants to start a new pizza order"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_intent(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "pizza_order"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_intent
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_intent(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      intent: "Cancel",
+      export: false
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_intents
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_intents(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      export: false
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_logs
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_logs(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"]
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_all_logs
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_all_logs(
+      filter: "language::en,request.context.metadata.deployment::deployment_1"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_message
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.message(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      input: { "text" => "Turn on the lights" },
+      context: nil
+    )
+    assert((200..299).cover?(service_response.status))
+
+    context = service_response.result["context"]
+    service_response = service.message(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      input: { "text" => "Turn on the lights" },
+      context: context
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_synonym
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_synonym(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "labor day",
+      synonym: "a"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_synonym(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "labor day",
+      synonym: "a",
+      new_synonym: "a"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_synonym(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "labor day",
+      synonym: "a"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_synonym
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_synonym(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "thanksgiving",
+      synonym: "turkey day"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_synonyms
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_synonyms(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "new years eve"
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_create_update_delete_value
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.create_value(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "aeiou"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.update_value(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "aeiou",
+      new_value: "BBQ sauce",
+      new_metadata: { "code" => 1422 },
+      new_synonyms: nil
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_value(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "BBQ sauce"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_value
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_value(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      value: "christmas",
+      export: true
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_list_values
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.list_values(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      entity: "holiday",
+      export: true
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_dialog_nodes
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+
+    service_response = service.create_dialog_node(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      dialog_node: "location-done"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.delete_dialog_node(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      dialog_node: "location-done"
+    )
+    assert(service_response.nil?)
+
+    service_response = service.get_dialog_node(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      dialog_node: "Hours of Operation"
+    )
+    assert((200..299).cover?(service_response.status))
+
+    service_response = service.list_dialog_nodes(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"]
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_delete_user_data
+    service = IBMWatson::AssistantV1.new(
+      iam_api_key: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.delete_user_data(
+      customer_id: "id"
+    )
+    assert(service_response.nil?)
+  end
+
+  def test_get_workspace_with_iam_user_pass_params
+    service = IBMWatson::AssistantV1.new(
+      username: "apikey",
+      password: ENV["ASSISTANT_IAM_APIKEY"],
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service_response = service.get_workspace(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      export: false
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_get_workspace_with_iam_user_pass_non_params
+    service = IBMWatson::AssistantV1.new(
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    service.username = "apikey"
+    service.password = ENV["ASSISTANT_IAM_APIKEY"]
+    service_response = service.get_workspace(
+      workspace_id: ENV["ASSISTANT_IAM_WORKSPACE_ID"],
+      export: false
+    )
+    assert((200..299).cover?(service_response.status))
+  end
+
+  def test_receive_error
+    service = IBMWatson::AssistantV1.new(
+      url: ENV["ASSISTANT_IAM_URL"],
+      version: "2018-02-16"
+    )
+    service.add_default_headers(
+      headers: {
+        "X-Watson-Learning-Opt-Out" => "1",
+        "X-Watson-Test" => "1"
+      }
+    )
+    error_received = false
+    begin
+      service._iam_api_key(iam_api_key: "bogus_api_key")
+      service.list_workspaces
+    rescue WatsonApiException => e
+      assert_equal("Provided API key could not be found", e.info["errorMessage"])
+      error_received = true
+    end
+    assert(error_received)
+  end
+end