From 61d8b91f8aab4e96d2508493abece410e1083cc0 Mon Sep 17 00:00:00 2001 From: Hugo Thunnissen Date: Mon, 22 Oct 2018 12:06:44 +0200 Subject: [PATCH] Test stomp-frame-header and fix encountered bug --- stomp.el | 2 +- test/test.el | 14 ++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/stomp.el b/stomp.el index 10511f5..c3207d2 100644 --- a/stomp.el +++ b/stomp.el @@ -53,7 +53,7 @@ (defun stomp-frame-header (header frame) "Shortcut to access frame headers" (assoc-default - "content-length" + header (alist-get 'headers frame nil) 'string-equal nil)) (defun stomp-read-frame (buffer) diff --git a/test/test.el b/test/test.el index 5024d5e..5c84768 100755 --- a/test/test.el +++ b/test/test.el @@ -93,3 +93,17 @@ (map-put frame 'headers headers) (map-put frame 'content "Some content") (should (equal "SEND\nheader:value\n\nSome content\u0000" (stomp-frame-to-string frame))))) + +(ert-deftest can-get-frame-header () + "stomp-frame-header should be able to read arbitrary headers + from a frame" + (setq frame '((headers . (("bla" . "ok") ("random" . "barry"))))) + + (should (equal "ok" (stomp-frame-header "bla" frame))) + (should (equal "barry" (stomp-frame-header "random" frame)))) + +(ert-deftest will-return-nil-for-non-existent-header () + "stomp-frame-header should return nil if a header does not exist" + (setq frame '((headers . (("okay" . "bye"))))) + + (should (equal nil (stomp-frame-header "bla" frame))))