presto-client-legacy 0.4.17

data/spec/statement_client_spec.rb

@@ -0,0 +1,183 @@
+require 'spec_helper'
+
+describe PrestoLegacy::Client::StatementClient do
+  let :options do
+    {
+      server: "localhost",
+      user: "frsyuki",
+      catalog: "native",
+      schema: "default",
+      time_zone: "US/Pacific",
+      language: "ja_JP",
+      properties: {"hello" => "world", "name"=>"value"},
+      debug: true,
+    }
+  end
+
+  let :query do
+    "select * from sys.node"
+  end
+
+  let :response_json do
+    {
+      id: "queryid",
+      stats: {}
+    }
+  end
+
+  it "sets headers" do
+    stub_request(:post, "localhost/v1/statement").
+      with(body: query,
+           headers: {
+              "User-Agent" => "presto-ruby/#{VERSION}",
+              "X-Presto-Catalog" => options[:catalog],
+              "X-Presto-Schema" => options[:schema],
+              "X-Presto-User" => options[:user],
+              "X-Presto-Language" => options[:language],
+              "X-Presto-Time-Zone" => options[:time_zone],
+              "X-Presto-Session" => options[:properties].map {|k,v| "#{k}=#{v}"}.join("\r\nX-Presto-Session: ")
+    }).to_return(body: response_json.to_json)
+
+    faraday = Faraday.new(url: "http://localhost")
+    StatementClient.new(faraday, query, options)
+  end
+
+  let :response_json2 do
+    {
+      id: "queryid",
+      nextUri: 'http://localhost/v1/next_uri',
+      stats: {}
+    }
+  end
+
+  it "sets headers" do
+    retry_p = false
+    stub_request(:post, "localhost/v1/statement").
+      with(body: query,
+           headers: {
+              "User-Agent" => "presto-ruby/#{VERSION}",
+              "X-Presto-Catalog" => options[:catalog],
+              "X-Presto-Schema" => options[:schema],
+              "X-Presto-User" => options[:user],
+              "X-Presto-Language" => options[:language],
+              "X-Presto-Time-Zone" => options[:time_zone],
+              "X-Presto-Session" => options[:properties].map {|k,v| "#{k}=#{v}"}.join("\r\nX-Presto-Session: ")
+    }).to_return(body: response_json2.to_json)
+
+    stub_request(:get, "localhost/v1/next_uri").
+      with(headers: {
+              "User-Agent" => "presto-ruby/#{VERSION}",
+              "X-Presto-Catalog" => options[:catalog],
+              "X-Presto-Schema" => options[:schema],
+              "X-Presto-User" => options[:user],
+              "X-Presto-Language" => options[:language],
+              "X-Presto-Time-Zone" => options[:time_zone],
+              "X-Presto-Session" => options[:properties].map {|k,v| "#{k}=#{v}"}.join("\r\nX-Presto-Session: ")
+    }).to_return(body: lambda{|req|if retry_p; response_json.to_json; else; retry_p=true; raise Timeout::Error.new("execution expired"); end })
+
+    faraday = Faraday.new(url: "http://localhost")
+    sc = StatementClient.new(faraday, query, options.merge(http_open_timeout: 1))
+    sc.has_next?.should be_true
+    sc.advance.should be_true
+    retry_p.should be_true
+  end
+
+  it "decodes DeleteHandle" do
+    dh = Models::DeleteHandle.decode({
+      "handle" => {
+        "connectorId" => "c1",
+        "connectorHandle" => {},
+      }
+    })
+    dh.handle.should be_a_kind_of Models::TableHandle
+    dh.handle.connector_id.should == "c1"
+    dh.handle.connector_handle.should == {}
+  end
+
+  it "validates models" do
+    lambda do
+      Models::DeleteHandle.decode({
+        "handle" => "invalid"
+      })
+    end.should raise_error(TypeError, /String to Hash/)
+  end
+
+  describe "ssl" do
+    it "is disabled by default" do
+      f = Query.__send__(:faraday_client, {
+        server: "localhost",
+      })
+      f.url_prefix.to_s.should == "http://localhost/"
+    end
+
+    it "is enabled with ssl: true" do
+      f = Query.__send__(:faraday_client, {
+        server: "localhost",
+        ssl: true,
+      })
+      f.url_prefix.to_s.should == "https://localhost/"
+      f.ssl.verify?.should == true
+    end
+
+    it "is enabled with ssl: {verify: false}" do
+      f = Query.__send__(:faraday_client, {
+        server: "localhost",
+        ssl: {verify: false}
+      })
+      f.url_prefix.to_s.should == "https://localhost/"
+      f.ssl.verify?.should == false
+    end
+
+    it "rejects invalid ssl: verify: object" do
+      lambda do
+        f = Query.__send__(:faraday_client, {
+          server: "localhost",
+          ssl: {verify: "??"}
+        })
+      end.should raise_error(ArgumentError, /String/)
+    end
+
+    it "is enabled with ssl: Hash" do
+      require 'openssl'
+
+      ssl = {
+        ca_file: "/path/to/dummy.pem",
+        ca_path: "/path/to/pemdir",
+        cert_store: OpenSSL::X509::Store.new,
+        client_cert: OpenSSL::X509::Certificate.new,
+        client_key: OpenSSL::PKey::DSA.new,
+      }
+
+      f = Query.__send__(:faraday_client, {
+        server: "localhost",
+        ssl: ssl,
+      })
+
+      f.url_prefix.to_s.should == "https://localhost/"
+      f.ssl.verify?.should == true
+      f.ssl.ca_file.should == ssl[:ca_file]
+      f.ssl.ca_path.should == ssl[:ca_path]
+      f.ssl.cert_store.should == ssl[:cert_store]
+      f.ssl.client_cert.should == ssl[:client_cert]
+      f.ssl.client_key.should == ssl[:client_key]
+    end
+
+    it "rejects an invalid string" do
+      lambda do
+        Query.__send__(:faraday_client, {
+          server: "localhost",
+          ssl: '??',
+        })
+      end.should raise_error(ArgumentError, /String/)
+    end
+
+    it "rejects an integer" do
+      lambda do
+        Query.__send__(:faraday_client, {
+          server: "localhost",
+          ssl: 3,
+        })
+      end.should raise_error(ArgumentError, /:ssl/)
+    end
+  end
+end

