def test_OmikujiAddEmpty(self): ds = DataStore.DataStore() # 返信制御クラスを作成 rt = ReplyTweet() mlist =[\ (100, u'user1', u'くじ追加「」'), \ (200, u'user2', u'くじ追加'), \ ] rt.omikujiAdd(mlist) lst = ds.getStatuses(DataStore.Type.omikuji) assert_equal(0, len(lst))
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()
def test_OmikujiAdd(self): ds = DataStore.DataStore() # 返信制御クラスを作成 rt = ReplyTweet() # updateをモック self.mocker.StubOutWithMock(rt, "update") mlist =[\ (100, u'user1', u'くじ追加「大吉」'), \ (200, u'user2', u'くじ追加「中吉」'), \ (300, u'user3', u'くじ追加「小吉」') \ ] rt.omikujiAdd(mlist) lst = ds.getStatuses(DataStore.Type.omikuji) assert_equal(u'大吉', lst[0][1]) assert_equal(u'中吉', lst[1][1]) assert_equal(u'小吉', lst[2][1])