test/test_website.rb in webgen-0.5.7 vs test/test_website.rb in webgen-0.5.8

@@ -17,10 +17,19 @@
     assert_nil(ws.config)
     assert_nil(ws.tree)
     assert_equal('dir', ws.directory)
     assert_throws(:called) { ws.init }
     assert_nil(Thread.current[:webgen_website])
+
+    ws = Webgen::Website.new
+    assert_equal(Dir.pwd, ws.directory)
+    ENV['WEBGEN_WEBSITE'] = 'hallo'
+    ws = Webgen::Website.new
+    assert_equal('hallo', ws.directory)
+    ENV['WEBGEN_WEBSITE'] = ''
+    ws = Webgen::Website.new
+    assert_equal(Dir.pwd, ws.directory)
   end
 
   def test_autoload_service
     ws = Webgen::Website.new('unknown', nil)
     ws.init
@@ -50,9 +59,17 @@
 
   def test_execute_in_env
     ws = Webgen::Website.new('hallo')
     assert_nil(Webgen::WebsiteAccess.website)
     ws.execute_in_env { assert_not_nil(Webgen::WebsiteAccess.website) }
+    assert_nil(Webgen::WebsiteAccess.website)
+    ws.execute_in_env do
+      assert_equal(ws, Webgen::WebsiteAccess.website)
+      ws2 = Webgen::Website.new("hallo2")
+      ws2.execute_in_env { assert_equal(ws2, Webgen::WebsiteAccess.website) }
+      assert_equal(ws, Webgen::WebsiteAccess.website)
+    end
+    assert_equal(nil, Webgen::WebsiteAccess.website)
   end
 
   def test_read_config_file
     dir = File.join(Dir.tmpdir, 'webgen-' + Process.pid.to_s)
     FileUtils.mkdir_p(dir)