module Feature.RpcPreRequestGucsSpec where

import qualified Data.ByteString.Lazy as BL (empty)

import Network.Wai      (Application)
import Network.Wai.Test (SResponse (simpleBody, simpleHeaders, simpleStatus))

import Network.HTTP.Types
import Test.Hspec          hiding (pendingWith)
import Test.Hspec.Wai
import Test.Hspec.Wai.JSON
import Text.Heredoc

import Protolude  hiding (get, put)
import SpecHelper

spec :: SpecWith ((), Application)
spec =
  describe "GUC headers on all methods via pre-request" $ do
    it "succeeds setting the headers on POST" $
      post "/items"
          [json|[{"id": 11111}]|]
        `shouldRespondWith` ""
          { matchStatus = 201
          , matchHeaders = ["X-Custom-Header" <:> "mykey=myval"]
          }

    it "succeeds setting the headers on GET and HEAD" $ do
      request methodGet "/items?id=eq.1"
          [("User-Agent", "MSIE 6.0")]
          ""
        `shouldRespondWith`
          [json|[{"id": 1}]|]
          { matchHeaders = ["Cache-Control" <:> "no-cache, no-store, must-revalidate"] }

      request methodHead "/items?id=eq.1"
          [("User-Agent", "MSIE 7.0")]
          ""
        `shouldRespondWith`
          ""
          { matchHeaders = ["Cache-Control" <:> "no-cache, no-store, must-revalidate"] }

      request methodHead "/projects"
          [("Accept", "text/csv")]
          ""
        `shouldRespondWith`
          ""
          { matchHeaders = ["Content-Disposition" <:> "attachment; filename=projects.csv"] }

    it "succeeds setting the headers on PATCH" $
        patch "/items?id=eq.1"
            [json|[{"id": 11111}]|]
          `shouldRespondWith` ""
            { matchStatus = 204
            , matchHeaders = ["X-Custom-Header" <:> "mykey=myval"]
            }

    it "succeeds setting the headers on PUT" $
      put "/items?id=eq.1"
          [json|[{"id": 1}]|]
        `shouldRespondWith` ""
          { matchStatus = 204
          , matchHeaders = ["X-Custom-Header" <:> "mykey=myval"]
          }

    it "succeeds setting the headers on DELETE" $
      delete "/items?id=eq.1"
        `shouldRespondWith`
          ""
          { matchStatus = 204
          , matchHeaders = ["X-Custom-Header" <:> "mykey=myval"]
          }
    it "can override the Content-Type header" $ do
      request methodHead "/clients?id=eq.1"
          []
          ""
        `shouldRespondWith`
          ""
          { matchStatus = 200
          , matchHeaders = ["Content-Type" <:> "application/custom+json"]
          }
      request methodHead "/rpc/getallprojects"
          []
          ""
        `shouldRespondWith`
          ""
          { matchStatus = 200
          , matchHeaders = ["Content-Type" <:> "application/custom+json"]
          }