turbot 0.0.2

data/spec/turbot/client/rendezvous_spec.rb

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+require "spec_helper"
+require "turbot/client/rendezvous"
+require "support/openssl_mock_helper"
+
+describe Turbot::Client, "rendezvous" do
+  before do
+    @rendezvous = Turbot::Client::Rendezvous.new({
+      :rendezvous_url => "https://turbot.local:1234/secret",
+      :output => $stdout
+    })
+  end
+  context "fixup" do
+    it "null" do
+      @rendezvous.send(:fixup, nil).should be_nil
+    end
+    it "an empty string" do
+      @rendezvous.send(:fixup, "").should eq ""
+    end
+    it "hash" do
+      @rendezvous.send(:fixup, { :x => :y }).should eq({ :x => :y })
+    end
+    it "default English UTF-8 data" do
+      @rendezvous.send(:fixup, "turbot").should eq "turbot"
+    end
+    it "default Japanese UTF-8 encoded data" do
+      @rendezvous.send(:fixup, "愛しています").should eq "愛しています"
+    end
+    if RUBY_VERSION >= "1.9"
+      it "ISO-8859-1 force-encoded data" do
+        @rendezvous.send(:fixup, "Хероку".force_encoding("ISO-8859-1")).should eq "Хероку".force_encoding("UTF-8")
+      end
+    end
+  end
+  context "with mock ssl" do
+    before :each do
+      mock_openssl
+      @ssl_socket_mock.should_receive(:puts).with("secret")
+      @ssl_socket_mock.should_receive(:readline).and_return(nil)
+    end
+    it "should connect to host:post" do
+      TCPSocket.should_receive(:open).with("turbot.local", 1234).and_return(@tcp_socket_mock)
+      IO.stub(:select).and_return(nil)
+      @ssl_socket_mock.stub(:write)
+      @ssl_socket_mock.stub(:flush) { raise Timeout::Error }
+      lambda { @rendezvous.start }.should raise_error(Timeout::Error)
+    end
+    it "should callback on_connect" do
+      @rendezvous.on_connect do 
+        raise "on_connect"
+      end
+      TCPSocket.should_receive(:open).and_return(@tcp_socket_mock)
+      lambda { @rendezvous.start }.should raise_error("on_connect")
+    end
+    it "should fixup received data" do
+      TCPSocket.should_receive(:open).and_return(@tcp_socket_mock)
+      @ssl_socket_mock.should_receive(:readpartial).and_return("The quick brown fox jumps over the lazy dog")
+      @rendezvous.stub(:fixup) { |data| raise "received: #{data}" }
+      lambda { @rendezvous.start }.should raise_error("received: The quick brown fox jumps over the lazy dog")
+    end
+  end
+end

data/spec/turbot/client/ssl_endpoint_spec.rb

