Added support for sentiment analysis endpoint

Smithaaaa · Jul 6, 2015 · 0118287 · 0118287
1 parent 211add5
commit 0118287
Show file tree

Hide file tree

Showing 4 changed files with 122 additions and 0 deletions.
diff --git a/lib/dandelionapi.rb b/lib/dandelionapi.rb
@@ -6,6 +6,7 @@
 require "dandelionapi/entity_extraction"
 require "dandelionapi/text_similarity"
 require "dandelionapi/language_detection"
+require "dandelionapi/sentiment_analysis"
 
 module Dandelionapi
 

diff --git a/lib/dandelionapi/sentiment_analysis.rb b/lib/dandelionapi/sentiment_analysis.rb
@@ -0,0 +1,25 @@
+# encoding: UTF-8
+
+require "faraday"
+require "faraday_middleware"
+require "json"
+
+module Dandelionapi
+
+  class SentimentAnalysis < Base
+
+    ENDPOINT = "/datatxt/sent/v1"
+
+    attr_accessor :text, :url, :html, :html_fragment, :lang
+
+    def analyze(options)
+
+      raise MissingParameters.new("Please provide one of the following parameters: text, url, html or html_fragment") if ([:text, :url, :html, :html_fragment] & options.keys).empty?
+
+      params = options
+      call(ENDPOINT, params)
+    end
+
+  end
+
+end
diff --git a/spec/cassettes/sentiment_analysis.yml b/spec/cassettes/sentiment_analysis.yml
diff --git a/spec/sentiment_analysis_spec.rb b/spec/sentiment_analysis_spec.rb
@@ -0,0 +1,50 @@
+# encoding: UTF-8
+
+require 'spec_helper'
+
+vcr_options = { :cassette_name => "sentiment_analysis", :record => :new_episodes }
+
+describe Dandelionapi::SentimentAnalysis, vcr: vcr_options do
+
+  before(:each) do
+    Dandelionapi.configure do |c|
+      # account: Mario Rossi <datatxt@mailinator.com>
+      c.app_id = "e0bca290"
+      c.app_key = "2294c676b8698383764514cc219fad92"
+      c.endpoint = "https://api.dandelion.eu/"
+    end
+  end
+
+  it "initialize a new nex request" do
+    request = Dandelionapi::SentimentAnalysis.new
+    expect(request).to be_an_instance_of(Dandelionapi::SentimentAnalysis)
+  end
+
+  it "make a request to sent using italian plain text" do
+    element = Dandelionapi::SentimentAnalysis.new
+    response = element.analyze(
+      text: "Mio padre che mi spinge a mangiare e guai se non finisco mio padre che vuol farmi guidare mi frena con il fischio"
+    )
+    puts response.inspect
+    expect(response).not_to be_empty
+    expect(response["sentiment"]).not_to be_empty
+    expect(response["sentiment"]["score"]).to be >= -1.0
+    expect(response["sentiment"]["score"]).to be <= 1.0
+    expect(["positive", "neutral", "negative"]).to include(response["sentiment"]["type"])
+  end
+
+  it "raise BadResponse exception on wrong config parameters" do
+    Dandelionapi.configure do |c|
+      c.app_id = "bad-app-id"
+      c.app_key = "bad-app-key"
+      c.endpoint = "not-an-url-endpoint"
+    end
+    element = Dandelionapi::SentimentAnalysis.new
+    expect { element.analyze(text: "test") }.to raise_error(Dandelionapi::BadResponse)
+  end
+
+  it "raise MissingParameters exception when required params are missing" do
+    element = Dandelionapi::SentimentAnalysis.new
+    expect { element.analyze(other: "test") }.to raise_error(Dandelionapi::MissingParameters)
+  end
+end