def test_MainHandler(self): # 周期ツイート制御のインスタンスを生成 ct = CycleTweet(datetime.datetime(2012, 5, 6, 0, 0)) # 各ハンドラをモックする self.mocker.StubOutWithMock(ct, "scheduleHandler") self.mocker.StubOutWithMock(ct, "cycleHandler") self.mocker.StubOutWithMock(ct, "anniversaryHandler") # 記録をリセット self.mocker.ResetAll() # 呼び出しの記録 ct.scheduleHandler() ct.cycleHandler() ct.anniversaryHandler() # 再現動作の開始 self.mocker.ReplayAll() # ツイート ct.handler() # 実動作の検証 self.mocker.VerifyAll()
def test_ScheduleHandlerTomorrow(self): # 周期ツイート制御のインスタンスを生成 ct = CycleTweet(datetime.datetime(2011, 11, 12, 22, 50)) # カレンダーのダミーファイルを指定 ct.calUrl = u'./gcaltest.xml' # tweepyAPIのモックを作成 apiMock = self.mocker.CreateMock(tweepy.API) # モックをテスト対象に設定 ct.api = apiMock # スケジュール取得関数をモック化 self.mocker.StubOutWithMock(ct, "todaySchedule") self.mocker.StubOutWithMock(ct, "tomorrowSchedule") # 記録をリセット self.mocker.ResetAll() # 呼び出しの記録 ct.tomorrowSchedule(mox.IsA(GCalendar.GCalendar)).AndReturn(u'tomorrow') apiMock.update_status(u'tomorrow') # 再現動作の開始 self.mocker.ReplayAll() # スケジュール制御のハンドラ呼び出し ct.scheduleHandler() # 実動作の検証 self.mocker.VerifyAll()