Add spec

cloudamqp · Nov 21, 2024 · 9f1adcb · 9f1adcb
1 parent c9725e8
commit 9f1adcb
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/spec/amqproxy/server_spec.cr b/spec/amqproxy/server_spec.cr
@@ -242,4 +242,21 @@ describe AMQProxy::Server do
       end
     end
   end
+
+  it "should treat all frames as heartbeats" do
+    with_server do |server, proxy_url|
+      Fiber.yield
+      AMQP::Client.start("#{proxy_url}?heartbeat=1") do |conn|
+        client = server.@clients.first?.should_not be_nil
+        last_heartbeat = client.@last_heartbeat
+        conn.channel
+        Fiber.yield
+        client.@last_heartbeat.should be > last_heartbeat
+        last_heartbeat = client.@last_heartbeat
+        conn.write AMQ::Protocol::Frame::Heartbeat.new
+        Fiber.yield
+        client.@last_heartbeat.should be > last_heartbeat
+      end
+    end
+  end
 end