test/cuba_test.rb in cuba-0.0.3 vs test/cuba_test.rb in cuba-0.0.4

@@ -1,6 +1,6 @@
-$:.unshift(File.join("..", "lib"))
+$:.unshift(File.dirname(__FILE__) + "/../lib/")
 
 require "cuba"
 require "cuba/test"
 
 form = <<EOS
@@ -16,16 +16,21 @@
       res.write "Enter your username"
       res.write form
     end
 
     on default do
-      res.redirect "/login"
+      if session[:user]
+        res.write "Hey #{session[:user]}!"
+      else
+        res.redirect "/login"
+      end
     end
   end
 
   on post, path("login") do
     on param("user") do |user|
+      session[:user] = user
       res.write "Got #{user}"
     end
   end
 end
 
@@ -38,8 +43,12 @@
 
       fill_in "user", :with => "Michel"
       click_button "Login"
 
       assert_contain "Got Michel"
+
+      visit "/"
+
+      assert_contain "Hey Michel!"
     end
   end
 end