lib/td/logger/agent/rails.rb in td-logger-0.3.19 vs lib/td/logger/agent/rails.rb in td-logger-0.3.20

@@ -17,10 +17,12 @@
       warn 'Disabling Treasure Data event logger.'
       ::TreasureData::Logger.open_null
       return false
     end
 
-    if c.agent_mode?
+    if c.test_mode?
+      ::TreasureData::Logger.open_test
+    elsif c.agent_mode?
       ::TreasureData::Logger.open_agent(c.tag, :host=>c.agent_host, :port=>c.agent_port, :debug=>c.debug_mode)
     else
       ::TreasureData::Logger.open(c.database, :apikey=>c.apikey, :auto_create_table=>c.auto_create_table, :debug=>c.debug_mode)
     end