def test_getState_validCookieFormat_nonExtendable(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
cookieKey = UserInQueueStateCookieRepository.getCookieKey(eventId)
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
issueTime = str(QueueitHelpers.getCurrentTime())
hashValue = UnitTestHelper.generateHash(eventId, queueId, 3, "idle",
issueTime, secretKey)
cookieValue = ("EventId=" + eventId + "&QueueId=" + queueId +
"&FixedValidityMins=3" + "&RedirectType=idle" +
"&IssueTime=" + issueTime + "&Hash=" + hashValue)
wfHandler.setCookie(cookieKey, cookieValue,
QueueitHelpers.getCookieExpirationDate(),
cookieDomain, isCookieHttpOnly, isCookieSecure)
state = testObject.getState(eventId, 10, secretKey, True)
assert (not state.isStateExtendable())
assert (state.isValid)
assert (state.isFound)
assert (state.queueId == queueId)
assert (state.redirectType == "idle")
def test_hasValidState_noCookie_stateIsNotValid(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieValidity = 10
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (not state.isValid)
def test_getState_NoCookie(self):
fakeContext = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(fakeContext)
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
state = testObject.getState(eventId, 10, secretKey, True)
assert (not state.isFound)
assert (not state.isValid)
def test_extendQueueCookie_cookieExist(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = True
isCookieSecure = True
queueId = "queueId"
cookieKey = UserInQueueStateCookieRepository.getCookieKey(eventId)
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, None, cookieDomain,
isCookieHttpOnly, isCookieSecure, "Queue", secretKey)
testObject.reissueQueueCookie(eventId, 12, cookieDomain,
isCookieHttpOnly, isCookieSecure,
secretKey)
state = testObject.getState(eventId, 5, secretKey, True)
assert (state.isValid)
assert (state.queueId == queueId)
assert (state.isStateExtendable())
expirationTimeDelta = wfHandler.cookieList[cookieKey][
"expiration"].replace(
microsecond=0) - QueueitHelpers.getCookieExpirationDate(
).replace(microsecond=0)
assert (str(expirationTimeDelta) == "0:00:00")
assert (
wfHandler.cookieList[cookieKey]["cookieDomain"] == cookieDomain)
assert (
wfHandler.cookieList[cookieKey]["isHttpOnly"] == isCookieHttpOnly)
assert (wfHandler.cookieList[cookieKey]["isSecure"] == isCookieSecure)
def test_store_hasValidState_expiredCookie_stateIsNotValid(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
cookieValidity = -1
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, None, cookieDomain,
isCookieHttpOnly, isCookieSecure, "Idle", secretKey)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (not state.isValid)
def test_store_hasValidState_nonExtendableCookie_CookieIsSaved(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = True
isCookieSecure = True
queueId = "queueId"
cookieValidity = 3
cookieKey = UserInQueueStateCookieRepository.getCookieKey(eventId)
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, cookieValidity, cookieDomain,
isCookieHttpOnly, isCookieSecure, "Idle", secretKey)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (state.isValid)
assert (state.queueId == queueId)
assert (state.isStateExtendable() == False)
assert (state.redirectType == 'Idle')
assert (state.fixedCookieValidityMinutes == 3)
expirationTimeDelta = wfHandler.cookieList[cookieKey][
"expiration"].replace(
microsecond=0) - QueueitHelpers.getCookieExpirationDate(
).replace(microsecond=0)
assert (str(expirationTimeDelta) == "0:00:00")
self.assertEqual(wfHandler.cookieList[cookieKey]["cookieDomain"],
cookieDomain)
self.assertEqual(wfHandler.cookieList[cookieKey]["isHttpOnly"],
isCookieHttpOnly)
self.assertEqual(wfHandler.cookieList[cookieKey]["isSecure"],
isCookieSecure)
def test_cancelQueueCookie(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
cookieValidity = 20
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, 20, cookieDomain, isCookieHttpOnly,
isCookieSecure, "Queue", secretKey)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (state.isValid)
testObject.cancelQueueCookie(eventId, cookieDomain, isCookieHttpOnly,
isCookieSecure)
state2 = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (not state2.isValid)
assert (int(wfHandler.setCookieCalls[1]["expiration"]) == -1)
assert (wfHandler.setCookieCalls[1]["cookieDomain"] == cookieDomain)
assert (wfHandler.setCookieCalls[1]["value"] == None)
def test_extendQueueCookie_cookieDoesNotExist(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store("event2", queueId, 20, cookieDomain, isCookieHttpOnly,
isCookieSecure, "Queue", secretKey)
testObject.reissueQueueCookie(eventId, 12, cookieDomain,
isCookieHttpOnly, isCookieSecure,
secretKey)
assert (len(wfHandler.setCookieCalls) == 1)
def test_hasValidState_invalidCookie_stateIsNotValid(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
cookieKey = UserInQueueStateCookieRepository.getCookieKey(eventId)
cookieValidity = 10
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, 20, cookieDomain, isCookieHttpOnly,
isCookieSecure, "Queue", secretKey)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (state.isValid)
wfHandler.cookieList[cookieKey][
"value"] = "IsCookieExtendable=ooOOO&Expires=|||&QueueId=000&Hash=23232"
state2 = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (not state2.isValid)
def test_store_hasValidState_tamperedCookie_stateIsNotValid_eventId(self):
eventId = "event1"
secretKey = "4e1deweb821-a82ew5-49da-acdqq0-5d3476f2068db"
cookieDomain = ".test.com"
isCookieHttpOnly = False
isCookieSecure = False
queueId = "queueId"
cookieValidity = 10
cookieKey = UserInQueueStateCookieRepository.getCookieKey(eventId)
wfHandler = HttpContextProviderMock()
testObject = UserInQueueStateCookieRepository(wfHandler)
testObject.store(eventId, queueId, 3, cookieDomain, isCookieHttpOnly,
isCookieSecure, "Idle", secretKey)
state = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (state.isValid)
oldCookieValue = wfHandler.cookieList[cookieKey]["value"]
wfHandler.cookieList[cookieKey]["value"] = oldCookieValue.replace(
"EventId=event1", "EventId=event2")
state2 = testObject.getState(eventId, cookieValidity, secretKey, True)
assert (not state2.isValid)
assert (not state2.isStateExtendable())