Skip to content

Commit f9f74fd

Browse files
committed
test_filter_record_transformer: pass a Time object to Timecop.freeze
Signed-off-by: Shizuo Fujita <fujita@clear-code.com>
1 parent 308b336 commit f9f74fd

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

test/plugin/test_filter_record_transformer.rb

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,10 @@ class RecordTransformerFilterTest < Test::Unit::TestCase
1313
@hostname = Socket.gethostname.chomp
1414
@tag = 'test.tag'
1515
@tag_parts = @tag.split('.')
16-
@time = event_time('2010-05-04 03:02:01 UTC')
17-
Timecop.freeze(@time)
16+
17+
@timestamp = Time.parse('2010-05-04 03:02:01 UTC')
18+
Timecop.freeze(@timestamp)
19+
@event_time = event_time
1820
end
1921

2022
teardown do
@@ -44,7 +46,7 @@ def filter(config, msgs = [''])
4446
d = create_driver(config)
4547
d.run {
4648
msgs.each { |msg|
47-
d.feed(@tag, @time, {'foo' => 'bar', 'message' => msg, 'nest' => {'k1' => 'v1', 'k2' => 'v2'}})
49+
d.feed(@tag, @event_time, {'foo' => 'bar', 'message' => msg, 'nest' => {'k1' => 'v1', 'k2' => 'v2'}})
4850
}
4951
}
5052
d.filtered
@@ -66,7 +68,7 @@ def filter(config, msgs = [''])
6668
assert_equal('bar', r['foo'])
6769
assert_equal(@hostname, r['hostname'])
6870
assert_equal(@tag, r['tag'])
69-
assert_equal(Time.at(@time).localtime.to_s, r['time'])
71+
assert_equal(Time.at(@timestamp).localtime.to_s, r['time'])
7072
assert_equal("#{@hostname} #{@tag_parts[-1]} #{msgs[i]}", r['message'])
7173
assert_equal({'k1' => 'v1', 'k2' => 'v2'}, r['nest'])
7274
end
@@ -79,7 +81,7 @@ def filter(config, msgs = [''])
7981
assert_not_include(r, 'foo')
8082
assert_equal(@hostname, r['hostname'])
8183
assert_equal(@tag, r['tag'])
82-
assert_equal(Time.at(@time).localtime.to_s, r['time'])
84+
assert_equal(Time.at(@timestamp).localtime.to_s, r['time'])
8385
assert_not_include(r, 'message')
8486
end
8587
end
@@ -100,7 +102,7 @@ def filter(config, msgs = [''])
100102
assert_not_include(r, 'foo')
101103
assert_equal(@hostname, r['hostname'])
102104
assert_equal(@tag, r['tag'])
103-
assert_equal(Time.at(@time).localtime.to_s, r['time'])
105+
assert_equal(Time.at(@timestamp).localtime.to_s, r['time'])
104106
assert_equal("#{@hostname} #{@tag_parts[-1]} #{msgs[i]}", r['message'])
105107
end
106108
end
@@ -222,7 +224,7 @@ def filter(config, msgs = [''])
222224
{ 'eventType0' => 'bar', 'message' => msg }
223225
end
224226
records.each do |record|
225-
d.feed(@tag, @time, record)
227+
d.feed(@tag, @event_time, record)
226228
end
227229
}
228230
d.filtered
@@ -294,9 +296,9 @@ def filter(config, msgs = [''])
294296
filtered = filter(config)
295297
filtered.each do |t, r|
296298
if enable_ruby == "yes"
297-
assert_equal(Time.at(@time).localtime, r['message'])
299+
assert_equal(Time.at(@timestamp).localtime, r['message'])
298300
else
299-
assert_equal(Time.at(@time).localtime.to_s, r['message'])
301+
assert_equal(Time.at(@timestamp).localtime.to_s, r['message'])
300302
end
301303
end
302304
end
@@ -533,7 +535,7 @@ def filter(config, msgs = [''])
533535
]
534536
d = create_driver(config)
535537
message = {"@timestamp" => "foo"}
536-
d.run { d.feed(@tag, @time, message) }
538+
d.run { d.feed(@tag, @event_time, message) }
537539
filtered = d.filtered
538540
filtered.each do |t, r|
539541
assert_equal(message["@timestamp"], r['foo'])
@@ -551,7 +553,7 @@ def filter(config, msgs = [''])
551553
]
552554
d = create_driver(config)
553555
message = {"@timestamp" => "foo"}
554-
d.run { d.feed(@tag, @time, message) }
556+
d.run { d.feed(@tag, @event_time, message) }
555557
filtered = d.filtered
556558
filtered.each do |t, r|
557559
assert_equal([message["@timestamp"]], r['foo'])

0 commit comments

Comments
 (0)