def test_Omikuji(self): ds = DataStore.DataStore() ds.addStatus(DataStore.Type.omikuji, u'大吉') ds.addStatus(DataStore.Type.omikuji, u'中吉') ds.addStatus(DataStore.Type.omikuji, u'小吉') # 返信制御クラスを作成 rt = ReplyTweet() # updateをモック self.mocker.StubOutWithMock(rt, "update") # 再現を初期化 self.mocker.ResetAll() # 動作を記録 rt.update(mox.Regex(u'@user1 .* \(1おみくじちょうだい！\)'), 100) rt.update(mox.Regex(u'@user2 .* \(2おみくじちょうだい！\)'), 200) rt.update(mox.Regex(u'@user3 .* \(3おみくじちょうだい！\)'), 300) # 再現を開始 self.mocker.ReplayAll() mlist =[\ (100, u'user1', u'1おみくじちょうだい！'), \ (200, u'user2', u'2おみくじちょうだい！'), \ (300, u'user3', u'3おみくじちょうだい！') \ ] rt.omikujiTweet(mlist) # 再現を検証 self.mocker.VerifyAll()
def test_MentionReplyHandler(self): # 返信制御クラスの作成 rt = ReplyTweet() mlist =[\ (100, u'user1', u'1ハンドラ'), \ (200, u'user2', u'2ハンドラ'), \ (300, u'user3', u'3ハンドラ') \ ] # 各エントリをモック self.mocker.StubOutWithMock(rt, "createMentionList") self.mocker.StubOutWithMock(rt, "iyahoTweet") self.mocker.StubOutWithMock(rt, "uhyouTweet") self.mocker.StubOutWithMock(rt, "peroperoTweet") self.mocker.StubOutWithMock(rt, "douiTweet") self.mocker.StubOutWithMock(rt, "enochTweet") self.mocker.StubOutWithMock(rt, "omikujiTweet") self.mocker.StubOutWithMock(rt, "omikujiAdd") # 再現を初期化 self.mocker.ResetAll() # 動作を記録 rt.createMentionList().AndReturn(mlist) rt.iyahoTweet(mlist) rt.uhyouTweet(mlist) rt.peroperoTweet(mlist) rt.douiTweet(mlist) rt.enochTweet(mlist) rt.omikujiTweet(mlist) rt.omikujiAdd(mlist) # 再現を開始 self.mocker.ReplayAll() rt.mentionReplyHandler() # 再現を検証 self.mocker.VerifyAll()