def setUp(self):
user = User.objects.get(pk=1)
twitter_search = Pipe.objects.get(pk=1)
twitter_stream = Pipe.objects.get(pk=2)
instagram = Pipe.objects.get(pk=4)
self.search_faucet = Faucet(
endpoint=twitter_search,
user=user,
task='ADHOC_SRCH'
)
self.stream_faucet = Faucet(
endpoint=twitter_stream,
user=user,
task='BKGD_SRCH'
)
self.other_faucet = Faucet(
endpoint=instagram,
user=user,
task='OTHER_TASK'
)
self.stamp = Stamp.objects.get(pk=1)
term = SearchTerm.objects.get(pk=1)
self.query = ReservoirQuery(searchterms=[term])
def _setup_objects(obj):
"""
"""
obj.user = User.objects.get(pk=1)
obj.twitter_search = Pipe.objects.get(pk=1)
obj.twitter_stream = Pipe.objects.get(pk=2)
obj.instagram = Pipe.objects.get(pk=4)
obj.search_faucet = Faucet(
endpoint=obj.twitter_search,
user=obj.user,
task='ADHOC_SRCH'
)
obj.stream_faucet = Faucet(
endpoint=obj.twitter_stream,
user=obj.user,
task='BKGD_SRCH'
)
obj.other_faucet = Faucet(
endpoint=obj.instagram,
user=obj.user,
task='OTHER_TASK'
)
obj.stamp = Stamp.objects.get(pk=1)
term = SearchTerm.objects.get(pk=1)
obj.query = ReservoirQuery(searchterms=[term])
def _create_faucet():
"""
Helper method that returns an example ApiHandler.
"""
pipe = Pipe.objects.get_by_natural_key('twitter', 'SearchAPI')
user = User.objects.get(pk=1)
return Faucet(endpoint=pipe, user=user, task='BKGD_SRCH')
def setUp(self):
create_pumps(self)
pipe = Pipe.objects.get_by_natural_key('twitter', 'SearchAPI')
user = User.objects.get(pk=1)
self.faucet = Faucet(endpoint=pipe, user=user, task=BKGD_SRCH)
self.faucet.process_request = Mock() # avoid NotImplementedError
self.faucet.stop = Mock()
self.controller = StreamController(faucet=self.faucet,
query=self.query)
self.stream = self.controller.stream