def test_get(self): request, response = new_mock_request_response(self.m) self.m.StubOutWithMock(ArtifactSourceAccessor, "find_artifact_counts") self.m.StubOutWithMock(ArtifactSourceAccessor, "find_artifact_counts_newer") self.m.StubOutWithMock(memcache, "get_stats") self.m.StubOutWithMock(ConfigurationAccessor, "get_or_create") ArtifactSourceAccessor.find_artifact_counts().AndReturn({}) ArtifactSourceAccessor.find_artifact_counts_newer(IgnoreArg()).AndReturn({}) memcache.get_stats().AndReturn({}) # token = "oauth_token_secret=hi&oauth_token=hi" # config = MockEntity(key_name="hi", twitter_oauth_enabled=True, twitter_access_token=token, twitter_read_only=False) # ConfigurationAccessor.get_or_create().AndReturn(config) api = new_mock_twitter_api(self.m) api.GetUserTimeline(count=IgnoreArg()).AndReturn(()) api.GetDirectMessages(since=IgnoreArg()).AndReturn(()) api.GetReplies(since=IgnoreArg()).AndReturn(()) handler = StatsHandler() handler.initialize(request, response) self.m.ReplayAll() handler.get() self.m.VerifyAll()
def get(self, **kw): helper = RequestHelper(self) result = {} today = datetime.combine(datetime.now(), time(0, 0, 0)) result['timestamp'] = str(datetime.now()) result['today'] = str(today) warnings = [] result['warnings'] = warnings # source/artifact counts result[ 'source_artifact_counts'] = ArtifactSourceAccessor.find_artifact_counts( ) # new artifacts result[ 'new_artifacts'] = ArtifactSourceAccessor.find_artifact_counts_newer( today) # newer_arts = ArtifactAccessor.find_newer(today, refresh=True) # new_art_stats = [] # result['new_artifacts'] = new_art_stats # for art in newer_arts: # new_art_stats.append(dict(guid=art.guid, source_name=art.source_name)) # memcache stats result['memcache'] = memcache.get_stats() try: twactor = TwitterActor() # outgoing messages result['statuses_out'] = [ describe_status_with_timestamp(s) for s in twactor.latest_statuses(5) ] # incoming messages direct_stats = [] mention_stats = [] result['directs'] = direct_stats result['mentions'] = mention_stats directs, mentions = twactor.messages(today) directs.reverse() mentions.reverse() direct_stats.extend( [describe_status_with_timestamp(s) for s in directs]) mention_stats.extend( [describe_status_with_timestamp(s) for s in mentions]) except TwitterException, ex: warnings.append(str(ex))
def get(self, **kw): helper = RequestHelper(self) result = {} today = datetime.combine(datetime.now(), time(0, 0, 0)) result['timestamp'] = str(datetime.now()) result['today'] = str(today) warnings = [] result['warnings'] = warnings # source/artifact counts result['source_artifact_counts'] = ArtifactSourceAccessor.find_artifact_counts() # new artifacts result['new_artifacts'] = ArtifactSourceAccessor.find_artifact_counts_newer(today) # newer_arts = ArtifactAccessor.find_newer(today, refresh=True) # new_art_stats = [] # result['new_artifacts'] = new_art_stats # for art in newer_arts: # new_art_stats.append(dict(guid=art.guid, source_name=art.source_name)) # memcache stats result['memcache'] = memcache.get_stats() try: twactor = TwitterActor() # outgoing messages result['statuses_out'] = [describe_status_with_timestamp(s) for s in twactor.latest_statuses(5)] # incoming messages direct_stats = [] mention_stats = [] result['directs'] = direct_stats result['mentions'] = mention_stats directs, mentions = twactor.messages(today) directs.reverse() mentions.reverse() direct_stats.extend([describe_status_with_timestamp(s) for s in directs]) mention_stats.extend([describe_status_with_timestamp(s) for s in mentions]) except TwitterException, ex: warnings.append(str(ex))