test/logster/middleware/test_viewer.rb in logster-0.0.2 vs test/logster/middleware/test_viewer.rb in logster-0.0.3

@@ -3,22 +3,50 @@
 require 'logster/redis_store'
 require 'logster/middleware/viewer'
 
 class TestViewer < Minitest::Test
 
+  class BrokenApp
+    def call(env)
+      [500, {}, ["broken"]]
+    end
+  end
+
+  def setup
+    Logster.store = Logster::RedisStore.new
+  end
+
   def teardown
+    Logster.config.subdirectory = nil
+    Logster.config.authorize_callback = nil
+    Logster.store = nil
   end
 
   def viewer
     @viewer ||= begin
-                  store = Logster::RedisStore.new
-                  Logster::Middleware::Viewer.new(nil, store: store, path: "/logsie")
+                  Logster.config.subdirectory = "/logsie"
+                  Logster::Middleware::Viewer.new(nil)
                 end
   end
 
+  def test_authorize_callback
+    Logster.config.authorize_callback = lambda{ |env|
+      env["authorized"]
+    }
+
+    viewer = Logster::Middleware::Viewer.new(BrokenApp.new)
+    status, _  = viewer.call({"PATH_INFO" => "/logs"})
+    assert_equal(500, status)
+
+    status, _  = viewer.call({"PATH_INFO" => "/logs", "authorized" => true})
+    assert_equal(200, status)
+  end
+
   def test_path_resolution
     assert_nil(viewer.send(:resolve_path, "/logs"))
+    assert_nil(viewer.send(:resolve_path, "/admin/logsie"))
+    assert_nil(viewer.send(:resolve_path, "/admin/logsie/bla"))
     assert_equal("/",viewer.send(:resolve_path, "/logsie"))
     assert_equal("/",viewer.send(:resolve_path, "/logsie/"))
     assert_equal("/hello/world",viewer.send(:resolve_path, "/logsie/hello/world"))
   end
 
@@ -27,8 +55,12 @@
     env["PATH_INFO"] = "/logsie/javascript/external/jquery.min.js"
     env["REQUEST_METHOD"] = "GET"
 
     result,  = viewer.call(env)
     assert_equal(200, result)
+  end
+
+  def test_regex_parse
+    assert_equal(/hello/i, viewer.send(:parse_regex, '/hello/i'))
   end
 
 end