def testGetFileLinesReturnsLinesOfFile(self): filename = "fooFile.dat" fileLines = ["one", "two", "three", "all of the lines"] mockFile = testhelper.createSimpleMock() #file object when(mockFile).readlines().thenReturn(fileLines) self.testObj._getFile = lambda fname: mockFile if filename == fname else None result = self.testObj.getFileLines(filename) self.assertEqual(fileLines, result) verify(mockFile).close()
def setUp(self): self.sessionStore = testhelper.createSimpleMock() self.key = "foouser" self.expectedValue = {"userid" : "2309asdlj3"} when(self.sessionStore).get(self.key).thenReturn(self.expectedValue) self._buildTestObj()
def testGetFileContentsReturnsContentsOfFile(self): filename = "bar_file.info" fileContents = "the best file ever\nis not at this location" mockFile = testhelper.createSimpleMock() #file object when(mockFile).read().thenReturn(fileContents) self.testObj._getFile = lambda fname: mockFile if filename == fname else None result = self.testObj.getFileContents(filename) self.assertEqual(fileContents, result) verify(mockFile).close()
def setUp(self): self.applicationUrl = "http://sweetapp.com" self.key = "some key" self.expectedResult = "some data" self.cookies = ["123", "sdlkj", "324jd"] self.request = testhelper.createSimpleMock() self.request.application_url = self.applicationUrl self.request.cookies = self.cookies when(self.request).get(self.key).thenReturn(self.expectedResult) self.testObj = util.RequestDataAccessor.getInstance(self.request)
def testDeserializesRoundCorrectly(self): players = "some players" serializedTurnTracker = "a turn tracker" serializedTrickEvaluator = "a trick evaluator" hands = {"1" : ["card 1", "card 2"], "2" : ["card 3", "card 4"]} curTrick = "trick 1" prevTricks = ["trick 2", "trick 3"] scores = {"1" : 3, "2" : 2} data = {"turnTracker" : serializedTurnTracker, "trickEvaluator" : serializedTrickEvaluator, "hands" : hands, "curTrick" : curTrick, "prevTricks" : prevTricks, "scores" : scores} when(self.turnTrackerSerializer).deserialize(serializedTurnTracker, players).thenReturn(self.turnTracker) when(self.trickEvaluatorSerializer).deserialize(serializedTrickEvaluator).thenReturn(self.trickEvaluator) expectedCards = {} for playerId, hand in hands.iteritems(): for card in hand: deserializedCard = self._randomCard() when(self.cardSerializer).deserialize(card).thenReturn(deserializedCard) expectedCards[card] = deserializedCard expectedTricks = {curTrick : testhelper.createMock(euchre.Trick)} when(self.trickSerializer).deserialize(curTrick, self.trump).thenReturn(expectedTricks[curTrick]) for trick in prevTricks: deserializedTrick = testhelper.createMock(euchre.Trick) expectedTricks[trick] = deserializedTrick when(self.trickSerializer).deserialize(trick, self.trump).thenReturn(deserializedTrick) testhelper.replaceClass(src.euchre, "CardTranslator", testhelper.createSimpleMock()) self.wasCalled = False def assertCorrectTrump(trump): self.assertEqual(self.trump, trump) self.wasCalled = True src.euchre.CardTranslator.getInstance = assertCorrectTrump obj = self.testObj.deserialize(data, players) self.assertTrue(self.wasCalled) self.assertEqual(self.turnTracker, obj._turnTracker) verify(self.turnTrackerSerializer).deserialize(serializedTurnTracker, players) self.assertEqual(self.trickEvaluator, obj._trickEvaluator) for playerId, hand in hands.iteritems(): for i in range(len(hand)): self.assertEqual(expectedCards[hand[i]], obj._hands[playerId][i]) self.assertEqual(expectedTricks[curTrick], obj.getCurrentTrick()) for i in range(len(prevTricks)): self.assertEqual(expectedTricks[prevTricks[i]], obj.getPreviousTricks()[i]) for playerId, score in scores.iteritems(): self.assertEqual(score, obj._scores[playerId])
def _doTraining(self): when(self.requestDataAccessor).getCookies().thenReturn(self.requestCookies) when(facebook).get_user_from_cookie(self.requestCookies, self.appId, self.appSecret).thenReturn(self.fbCookie) when(facebook).GraphAPI(self.accessToken).thenReturn(self.graph) when(self.graph).get_object(self.playerId).thenReturn(self.profile) testhelper.replaceClass(src.social, "User", testhelper.createSimpleMock()) when(src.social.User).getInstance(self.playerId, self.expectedName).thenReturn(self.expectedUser) when(src.social.User).getInstance().thenReturn(self.nullUser) when(self.configManager).get("FB.sessionKey").thenReturn(self.sessionKey) when(self.configManager).get("FB.appId").thenReturn(self.appId) when(self.configManager).get("FB.appSecret").thenReturn(self.appSecret)
def setUp(self): self.response = testhelper.createSimpleMock() self.testObj = util.ResponseWriter.getInstance(self.response)