def test_default_or_logic(self):
"""
Tests the _handle_trm_trm_logic method for API with OR operator.
"""
specs = PipeSpecSheet.objects.get(pipe=1) # Twitter Search API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_trm_logic([self.query])
self.assertEqual(len(queries), 1)
def test_no_or_logic(self):
"""
Tests the _handle_trm_trm_logic method for API without OR logic.
"""
specs = PipeSpecSheet.objects.get(pipe=4) # Instagram API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_trm_logic([self.query])
self.assertEqual(len(queries), 3)
def test_default_or_logic(self):
"""
Tests the _handle_trm_trm_logic method for API with default OR logic.
"""
specs = PipeSpecSheet.objects.get(pipe=2) # Twitter Public Streams API
engineer = Engineer(query=self.query, specs=specs)
queries = engineer._handle_trm_trm_logic([self.query])
self.assertEqual(len(queries), 1)
