def setUp(self, *args, **kwargs): yield super(TestFoursquareWeb, self).setUp(*args, **kwargs) self.service = FoursquareWeb(datastore=self.app.db) self.daemon = self.service.daemon_object() recorder.wrap(self.service.actually_request) recorder.wrap(self.daemon.actually_request) recorder.play()
def test_response_filter(self): ms = MockService() recorder.use_filters = True recorder.wrap(ms.actually_request) recorder.record() args = [] kwargs = {"url": "http://example.com/year"} resp = yield ms.request(**kwargs) self.assertEqual(resp["year"], 2013) recorder.stop() recorder.save() resp = yield ms.request(url="http://example.com/year") self.assertEqual(resp["year"], 3013) # we must turn off filters here, or we'll get strange database errors on setup for other # tests recorder.use_filters = False
def setUp(self): yield super(TestRedditWeb, self).setUp() self.service = RedditWebMock( datastore=self.app.db, ) self.daemon = self.service.daemon_object() recorder.wrap(self.service.actually_request) recorder.wrap(self.daemon.actually_request) recorder.wrap(requests.request)
def setUp(self, *args, **kwargs): yield super(TestFlickrWeb, self).setUp(*args, **kwargs) self.service = FlickrWeb( datastore=self.app.db, ) self.daemon = self.service.daemon_object() recorder.wrap(self.service.actually_request) recorder.wrap(self.daemon.actually_request) recorder.wrap(requests.request)
from bs4 import BeautifulSoup from lightning.recorder import recorder from lightning.service.base import Profile import logging import os import time import re import requests from subprocess import Popen, PIPE, STDOUT from twistedpyres import ResQ USER_AGENT = dict( firefox='Mozilla/5.0 (X11; Ubuntu; Linux i686; rv:13.0) Gecko/20100101 Firefox/13.0.1', ) recorder.wrap(requests.request) class TestService(TestLightning): def _test_method_result_keys(self, method_name, result, expected_result=None): """ This method verifies that the result value passed in matches the values that we expect for the given method. For example, we could test the method return value of profile. This method will verify that all the profile keys (first_name, last_name, etc) are contained in the return value. method_name: the method to compare the result again result: the result to test against the expected result """