td_logger_spec.rb in td-logger-0.3.24

- old
+ new

@@ -2,45 +2,67 @@
 require 'spec_helper'
 
 describe TreasureData::Logger::TreasureDataLogger do
   context 'init' do
     it 'with apikey' do
-      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_1')
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1')
       td.instance_variable_get(:@client).api.apikey.should == 'test_1'
       td.instance_variable_get(:@client).api.instance_variable_get(:@ssl).should be_false
     end
 
     it 'with apikey and use_ssl' do
-      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_1', :use_ssl => true)
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1', :use_ssl => true)
       td.instance_variable_get(:@client).api.apikey.should == 'test_1'
       td.instance_variable_get(:@client).api.instance_variable_get(:@ssl).should be_true
     end
 
+    it 'with apikey and ssl' do
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1', :ssl => true)
+      td.instance_variable_get(:@client).api.apikey.should == 'test_1'
+      td.instance_variable_get(:@client).api.instance_variable_get(:@ssl).should be_true
+    end
+
+    it 'with apikey and HTTP endpoint' do
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1', :endpoint => "http://idontexi.st")
+      td.instance_variable_get(:@client).api.apikey.should == 'test_1'
+      td.instance_variable_get(:@client).api.instance_variable_get(:@host).should == "idontexi.st"
+      td.instance_variable_get(:@client).api.instance_variable_get(:@port).should == 80
+      td.instance_variable_get(:@client).api.instance_variable_get(:@ssl).should be_false
+    end
+
+    it 'with apikey and HTTPS endpoint' do
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1', :endpoint => "https://idontexi.st")
+      td.instance_variable_get(:@client).api.apikey.should == 'test_1'
+      td.instance_variable_get(:@client).api.instance_variable_get(:@host).should == "idontexi.st"
+      td.instance_variable_get(:@client).api.instance_variable_get(:@port).should == 443
+      td.instance_variable_get(:@client).api.instance_variable_get(:@ssl).should be_true
+    end
+
     it 'db config' do
-      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_1')
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_1')
       time = Time.now
-      td.should_receive(:add).with('db1', 'table1', {:foo=>:bar, :time=>time.to_i})
-      td.post_with_time('table1', {:foo=>:bar}, time)
+      td.should_receive(:add).with('db1', 'table1', {:foo => :bar, :time => time.to_i})
+      td.post_with_time('table1', {:foo => :bar}, time)
     end
 
     it 'fluent-logger-td compat' do
-      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_2')
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_2')
       time = Time.now
-      td.should_receive(:add).with('overwrite', 'table1', {:foo=>:bar, :time=>time.to_i})
-      td.post_with_time('overwrite.table1', {:foo=>:bar}, time)
+      td.should_receive(:add).with('overwrite', 'table1', {:foo => :bar, :time => time.to_i})
+      td.post_with_time('overwrite.table1', {:foo => :bar}, time)
     end
 
     ## TODO this causes real upload
     #it 'success' do
-    #  td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_3')
+    #  td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_3')
     #  td.post('valid', {}).should == true
     #end
   end
 
   context 'validate' do
     it 'validate table name' do
-      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey=>'test_4')
+      td = TreasureData::Logger::TreasureDataLogger.new('db1', :apikey => 'test_4')
       proc {
         td.post('invalid-name', {})
       }.should raise_error(RuntimeError)
       proc {
         td.post('', {})
@@ -49,10 +71,10 @@
         td.post('9', {})
       }.should raise_error(RuntimeError)
     end
 
     it 'validate database name' do
-      td = TreasureData::Logger::TreasureDataLogger.new('invalid-db-name', :apikey=>'test_5')
+      td = TreasureData::Logger::TreasureDataLogger.new('invalid-db-name', :apikey => 'test_5')
       proc {
         td.post('table', {})
       }.should raise_error(RuntimeError)
     end
   end