From 823f992646d4ad9eb57c5757bcad450047977e97 Mon Sep 17 00:00:00 2001
From: Felipe Lessa <felipe.lessa@gmail.com>
Date: Fri, 29 May 2015 00:40:55 -0300
Subject: [PATCH] Tests for session invalidation.

---
 serversession/tests/Main.hs | 47 +++++++++++++++++++++++++++++++++++--
 1 file changed, 45 insertions(+), 2 deletions(-)

diff --git a/serversession/tests/Main.hs b/serversession/tests/Main.hs
index e8165a6..3bea094 100644
--- a/serversession/tests/Main.hs
+++ b/serversession/tests/Main.hs
@@ -214,7 +214,50 @@ main = hspec $ parallel $ do
     it "works for a complex example" pending
 
   describe "invalidateIfNeeded" $ do
-    it "should have more tests" pending
+    let prepareInvalidateIfNeeded authId = do
+          let oldSession = Session
+                { sessionKey        = S "123456789-123456789-1234"
+                , sessionAuthId     = authId
+                , sessionData       = M.empty
+                , sessionCreatedAt  = TI.addUTCTime (-10) fakenow
+                , sessionAccessedAt = TI.addUTCTime (-5)  fakenow }
+          sto <- prepareMockStorage [oldSession]
+          st  <- createState sto
+          return (oldSession, sto, st)
+        allEdges = let x = [Nothing, Just "john", Just "jane"] in (,) <$> x <*> x
+
+    it "does not invalidate when not changing auth ID nor explicitly requesting" $ do
+      forM_ [Nothing, Just "john"] $ \authId -> do
+        (session, sto, st) <- prepareInvalidateIfNeeded authId
+        let d = DecomposedSession authId DoNotForceInvalidate M.empty
+        invalidateIfNeeded st (Just session) d `shouldReturn` Just session
+        getMockOperations sto `shouldReturn` []
+
+    it "invalidates the current session when changing auth ID" $ do
+      forM_ [ (Just "john",  Just "jane")
+            , (Just "admin", Nothing)
+            , (Nothing,      Just "joe") ] $ \edgeTransition -> do
+        (session, sto, st) <- prepareInvalidateIfNeeded (fst edgeTransition)
+        let d = DecomposedSession (snd edgeTransition) DoNotForceInvalidate M.empty
+        invalidateIfNeeded st (Just session) d `shouldReturn` Nothing
+        getMockOperations sto `shouldReturn` [DeleteSession (sessionKey session)]
+
+    it "invalidates the current session when CurrentSessionId is forced" $ do
+      forM_ allEdges $ \edgeTransition -> do
+        (session, sto, st) <- prepareInvalidateIfNeeded (fst edgeTransition)
+        let d = DecomposedSession (snd edgeTransition) CurrentSessionId M.empty
+        invalidateIfNeeded st (Just session) d `shouldReturn` Nothing
+        getMockOperations sto `shouldReturn` [DeleteSession (sessionKey session)]
+
+    it "invalidates all of the user's sessions when AllSessionIdsOfLoggedUser is forced" $ do
+      forM_ allEdges $ \edgeTransition -> do
+        (session, sto, st) <- prepareInvalidateIfNeeded (fst edgeTransition)
+        let d = DecomposedSession (snd edgeTransition) AllSessionIdsOfLoggedUser M.empty
+        invalidateIfNeeded st (Just session) d `shouldReturn` Nothing
+        let expected = DeleteSession (sessionKey session) :
+                       maybe [] ((:[]) . DeleteAllSessionsOfAuthId) (snd edgeTransition)
+                       -- It deletes all sessions only when there's an authId.
+        getMockOperations sto `shouldReturn` expected
 
   describe "saveSessionOnDb" $ do
     let prepareSaveSessionOnDb = do
@@ -436,7 +479,7 @@ prepareMockStorage sessions = do
 -- | Get the list of mock operations that were made and clear
 -- them.  The operations are listed in chronological order.
 getMockOperations :: MockStorage -> IO [MockOperation]
-getMockOperations = flip I.atomicModifyIORef' ((,) []) . mockOperations
+getMockOperations = flip I.atomicModifyIORef' ((,) [] . reverse) . mockOperations
 
 
 -- | Add a mock operations to the log.