From 6c74d14bb7534b2d526b2ffbb548e580421745d4 Mon Sep 17 00:00:00 2001 From: Jack Heysel Date: Mon, 29 Apr 2024 08:54:57 -0700 Subject: [PATCH] Tested attempt to fix rspec --- spec/lib/msf/core/auxiliary/rocketmq_spec.rb | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/spec/lib/msf/core/auxiliary/rocketmq_spec.rb b/spec/lib/msf/core/auxiliary/rocketmq_spec.rb index aa0642da445a..ce961b460969 100644 --- a/spec/lib/msf/core/auxiliary/rocketmq_spec.rb +++ b/spec/lib/msf/core/auxiliary/rocketmq_spec.rb @@ -9,7 +9,7 @@ end let(:mock_name_server_response) do - "\x00\x00\x00_{\"code\":105,\"extFields\":{\"Signature\":\"/u5P/wZUbhjanu4LM/UzEdo2u2I=\",\"topic\":\"TBW102\",\"AccessKey\":\"rocketmq2\"},\"flag\":0,\"language\":\"JAVA\",\"opaque\":1,\"serializeTypeCurrentRPC\":\"JSON\",\"version\":401}".b + "\x00\x00\x00\xC7\x00\x00\x00\xC3{\"code\":105,\"extFields\":{\"Signature\":\"/u5P/wZUbhjanu4LM/UzEdo2u2I=\",\"topic\":\"TBW102\",\"AccessKey\":\"rocketmq2\"},\"flag\":0,\"language\":\"JAVA\",\"opaque\":1,\"serializeTypeCurrentRPC\":\"JSON\",\"version\":401}".b end let(:expected_name_server_response) do @@ -55,6 +55,9 @@ describe '#send_version_request' do it 'returns version info' do expect(mock_sock).to receive(:send).with(mock_name_server_response, 0) + expect(mock_sock).to receive(:timed_read).with(4).and_return([expected_name_server_response.length].pack('N')) + expect(mock_sock).to receive(:timed_read).with(expected_name_server_response.length).and_return(expected_name_server_response) + expect(subject.send_version_request).to eq(expected_name_server_response) end end @@ -74,4 +77,4 @@ expect(subject.get_broker_port(expected_parsed_data_response, '172.16.199.1', default_broker_port: 10000)).to eq(10000) end end -end \ No newline at end of file +end