@@ -0,0 +1,48 @@
+require "spec_helper"
+require "turbot/client/ssl_endpoint"
+
+describe Turbot::Client, "ssl endpoints" do
+  before do
+    @client = Turbot::Client.new(nil, nil)
+  end
+
+  it "adds an ssl endpoint" do
+    stub_request(:post, "https://api.turbot.com/bots/example/ssl-endpoints").
+      with(:body => { :accept => "json", :pem => "pem content", :key => "key content" }).
+      to_return(:body => %{ {"cname": "tokyo-1050" } })
+    @client.ssl_endpoint_add("example", "pem content", "key content").should == { "cname" => "tokyo-1050" }
+  end
+
+  it "gets info on an ssl endpoint" do
+    stub_request(:get, "https://api.turbot.com/bots/example/ssl-endpoints/tokyo-1050").
+      to_return(:body => %{ {"cname": "tokyo-1050" } })
+    @client.ssl_endpoint_info("example", "tokyo-1050").should == { "cname" => "tokyo-1050" }
+  end
+
+  it "lists ssl endpoints for an bot" do
+    stub_request(:get, "https://api.turbot.com/bots/example/ssl-endpoints").
+      to_return(:body => %{ [{"cname": "tokyo-1050" }, {"cname": "tokyo-1051" }] })
+    @client.ssl_endpoint_list("example").should == [
+      { "cname" => "tokyo-1050" },
+      { "cname" => "tokyo-1051" },
+    ]
+  end
+
+  it "removes an ssl endpoint" do
+    stub_request(:delete, "https://api.turbot.com/bots/example/ssl-endpoints/tokyo-1050")
+    @client.ssl_endpoint_remove("example", "tokyo-1050")
+  end
+
+  it "rolls back an ssl endpoint" do
+    stub_request(:post, "https://api.turbot.com/bots/example/ssl-endpoints/tokyo-1050/rollback").
+      to_return(:body => %{ {"cname": "tokyo-1050" } })
+    @client.ssl_endpoint_rollback("example", "tokyo-1050").should == { "cname" => "tokyo-1050" }
+  end
+
+  it "updates an ssl endpoint" do
+    stub_request(:put, "https://api.turbot.com/bots/example/ssl-endpoints/tokyo-1050").
+      with(:body => { :accept => "json", :pem => "pem content", :key => "key content" }).
+      to_return(:body => %{ {"cname": "tokyo-1050" } })
+    @client.ssl_endpoint_update("example", "tokyo-1050", "pem content", "key content").should == { "cname" => "tokyo-1050" }
+  end
+end

data/spec/turbot/client/turbot_postgresql_spec.rb

@@ -0,0 +1,71 @@
+require "spec_helper"
+require "turbot/client/turbot_postgresql"
+require "digest"
+
+describe Turbot::Client::TurbotPostgresql do
+  include Turbot::Helpers
+
+  before do
+    Turbot::Auth.stub :user => 'user@example.com', :password => 'apitoken'
+  end
+
+  let(:attachment) { double('attachment', :resource_name => 'something-something-42', :starter_plan? => false) }
+  let(:client)  { Turbot::Client::TurbotPostgresql.new(attachment) }
+
+  describe 'api choosing' do
+    it "sends an ingress request to the client for production plans" do
+      attachment.stub! :starter_plan? => false
+      host = 'postgres-api.turbot.com'
+      url  = "https://user@example.com:apitoken@#{host}/client/v11/databases/#{attachment.resource_name}/ingress"
+
+      stub_request(:put, url).to_return(
+        :body => json_encode({"message" => "ok"}),
+        :status => 200
+      )
+
+      client.ingress
+
+      a_request(:put, url).should have_been_made.once
+    end
+
+    it "sends an ingress request to the client for production plans" do
+      attachment.stub! :starter_plan? => true
+      host = 'postgres-starter-api.turbot.com'
+      url  = "https://user@example.com:apitoken@#{host}/client/v11/databases/#{attachment.resource_name}/ingress"
+
+      stub_request(:put, url).to_return(
+        :body => json_encode({"message" => "ok"}),
+        :status => 200
+      )
+
+      client.ingress
+
+      a_request(:put, url).should have_been_made.once
+    end
+  end
+
+  describe '#get_database' do
+    let(:url) { "https://user@example.com:apitoken@postgres-api.turbot.com/client/v11/databases/#{attachment.resource_name}" }
+
+    it 'works without the extended option' do
+      stub_request(:get, url).to_return :body => '{}'
+      client.get_database
+      a_request(:get, url).should have_been_made.once
+    end
+
+    it 'works with the extended option' do
+      url2 = url + '?extended=true'
+      stub_request(:get, url2).to_return :body => '{}'
+      client.get_database(true)
+      a_request(:get, url2).should have_been_made.once
+    end
+
+    it "retries on error, then raises" do
+      stub_request(:get, url).to_return(:body => "error", :status => 500)
+      client.stub(:sleep)
+      lambda { client.get_database }.should raise_error RestClient::InternalServerError
+      a_request(:get, url).should have_been_made.times(4)
+    end
+  end
+
+end