test/test_marlowe.rb in marlowe-1.0.2 vs test/test_marlowe.rb in marlowe-1.0.3

@@ -14,6 +14,19 @@
   def test_with_header
     @middleware.call({'HTTP_CORRELATION_ID' => 'testvalue'})
     refute_empty @app.coordination_id
     assert_equal 'testvalue', @app.coordination_id
   end
+
+  def test_with_custom_no_header
+    @customized_middleware = Marlowe::Middleware.new(@app, correlation_header: "Custom-Header")
+    @customized_middleware.call({})
+    refute_empty @app.coordination_id
+  end
+
+  def test_with_custom_header
+    @customized_middleware = Marlowe::Middleware.new(@app, correlation_header: "Custom-Header")
+    @customized_middleware.call({'HTTP_CUSTOM_HEADER' => 'testvalue'})
+    refute_empty @app.coordination_id
+    assert_equal 'testvalue', @app.coordination_id
+  end
 end