metadata

@@ -0,0 +1,160 @@
+--- !ruby/object:Gem::Specification
+name: presto-client-legacy
+version: !ruby/object:Gem::Version
+  version: 0.4.17
+platform: ruby
+authors:
+- Sadayuki Furuhash
+- Kai Sasaki
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-08-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.8
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.8
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.16.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.16.1
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+description: Presto client library for legacy presto version
+email:
+- sf@treasure-data.com
+- lewuathe@me.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- ChangeLog
+- Gemfile
+- README.md
+- Rakefile
+- lib/presto-client-legacy.rb
+- lib/presto/client.rb
+- lib/presto/client/client.rb
+- lib/presto/client/errors.rb
+- lib/presto/client/models.rb
+- lib/presto/client/query.rb
+- lib/presto/client/statement_client.rb
+- lib/presto/client/version.rb
+- modelgen/modelgen.rb
+- modelgen/models.rb
+- modelgen/presto_models.rb
+- presto-client-legacy.gemspec
+- spec/client_spec.rb
+- spec/spec_helper.rb
+- spec/statement_client_spec.rb
+homepage: https://github.com/Lewuathe/presto-client-ruby
+licenses:
+- Apache-2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Presto client library for legacy presto version (before 0.151)
+test_files:
+- spec/client_spec.rb
+- spec/spec_helper.rb
+- spec/statement_client_spec.rb