test/composition.rb in cuba-3.4.0 vs test/composition.rb in cuba-3.5.0

@@ -17,5 +17,53 @@
 
    _, _, resp = Cuba.call(env)
 
    assert_response resp, ["View 101"]
 end
+
+test "redefining not_found" do
+  class Users < Cuba
+    def not_found
+      res.status = 404
+      res.write "Not found!"
+    end
+
+    define do
+      on root do
+        res.write "Users"
+      end
+    end
+  end
+
+  class Cuba
+    def not_found
+      res.status = 404
+      res.write "Error 404"
+    end
+  end
+
+  Cuba.define do
+    on "users" do
+      run Users
+    end
+  end
+
+  env = { "SCRIPT_NAME" => "/", "PATH_INFO" => "/users" }
+
+  _, _, resp = Cuba.call(env)
+
+  assert_response resp, ["Users"]
+
+  env = { "SCRIPT_NAME" => "/", "PATH_INFO" => "/users/42" }
+
+  status, _, resp = Cuba.call(env)
+
+  assert_response resp, ["Not found!"]
+  assert_equal status,  404
+
+  env = { "SCRIPT_NAME" => "/", "PATH_INFO" => "/guests" }
+
+  status, _, resp = Cuba.call(env)
+
+  assert_response resp, ["Error 404"]
+  assert_equal status,  404
+end