|
4 | 4 | require "java" |
5 | 5 |
|
6 | 6 | describe LogStash::Outputs::Jdbc do |
7 | | - def fetch_log_table_rowcount |
8 | | - # sleep for a second to let the flush happen |
9 | | - sleep 1 |
10 | | - |
11 | | - stmt = @sql.createStatement() |
12 | | - rs = stmt.executeQuery("select count(*) as total from log") |
13 | | - count = 0 |
14 | | - while rs.next() |
15 | | - count = rs.getInt("total") |
16 | | - end |
17 | | - stmt.close() |
18 | 7 |
|
19 | | - return count |
| 8 | + let(:derby_settings) do |
| 9 | + { |
| 10 | + "driver_class" => "org.apache.derby.jdbc.EmbeddedDriver", |
| 11 | + "connection_string" => "jdbc:derby:memory:testdb;create=true", |
| 12 | + "driver_jar_path" => ENV['JDBC_DERBY_JAR'], |
| 13 | + # Grumble. Grumble. |
| 14 | + # Derby doesn't like 'T' in timestamps as of current writing, so for now |
| 15 | + # we'll just use CURRENT_TIMESTAMP as opposed to the event @timestamp |
| 16 | + "statement" => [ "insert into log (created_at, message) values(CURRENT_TIMESTAMP, ?)", "message" ] |
| 17 | + } |
20 | 18 | end |
21 | 19 |
|
22 | | - let(:base_settings) { { |
23 | | - "driver_jar_path" => @driver_jar_path, |
24 | | - "connection_string" => @test_connection_string, |
25 | | - "username" => ENV['SQL_USERNAME'], |
26 | | - "password" => ENV['SQL_PASSWORD'], |
27 | | - "statement" => [ "insert into log (message) values(?)", "message" ], |
28 | | - "max_pool_size" => 1, |
29 | | - "flush_size" => 1, |
30 | | - "max_flush_exceptions" => 1 |
31 | | - } } |
32 | | - let(:test_settings) { {} } |
33 | | - let(:plugin) { LogStash::Outputs::Jdbc.new(base_settings.merge(test_settings)) } |
34 | | - let(:event_fields) { { "message" => "This is a message!" } } |
35 | | - let(:event) { LogStash::Event.new(event_fields) } |
36 | | - |
37 | | - before(:all) do |
38 | | - @driver_jar_path = File.absolute_path(ENV['SQL_JAR']) |
39 | | - @test_db_path = File.join(Stud::Temporary.directory, "test.db") |
40 | | - @test_connection_string = "jdbc:sqlite:#{@test_db_path}" |
41 | | - |
42 | | - require @driver_jar_path |
43 | | - |
44 | | - @sql = java.sql.DriverManager.get_connection(@test_connection_string, ENV['SQL_USERNAME'].to_s, ENV['SQL_PASSWORD'].to_s) |
45 | | - stmt = @sql.createStatement() |
46 | | - stmt.executeUpdate("CREATE table log (host text, timestamp datetime, message text);") |
47 | | - stmt.close() |
48 | | - end |
| 20 | + context 'rspec setup' do |
49 | 21 |
|
50 | | - before(:each) do |
51 | | - stmt = @sql.createStatement() |
52 | | - stmt.executeUpdate("delete from log") |
53 | | - stmt.close() |
| 22 | + it 'ensure derby is available' do |
| 23 | + j = ENV['JDBC_DERBY_JAR'] |
| 24 | + expect(j).not_to be_nil, "JDBC_DERBY_JAR not defined, required to run tests" |
| 25 | + expect(File.exists?(j)).to eq(true), "JDBC_DERBY_JAR defined, but not valid" |
| 26 | + end |
| 27 | + |
54 | 28 | end |
55 | 29 |
|
56 | | - after(:all) do |
57 | | - File.unlink(@test_db_path) |
58 | | - Dir.rmdir(File.dirname(@test_db_path)) |
59 | | - end |
| 30 | + context 'when initializing' do |
60 | 31 |
|
61 | | - describe "safe statement" do |
62 | | - it "should register without errors" do |
63 | | - expect { plugin.register }.to_not raise_error |
| 32 | + it 'shouldn\'t register without a config' do |
| 33 | + expect { |
| 34 | + LogStash::Plugin.lookup("output", "jdbc").new() |
| 35 | + }.to raise_error(LogStash::ConfigurationError) |
64 | 36 | end |
65 | 37 |
|
66 | | - it "receive event, without error" do |
67 | | - plugin.register |
68 | | - expect { plugin.receive(event) }.to_not raise_error |
| 38 | + it 'shouldn\'t register with a missing jar file' do |
| 39 | + derby_settings['driver_jar_path'] = nil |
| 40 | + plugin = LogStash::Plugin.lookup("output", "jdbc").new(derby_settings) |
| 41 | + expect { plugin.register }.to raise_error |
| 42 | + end |
69 | 43 |
|
70 | | - expect(fetch_log_table_rowcount).to eq(1) |
| 44 | + it 'shouldn\'t register with a missing jar file' do |
| 45 | + derby_settings['connection_string'] = nil |
| 46 | + plugin = LogStash::Plugin.lookup("output", "jdbc").new(derby_settings) |
| 47 | + expect { plugin.register }.to raise_error |
71 | 48 | end |
72 | 49 |
|
73 | 50 | end |
74 | 51 |
|
75 | | - describe "unsafe statement" do |
76 | | - let(:event_fields) { |
77 | | - { "message" => "This is a message!", "table" => "log" } |
78 | | - } |
79 | | - let(:test_settings) { { |
80 | | - "statement" => [ "insert into %{table} (message) values(?)", "message" ], |
81 | | - "unsafe_statement" => true |
82 | | - } } |
83 | | - |
84 | | - it "should register without errors" do |
85 | | - expect { plugin.register }.to_not raise_error |
| 52 | + context 'when outputting messages' do |
| 53 | + |
| 54 | + let(:event_fields) do |
| 55 | + { message: 'test-message' } |
86 | 56 | end |
| 57 | + let(:event) { LogStash::Event.new(event_fields) } |
| 58 | + let(:plugin) { |
| 59 | + # Setup plugin |
| 60 | + output = LogStash::Plugin.lookup("output", "jdbc").new(derby_settings) |
| 61 | + output.register |
| 62 | + if ENV['JDBC_DEBUG'] == '1' |
| 63 | + output.logger.subscribe(STDOUT) |
| 64 | + end |
| 65 | + |
| 66 | + # Setup table |
| 67 | + c = output.instance_variable_get(:@pool).getConnection() |
| 68 | + stmt = c.createStatement() |
| 69 | + stmt.executeUpdate("CREATE table log (created_at timestamp, message varchar(512))") |
| 70 | + stmt.close() |
| 71 | + c.close() |
| 72 | + |
| 73 | + output |
| 74 | + } |
87 | 75 |
|
88 | | - it "receive event, without error" do |
89 | | - plugin.register |
90 | | - plugin.receive(event) |
91 | | - expect(fetch_log_table_rowcount).to eq(1) |
| 76 | + it 'should save a event' do |
| 77 | + expect { plugin.receive(event) }.to_not raise_error |
| 78 | + |
| 79 | + # Wait for 1 second, for the buffer to flush |
| 80 | + sleep 1 |
| 81 | + |
| 82 | + c = plugin.instance_variable_get(:@pool).getConnection() |
| 83 | + stmt = c.createStatement() |
| 84 | + rs = stmt.executeQuery("select count(*) as total from log") |
| 85 | + count = 0 |
| 86 | + while rs.next() |
| 87 | + count = rs.getInt("total") |
| 88 | + end |
| 89 | + stmt.close() |
| 90 | + c.close() |
| 91 | + expect(count).to be > 0 |
92 | 92 | end |
93 | | - |
| 93 | + |
94 | 94 | end |
| 95 | + |
95 | 96 | end |
0 commit comments