Support header extractions

rinconjc · Apr 22, 2020 · cdd2cc5 · cdd2cc5
1 parent 3465ee2
commit cdd2cc5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/example/sample-tests.yaml b/example/sample-tests.yaml
@@ -22,8 +22,9 @@ Fake API:
         {id:"#\\d+"}
       headers:
         Content-Type: application/json; charset=utf-8
-      extract:
-        postId: $.id
+    extract:
+      postId: $.id
+      respType: header.Content-Type
   Retrieve Post:
     get: $fakeServer$/posts/$postId$
     headers:

diff --git a/src/rester/core.clj b/src/rester/core.clj
@@ -138,6 +138,7 @@
                (let [result (cond
                               (.startsWith path "$") (piped-json-path path body)
                               (.startsWith path "#") (re-find (re-pattern path) body)
+                              (.startsWith path "header.") (headers (str/replace-first path "header." ""))
                               true (piped-json-path (str "$." path) resp))]
                  (log/info "extracted:" name "=" result)
                  [name result])

diff --git a/test/rester/core_test.clj b/test/rester/core_test.clj
@@ -42,7 +42,9 @@
 
 (deftest test-extract-data
   (testing "simple json path"
-    (is (= {"id" 100} (extract-data {:body {:id 100 "name" "blah"}} {"id" "$.id"})))))
+    (is (= {"id" 100} (extract-data {:body {:id 100 "name" "blah"}} {"id" "$.id"}))))
+  (testing "extract headers"
+    (is (= {"x" "1234"} (extract-data {:body {} :headers {"header 123" "1234"}} {"x" "header.header 123"})))))
 
 (deftest test-ordered-iterator
   (testing "test ordered iterator"