Support regex header matching via expect_header_matches

brooklynDev · Nov 5, 2015 · a413dfe · a413dfe
1 parent fedc467
commit a413dfe
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/airborne/request_expectations.rb b/lib/airborne/request_expectations.rb
@@ -44,6 +44,10 @@ def expect_header_contains(key, content)
       expect_header_impl(key, content, true)
     end
 
+    def expect_header_matches(key, content)
+      expect_header_impl(key, content, false, true)
+    end
+
     def optional(hash)
       OptionalHashTypeExpectations.new(hash)
     end
@@ -58,11 +62,13 @@ def date
 
     private
 
-    def expect_header_impl(key, content, contains = nil)
+    def expect_header_impl(key, content, contains = nil, matches = nil)
       header = headers[key]
       if header
         if contains
           expect(header.downcase).to include(content.downcase)
+        elsif matches
+          expect(header.downcase).to match(content)
         else
           expect(header.downcase).to eq(content.downcase)
         end