WaiSpec.hs

module Network.Bugsnag.WaiSpec
    ( spec
    ) where

import Prelude

import qualified Control.Exception as Exception
import Data.Bugsnag
import qualified Data.HashMap.Strict as HashMap
import Network.Bugsnag
import Network.Bugsnag.Wai
import Test.Hspec

data TestException = TestException
    deriving stock Show
    deriving anyclass Exception.Exception

spec :: Spec
spec = do
    describe "redactRequestHeaders" $ do
        it "redacts the given headers" $ do
            let bn = redactRequestHeaders ["Authorization"]

                event = runBeforeNotify bn TestException $ defaultEvent
                    { event_request = Just defaultRequest
                        { request_headers =
                            Just $ HashMap.fromList
                                [("Authorization", "secret"), ("X-Foo", "Bar")]
                        }
                    }

                lookupEventRequestHeader k e = do
                    r <- event_request e
                    hs <- request_headers r
                    HashMap.lookup k hs

            lookupEventRequestHeader "Authorization" event
                `shouldBe` Just "<redacted>"
            lookupEventRequestHeader "X-Foo" event `shouldBe` Just "Bar"

    describe "readForwardedFor" $ do
        it "handles empty" $ do
            readForwardedFor "" `shouldBe` Nothing

        it "reads a single value" $ do
            readForwardedFor "123.123.123" `shouldBe` Just "123.123.123"

        it "reads the first of many values" $ do
            readForwardedFor "123.123.123, 45.45.45"
                `shouldBe` Just "123.123.123"