Tested attempt to fix rspec

jheysel-r7 · Apr 29, 2024 · 6c74d14 · 6c74d14
1 parent f5f1dea
commit 6c74d14
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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
+end