test/default_headers_test.rb in tynn-1.4.0 vs test/default_headers_test.rb in tynn-2.0.0.alpha

@@ -1,14 +1,26 @@
-test "default headers" do
-  Tynn.set(:default_headers, "Content-Type" => "text/plain")
+# frozen_string_literal: true
 
-  Tynn.define do
-    root do
-      res.write("hei")
-    end
+require_relative "helper"
+
+class DefaultHeadersTest < Minitest::Test
+  def setup
+    @app = Class.new(Tynn)
   end
 
-  app = Tynn::Test.new(Tynn)
-  app.get("/")
+  def test_set_and_get_headers
+    @app.set(:default_headers, "Content-Type" => "text/plain")
 
-  assert_equal "text/plain", app.res.headers["Content-Type"]
+    assert_equal "text/plain", @app.default_headers["Content-Type"]
+  end
+
+  def test_respond_with_headers
+    @app.set(:default_headers, "Content-Type" => "text/plain")
+
+    @app.define {}
+
+    ts = Tynn::Test.new(@app)
+    ts.get("/")
+
+    assert_equal @app.default_headers, ts.res.headers
+  end
 end