feat: add Server-Timing header with JWT duration

2023-09-15 14:39:12 -03:00
parent fa182c216e
commit c195eece65
8 changed files with 197 additions and 59 deletions
@@ -0,0 +1,79 @@
+module Feature.Query.ServerTimingSpec where
+
+import Network.Wai (Application)
+
+import Network.HTTP.Types
+import Test.Hspec
+import Test.Hspec.Wai
+import Test.Hspec.Wai.JSON
+
+import Protolude  hiding (get)
+import SpecHelper
+
+spec :: SpecWith ((), Application)
+spec =
+  describe "Show Duration on Server-Timing header" $ do
+
+    context "responds with Server-Timing header" $ do
+      it "works with get request" $ do
+        request methodGet  "/organizations?id=eq.6"
+          []
+          ""
+          `shouldRespondWith`
+          [json|[{"id":6,"name":"Oscorp","referee":3,"auditor":4,"manager_id":6}]|]
+          { matchStatus  = 200
+          , matchHeaders = [ matchContentTypeJson
+                           , matchHeaderPresent "Server-Timing"]
+          }
+
+      it "works with post request" $
+        request methodPost  "/organizations?select=*"
+          [("Prefer","return=representation")]
+          [json|{"id":7,"name":"John","referee":null,"auditor":null,"manager_id":6}|]
+          `shouldRespondWith`
+          [json|[{"id":7,"name":"John","referee":null,"auditor":null,"manager_id":6}]|]
+          { matchStatus  = 201
+          , matchHeaders = [ matchContentTypeJson
+                           , matchHeaderPresent "Server-Timing"]
+          }
+
+      it "works with patch request" $
+        request methodPatch "/no_pk?b=eq.0" mempty
+            [json| { b: "1" } |]
+          `shouldRespondWith`
+            ""
+            { matchStatus  = 204
+            , matchHeaders = [ matchHeaderAbsent hContentType
+                             , matchHeaderPresent "Server-Timing" ]
+            }
+
+      it "works with put request" $
+        request methodPut "/tiobe_pls?name=eq.Go"
+            [("Prefer", "return=representation")]
+            [json| [ { "name": "Go", "rank": 19 } ]|]
+          `shouldRespondWith`
+            [json| [ { "name": "Go", "rank": 19 } ]|]
+            { matchStatus  = 200
+            , matchHeaders = [ matchHeaderPresent "Server-Timing" ]
+            }
+
+      it "works with delete request" $
+        request methodDelete "/items?id=eq.1"
+            []
+            ""
+          `shouldRespondWith`
+            ""
+            { matchStatus  = 204
+            , matchHeaders = [ matchHeaderAbsent hContentType
+                             , matchHeaderPresent "Server-Timing" ]
+            }
+
+      it "works with rpc call" $
+        request methodPost "/rpc/ret_point_overloaded"
+          []
+          [json|{"x": 1, "y": 2}|]
+          `shouldRespondWith`
+          [json|{"x": 1, "y": 2}|]
+          { matchStatus  = 200
+          , matchHeaders = [ matchHeaderPresent "Server-Timing" ]
+          }