# encoding: utf-8 require "logstash/devutils/rspec/spec_helper" require "logstash/filters/kv" describe LogStash::Filters::KV do describe "defaults" do # The logstash config goes here. # At this time, only filters are supported. config <<-CONFIG filter { kv { } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world' bracketsone=(hello world) bracketstwo=[hello world] bracketsthree=" do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" insist { subject.get("bracketsone") } == "hello world" insist { subject.get("bracketstwo") } == "hello world" insist { subject.get("bracketsthree") } == "hello world" end end describe "test transforming keys to uppercase and values to lowercase" do config <<-CONFIG filter { kv { transform_key => "uppercase" transform_value => "lowercase" } } CONFIG sample "hello = world Foo =Bar BAZ= FIZZ doublequoteD = \"hellO worlD\" Singlequoted= 'Hello World' brAckets =(hello World)" do insist { subject.get("HELLO") } == "world" insist { subject.get("FOO") } == "bar" insist { subject.get("BAZ") } == "fizz" insist { subject.get("DOUBLEQUOTED") } == "hello world" insist { subject.get("SINGLEQUOTED") } == "hello world" insist { subject.get("BRACKETS") } == "hello world" end end describe "test transforming keys to lowercase and values to uppercase" do config <<-CONFIG filter { kv { transform_key => "lowercase" transform_value => "uppercase" } } CONFIG sample "Hello = World fOo =bar baz= FIZZ DOUBLEQUOTED = \"hellO worlD\" singlequoted= 'hEllo wOrld' brackets =(HELLO world)" do insist { subject.get("hello") } == "WORLD" insist { subject.get("foo") } == "BAR" insist { subject.get("baz") } == "FIZZ" insist { subject.get("doublequoted") } == "HELLO WORLD" insist { subject.get("singlequoted") } == "HELLO WORLD" insist { subject.get("brackets") } == "HELLO WORLD" end end describe "test transforming keys and values to capitals" do config <<-CONFIG filter { kv { transform_key => "capitalize" transform_value => "capitalize" } } CONFIG sample "Hello = World fOo =bar baz= FIZZ DOUBLEQUOTED = \"hellO worlD\" singlequoted= 'hEllo wOrld' brackets =(HELLO world)" do insist { subject.get("Hello") } == "World" insist { subject.get("Foo") } == "Bar" insist { subject.get("Baz") } == "Fizz" insist { subject.get("Doublequoted") } == "Hello world" insist { subject.get("Singlequoted") } == "Hello world" insist { subject.get("Brackets") } == "Hello world" end end describe "test spaces attached to the field_split" do config <<-CONFIG filter { kv { } } CONFIG sample "hello = world foo =bar baz= fizz doublequoted = \"hello world\" singlequoted= 'hello world' brackets =(hello world)" do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" insist { subject.get("brackets") } == "hello world" end end describe "LOGSTASH-624: allow escaped space in key or value " do config <<-CONFIG filter { kv { value_split => ':' } } CONFIG sample 'IKE:=Quick\ Mode\ completion IKE\ IDs:=subnet:\ x.x.x.x\ (mask=\ 255.255.255.254)\ and\ host:\ y.y.y.y' do insist { subject.get("IKE") } == '=Quick\ Mode\ completion' insist { subject.get('IKE\ IDs') } == '=subnet:\ x.x.x.x\ (mask=\ 255.255.255.254)\ and\ host:\ y.y.y.y' end end describe "test value_split" do context "using an alternate splitter" do config <<-CONFIG filter { kv { value_split => ':' } } CONFIG sample "hello:=world foo:bar baz=:fizz doublequoted:\"hello world\" singlequoted:'hello world' brackets:(hello world)" do insist { subject.get("hello") } == "=world" insist { subject.get("foo") } == "bar" insist { subject.get("baz=") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" insist { subject.get("brackets") } == "hello world" end end end # these specs are quite implementation specific by testing on the private method # has_value_splitter? - this is what I figured would help fixing the short circuit # broken code that was previously in place describe "short circuit" do subject do plugin = LogStash::Filters::KV.new(options) plugin.register plugin end let(:data) { {"message" => message} } let(:event) { LogStash::Event.new(data) } context "plain message" do let(:options) { {} } context "without splitter" do let(:message) { "foo:bar" } it "should short circuit" do expect(subject.send(:has_value_splitter?, message)).to be_falsey expect(subject).to receive(:has_value_splitter?).with(message).once.and_return(false) subject.filter(event) end end context "with splitter" do let(:message) { "foo=bar" } it "should not short circuit" do expect(subject.send(:has_value_splitter?, message)).to be_truthy expect(subject).to receive(:has_value_splitter?).with(message).once.and_return(true) subject.filter(event) end end end context "recursive message" do context "without inner splitter" do let(:inner) { "bar" } let(:message) { "foo=#{inner}" } let(:options) { {"recursive" => "true"} } it "should extract kv" do subject.filter(event) expect(event.get("foo")).to eq(inner) end it "should short circuit" do expect(subject.send(:has_value_splitter?, message)).to be_truthy expect(subject.send(:has_value_splitter?, inner)).to be_falsey expect(subject).to receive(:has_value_splitter?).with(message).once.and_return(true) expect(subject).to receive(:has_value_splitter?).with(inner).once.and_return(false) subject.filter(event) end end context "with inner splitter" do let(:foo_val) { "1" } let(:baz_val) { "2" } let(:inner) { "baz=#{baz_val}" } let(:message) { "foo=#{foo_val} bar=(#{inner})" } # foo=1 bar=(baz=2) let(:options) { {"recursive" => "true"} } it "should extract kv" do subject.filter(event) expect(event.get("foo")).to eq(foo_val) expect(event.get("[bar][baz]")).to eq(baz_val) end it "should short circuit" do expect(subject.send(:has_value_splitter?, message)).to be_truthy expect(subject.send(:has_value_splitter?, foo_val)).to be_falsey expect(subject.send(:has_value_splitter?, inner)).to be_truthy expect(subject.send(:has_value_splitter?, baz_val)).to be_falsey expect(subject).to receive(:has_value_splitter?).with(message).once.and_return(true) expect(subject).to receive(:has_value_splitter?).with(foo_val).once.and_return(false) expect(subject).to receive(:has_value_splitter?).with(inner).once.and_return(true) expect(subject).to receive(:has_value_splitter?).with(baz_val).once.and_return(false) subject.filter(event) end end end end describe "test field_split" do config <<-CONFIG filter { kv { field_split => '?&' } } CONFIG sample "?hello=world&foo=bar&baz=fizz&doublequoted=\"hello world\"&singlequoted='hello world'&ignoreme&foo12=bar12" do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" insist { subject.get("foo12") } == "bar12" end end describe "test include_brackets is false" do config <<-CONFIG filter { kv { include_brackets => "false" } } CONFIG sample "bracketsone=(hello world) bracketstwo=[hello world]" do insist { subject.get("bracketsone") } == "(hello" insist { subject.get("bracketstwo") } == "[hello" end end describe "test recursive" do config <<-CONFIG filter { kv { recursive => 'true' } } CONFIG sample 'IKE="Quick Mode completion" IKE\ IDs = (subnet= x.x.x.x mask= 255.255.255.254 and host=y.y.y.y)' do insist { subject.get("IKE") } == 'Quick Mode completion' insist { subject.get('IKE\ IDs')['subnet'] } == 'x.x.x.x' insist { subject.get('IKE\ IDs')['mask'] } == '255.255.255.254' insist { subject.get('IKE\ IDs')['host'] } == 'y.y.y.y' end end describe "delimited fields should override space default (reported by LOGSTASH-733)" do config <<-CONFIG filter { kv { field_split => "|" } } CONFIG sample "field1=test|field2=another test|field3=test3" do insist { subject.get("field1") } == "test" insist { subject.get("field2") } == "another test" insist { subject.get("field3") } == "test3" end end describe "test prefix" do config <<-CONFIG filter { kv { prefix => '__' } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("__hello") } == "world" insist { subject.get("__foo") } == "bar" insist { subject.get("__baz") } == "fizz" insist { subject.get("__doublequoted") } == "hello world" insist { subject.get("__singlequoted") } == "hello world" end end describe "speed test", :performance => true do count = 10000 + rand(3000) config <<-CONFIG input { generator { count => #{count} type => foo message => "hello=world bar='baz fizzle'" } } filter { kv { } } output { null { } } CONFIG start = Time.now agent do duration = (Time.now - start) puts "filters/kv rate: #{"%02.0f/sec" % (count / duration)}, elapsed: #{duration}s" end end describe "add_tag" do context "should activate when successful" do config <<-CONFIG filter { kv { add_tag => "hello" } } CONFIG sample "hello=world" do insist { subject.get("hello") } == "world" insist { subject.get("tags") }.include?("hello") end end context "should not activate when failing" do config <<-CONFIG filter { kv { add_tag => "hello" } } CONFIG sample "this is not key value" do insist { subject.get("tags") }.nil? end end end describe "add_field" do context "should activate when successful" do config <<-CONFIG filter { kv { add_field => [ "whoa", "fancypants" ] } } CONFIG sample "hello=world" do insist { subject.get("hello") } == "world" insist { subject.get("whoa") } == "fancypants" end end context "should not activate when failing" do config <<-CONFIG filter { kv { add_tag => "hello" } } CONFIG sample "this is not key value" do reject { subject.get("whoa") } == "fancypants" end end end #New tests describe "test target" do config <<-CONFIG filter { kv { target => 'kv' } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("kv")["hello"] } == "world" insist { subject.get("kv")["foo"] } == "bar" insist { subject.get("kv")["baz"] } == "fizz" insist { subject.get("kv")["doublequoted"] } == "hello world" insist { subject.get("kv")["singlequoted"] } == "hello world" insist {subject.get("kv").count } == 5 end end describe "test empty target" do config <<-CONFIG filter { kv { target => 'kv' } } CONFIG sample "hello:world:foo:bar:baz:fizz" do insist { subject.get("kv") } == nil end end describe "test data from specific sub source" do config <<-CONFIG filter { kv { source => "data" } } CONFIG sample("data" => "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'") do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" end end describe "test data from specific top source" do config <<-CONFIG filter { kv { source => "@data" } } CONFIG sample({"@data" => "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'"}) do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" end end describe "test data from specific sub source and target" do config <<-CONFIG filter { kv { source => "data" target => "kv" } } CONFIG sample("data" => "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'") do insist { subject.get("kv")["hello"] } == "world" insist { subject.get("kv")["foo"] } == "bar" insist { subject.get("kv")["baz"] } == "fizz" insist { subject.get("kv")["doublequoted"] } == "hello world" insist { subject.get("kv")["singlequoted"] } == "hello world" insist { subject.get("kv").count } == 5 end end describe "test data from nil sub source, should not issue a warning" do config <<-CONFIG filter { kv { source => "non-exisiting-field" target => "kv" } } CONFIG sample "" do insist { subject.get("non-exisiting-field") } == nil insist { subject.get("kv") } == nil end end describe "test include_keys" do config <<-CONFIG filter { kv { include_keys => [ "foo", "singlequoted" ] } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("foo") } == "bar" insist { subject.get("singlequoted") } == "hello world" end end describe "test exclude_keys" do config <<-CONFIG filter { kv { exclude_keys => [ "foo", "singlequoted" ] } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("hello") } == "world" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" end end describe "test include_keys with prefix" do config <<-CONFIG filter { kv { include_keys => [ "foo", "singlequoted" ] prefix => "__" } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("__foo") } == "bar" insist { subject.get("__singlequoted") } == "hello world" end end describe "test exclude_keys with prefix" do config <<-CONFIG filter { kv { exclude_keys => [ "foo", "singlequoted" ] prefix => "__" } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("__hello") } == "world" insist { subject.get("__baz") } == "fizz" insist { subject.get("__doublequoted") } == "hello world" end end describe "test include_keys with dynamic key" do config <<-CONFIG filter { kv { source => "data" include_keys => [ "%{key}"] } } CONFIG sample({"data" => "foo=bar baz=fizz", "key" => "foo"}) do insist { subject.get("foo") } == "bar" insist { subject.get("baz") } == nil end end describe "test exclude_keys with dynamic key" do config <<-CONFIG filter { kv { source => "data" exclude_keys => [ "%{key}"] } } CONFIG sample({"data" => "foo=bar baz=fizz", "key" => "foo"}) do insist { subject.get("foo") } == nil insist { subject.get("baz") } == "fizz" end end describe "test include_keys and exclude_keys" do config <<-CONFIG filter { kv { # This should exclude everything as a result of both settings. include_keys => [ "foo", "singlequoted" ] exclude_keys => [ "foo", "singlequoted" ] } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do %w(hello foo baz doublequoted singlequoted).each do |field| reject { subject }.include?(field) end end end describe "test default_keys" do config <<-CONFIG filter { kv { default_keys => [ "foo", "xxx", "goo", "yyy" ] } } CONFIG sample "hello=world foo=bar baz=fizz doublequoted=\"hello world\" singlequoted='hello world'" do insist { subject.get("hello") } == "world" insist { subject.get("foo") } == "bar" insist { subject.get("goo") } == "yyy" insist { subject.get("baz") } == "fizz" insist { subject.get("doublequoted") } == "hello world" insist { subject.get("singlequoted") } == "hello world" end end describe "overwriting a string field (often the source)" do config <<-CONFIG filter { kv { source => "happy" target => "happy" } } CONFIG sample("happy" => "foo=bar baz=fizz") do insist { subject.get("[happy][foo]") } == "bar" insist { subject.get("[happy][baz]") } == "fizz" end end describe "Removing duplicate key/value pairs" do config <<-CONFIG filter { kv { field_split => "&" source => "source" allow_duplicate_values => false } } CONFIG sample("source" => "foo=bar&foo=yeah&foo=yeah") do insist { subject.get("[foo]") } == ["bar", "yeah"] end end describe "Allow duplicate key/value pairs by default" do config <<-CONFIG filter { kv { field_split => "&" source => "source" } } CONFIG sample("source" => "foo=bar&foo=yeah&foo=yeah") do insist { subject.get("[foo]") } == ["bar", "yeah", "yeah"] end end describe "keys without values (reported in #22)" do subject do plugin = LogStash::Filters::KV.new(options) plugin.register plugin end let(:f1) { "AccountStatus" } let(:v1) { "4" } let(:f2) { "AdditionalInformation" } let(:f3) { "Code" } let(:f4) { "HttpStatusCode" } let(:f5) { "IsSuccess" } let(:v5) { "True" } let(:f6) { "Message" } let(:message) { "#{f1}: #{v1}\r\n#{f2}\r\n\r\n#{f3}: \r\n#{f4}: \r\n#{f5}: #{v5}\r\n#{f6}: \r\n" } let(:data) { {"message" => message} } let(:event) { LogStash::Event.new(data) } let(:options) { { "field_split" => "\r\n", "value_split" => " ", "trim_key" => ":" } } context "key and splitters with no value" do it "should ignore the incomplete key/value pairs" do subject.filter(event) expect(event.get(f1)).to eq(v1) expect(event.get(f5)).to eq(v5) expect(event.include?(f2)).to be false expect(event.include?(f3)).to be false expect(event.include?(f4)).to be false expect(event.include?(f6)).to be false end end end describe "trim_key/trim_value options : trim only leading and trailing spaces in keys/values (reported in #10)" do subject do plugin = LogStash::Filters::KV.new(options) plugin.register plugin end let(:message) { "key1= value1 with spaces | key2 with spaces =value2" } let(:data) { {"message" => message} } let(:event) { LogStash::Event.new(data) } let(:options) { { "field_split" => "\|", "value_split" => "=", "trim_value" => " ", "trim_key" => " " } } context "key and value with leading, trailing and middle spaces" do it "should trim only leading and trailing spaces" do subject.filter(event) expect(event.get("key1")).to eq("value1 with spaces") expect(event.get("key2 with spaces")).to eq("value2") end end end describe "remove_char_key/remove_char_value options : remove all characters in keys/values whatever their position" do subject do plugin = LogStash::Filters::KV.new(options) plugin.register plugin end let(:message) { "key1= value1 with spaces | key2 with spaces =value2" } let(:data) { {"message" => message} } let(:event) { LogStash::Event.new(data) } let(:options) { { "field_split" => "\|", "value_split" => "=", "remove_char_value" => " ", "remove_char_key" => " " } } context "key and value with leading, trailing and middle spaces" do it "should remove all spaces" do subject.filter(event) expect(event.get("key1")).to eq("value1withspaces") expect(event.get("key2withspaces")).to eq("value2") end end end describe "an empty value_split option should be reported" do config <<-CONFIG filter { kv { value_split => "" } } CONFIG sample("message" => "random message") do insist { subject }.raises(LogStash::ConfigurationError) end end end describe "multi character splitting" do subject do plugin = LogStash::Filters::KV.new(options) plugin.register plugin end let(:data) { {"message" => message} } let(:event) { LogStash::Event.new(data) } shared_examples "parsing all fields and values" do it "parses all fields and values" do subject.filter(event) expect(event.get("hello")).to eq("world") expect(event.get("foo")).to eq("bar") expect(event.get("baz")).to eq("fizz") expect(event.get("doublequoted")).to eq("hello world") expect(event.get("singlequoted")).to eq("hello world") expect(event.get("bracketsone")).to eq("hello world") expect(event.get("bracketstwo")).to eq("hello world") expect(event.get("bracketsthree")).to eq("hello world") end end context "empty value_split_pattern" do let(:options) { { "value_split_pattern" => "" } } it "should raise ConfigurationError" do expect{subject}.to raise_error(LogStash::ConfigurationError) end end context "empty field_split_pattern" do let(:options) { { "field_split_pattern" => "" } } it "should raise ConfigurationError" do expect{subject}.to raise_error(LogStash::ConfigurationError) end end context "single split" do let(:message) { "hello:world foo:bar baz:fizz doublequoted:\"hello world\" singlequoted:'hello world' bracketsone:(hello world) bracketstwo:[hello world] bracketsthree:" } let(:options) { { "field_split" => " ", "value_split" => ":", } } it_behaves_like "parsing all fields and values" end context "value split multi" do let(:message) { "hello::world foo::bar baz::fizz doublequoted::\"hello world\" singlequoted::'hello world' bracketsone::(hello world) bracketstwo::[hello world] bracketsthree::" } let(:options) { { "field_split" => " ", "value_split_pattern" => "::", } } it_behaves_like "parsing all fields and values" end context "field and value split multi" do let(:message) { "hello::world__foo::bar__baz::fizz__doublequoted::\"hello world\"__singlequoted::'hello world'__bracketsone::(hello world)__bracketstwo::[hello world]__bracketsthree::" } let(:options) { { "field_split_pattern" => "__", "value_split_pattern" => "::", } } it_behaves_like "parsing all fields and values" end context "field and value split multi with regex" do let(:message) { "hello:world_foo::bar__baz:::fizz___doublequoted:::\"hello world\"____singlequoted:::::'hello world'____bracketsone:::(hello world)__bracketstwo:[hello world]_bracketsthree::::::" } let(:options) { { "field_split_pattern" => "_+", "value_split_pattern" => ":+", } } it_behaves_like "parsing all fields and values" end context "field and value split multi using singe char" do let(:message) { "hello:world foo:bar baz:fizz doublequoted:\"hello world\" singlequoted:'hello world' bracketsone:(hello world) bracketstwo:[hello world] bracketsthree:" } let(:options) { { "field_split_pattern" => " ", "value_split_pattern" => ":", } } it_behaves_like "parsing all fields and values" end context "field and value split multi using escaping" do let(:message) { "hello++world??foo++bar??baz++fizz??doublequoted++\"hello world\"??singlequoted++'hello world'??bracketsone++(hello world)??bracketstwo++[hello world]??bracketsthree++" } let(:options) { { "field_split_pattern" => "\\?\\?", "value_split_pattern" => "\\+\\+", } } it_behaves_like "parsing all fields and values" end context "example from @guyboertje in #15" do let(:message) { 'key1: val1; key2: val2; key3: https://site/?g={......"...; CLR rv:11.0)"..}; key4: val4;' } let(:options) { { "field_split_pattern" => ";\s*(?=key.+?:)|;$", "value_split_pattern" => ":\s+", } } it "parses all fields and values" do subject.filter(event) expect(event.get("key1")).to eq("val1") expect(event.get("key2")).to eq("val2") expect(event.get("key3")).to eq("https://site/?g={......\"...; CLR rv:11.0)\"..}") expect(event.get("key4")).to eq("val4") end end end