diff --git a/test/plugin/out_forward/test_socket_cache.rb b/test/plugin/out_forward/test_socket_cache.rb index f7dec5a0f1..1d584d6b4e 100644 --- a/test/plugin/out_forward/test_socket_cache.rb +++ b/test/plugin/out_forward/test_socket_cache.rb @@ -17,7 +17,7 @@ class SocketCacheTest < Test::Unit::TestCase assert_equal(socket, c.checkout_or('key') { socket }) c.checkin(socket) - sock = dont_allow(mock!).open + sock = mock!.open.never.subject assert_equal(socket, c.checkout_or('key') { sock.open }) end @@ -130,7 +130,7 @@ def teardown c = Fluent::Plugin::ForwardOutput::SocketCache.new(10, $log) sock = mock!.close { 'closed' }.subject - sock2 = dont_allow(mock!).close + sock2 = mock!.close.never.subject stub(sock).inspect stub(sock2).inspect @@ -154,7 +154,7 @@ def teardown Timecop.freeze(Time.parse('2016-04-13 14:00:00 +0900')) c = Fluent::Plugin::ForwardOutput::SocketCache.new(10, $log) - sock = dont_allow(mock!).close + sock = mock!.close.never.subject stub(sock).inspect c.checkout_or('key') { sock } diff --git a/test/test_event_router.rb b/test/test_event_router.rb index 3601ddf10f..fda89db230 100644 --- a/test/test_event_router.rb +++ b/test/test_event_router.rb @@ -175,7 +175,7 @@ def event_router test "don't call default collector when tag matched" do event_router.add_rule('test', output) assert_rr do - dont_allow(default_collector).emit_events('test', is_a(OneEventStream)) + mock(default_collector).emit_events('test', is_a(OneEventStream)).never event_router.emit('test', Engine.now, 'k' => 'v') end # check emit handler doesn't catch rr error @@ -201,7 +201,7 @@ def filter_stream(_tag, es); end event_router.add_rule('test', filter) assert_rr do - dont_allow(filter).filter_stream('test', is_a(OneEventStream)) { events } + mock(filter).filter_stream('test', is_a(OneEventStream)).never event_router.emit('foo', Engine.now, 'k' => 'v') end end