def test_default_or_logic(self):
"""
Tests the _handle_trm_loc_logic method for an API tha joins SearchTerms
and Locations using OR logic.
"""
self.query.trm_loc_logic = 'OR'
specs = PipeSpecSheet.objects.get(pipe=2) # Twitter Public Streams API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_loc_logic([self.query])
self.assertEqual(len(queries), 1)
def test_query_and_logic(self):
"""
Tests the _handle_trm_loc_logic method for a query that joins
SearchTerms and Locations with AND logic.
"""
self.query.trm_loc_logic = 'AND'
specs = PipeSpecSheet.objects.get(pipe=1) # Twitter Search API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_loc_logic([self.query])
self.assertEqual(len(queries), 1)
def test_default_and_logic_multi_qs(self):
"""
Tests the _handle_trm_loc_logic method for multiple OR queries with an
API that joins SearchTerms and Locations using AND logic.
"""
self.query.trm_loc_logic = 'OR'
specs = PipeSpecSheet.objects.get(pipe=1) # Twitter Search API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_loc_logic([self.query, self.query])
self.assertEqual(len(queries), 4)
