96 lines
2.6 KiB
Ruby
96 lines
2.6 KiB
Ruby
require "logstash/devutils/rspec/spec_helper"
|
|
require "logstash/outputs/jdbc"
|
|
require "stud/temporary"
|
|
require "java"
|
|
|
|
describe LogStash::Outputs::Jdbc do
|
|
def fetch_log_table_rowcount
|
|
# sleep for a second to let the flush happen
|
|
sleep 1
|
|
|
|
stmt = @sql.createStatement()
|
|
rs = stmt.executeQuery("select count(*) as total from log")
|
|
count = 0
|
|
while rs.next()
|
|
count = rs.getInt("total")
|
|
end
|
|
stmt.close()
|
|
|
|
return count
|
|
end
|
|
|
|
let(:base_settings) { {
|
|
"driver_jar_path" => @driver_jar_path,
|
|
"connection_string" => @test_connection_string,
|
|
"username" => ENV['SQL_USERNAME'],
|
|
"password" => ENV['SQL_PASSWORD'],
|
|
"statement" => [ "insert into log (message) values(?)", "message" ],
|
|
"max_pool_size" => 1,
|
|
"flush_size" => 1,
|
|
"max_flush_exceptions" => 1
|
|
} }
|
|
let(:test_settings) { {} }
|
|
let(:plugin) { LogStash::Outputs::Jdbc.new(base_settings.merge(test_settings)) }
|
|
let(:event_fields) { { "message" => "This is a message!" } }
|
|
let(:event) { LogStash::Event.new(event_fields) }
|
|
|
|
before(:all) do
|
|
@driver_jar_path = File.absolute_path(ENV['SQL_JAR'])
|
|
@test_db_path = File.join(Stud::Temporary.directory, "test.db")
|
|
@test_connection_string = "jdbc:sqlite:#{@test_db_path}"
|
|
|
|
require @driver_jar_path
|
|
|
|
@sql = java.sql.DriverManager.get_connection(@test_connection_string, ENV['SQL_USERNAME'].to_s, ENV['SQL_PASSWORD'].to_s)
|
|
stmt = @sql.createStatement()
|
|
stmt.executeUpdate("CREATE table log (host text, timestamp datetime, message text);")
|
|
stmt.close()
|
|
end
|
|
|
|
before(:each) do
|
|
stmt = @sql.createStatement()
|
|
stmt.executeUpdate("delete from log")
|
|
stmt.close()
|
|
end
|
|
|
|
after(:all) do
|
|
File.unlink(@test_db_path)
|
|
Dir.rmdir(File.dirname(@test_db_path))
|
|
end
|
|
|
|
describe "safe statement" do
|
|
it "should register without errors" do
|
|
expect { plugin.register }.to_not raise_error
|
|
end
|
|
|
|
it "receive event, without error" do
|
|
plugin.register
|
|
expect { plugin.receive(event) }.to_not raise_error
|
|
|
|
expect(fetch_log_table_rowcount).to eq(1)
|
|
end
|
|
|
|
end
|
|
|
|
describe "unsafe statement" do
|
|
let(:event_fields) {
|
|
{ "message" => "This is a message!", "table" => "log" }
|
|
}
|
|
let(:test_settings) { {
|
|
"statement" => [ "insert into %{table} (message) values(?)", "message" ],
|
|
"unsafe_statement" => true
|
|
} }
|
|
|
|
it "should register without errors" do
|
|
expect { plugin.register }.to_not raise_error
|
|
end
|
|
|
|
it "receive event, without error" do
|
|
plugin.register
|
|
plugin.receive(event)
|
|
expect(fetch_log_table_rowcount).to eq(1)
|
|
end
|
|
|
|
end
|
|
end
|