def test_get_by_field_multiple(self): results = ResultsCollection() results.append(BondGirl1()) results.append(BondGirl2()) results.append(BondGirl3()) # Get by location london = results.get_by_field('location', 'London') self.assertEqual(len(london), 2)
def test_get_by_field_single(self): results = ResultsCollection() results.append(BondGirl1()) results.append(BondGirl2()) results.append(BondGirl3()) # Get by name honey = results.get_by_field('name', 'Honey Ryder') self.assertEqual(len(honey), 1) self.assertEqual(honey[0].name, 'Honey Ryder')
def run(self, access_token, storage=None, fql=None): """ Runs an FQLTask with the specified query - pass through the instance of `storage` """ if fql is None: fql = self.fql task = FqlTask(fql, access_token, storage) # Get the results results = task.get_results() # Assign the results to the result class results_collection = ResultsCollection() for result in results: try: results_collection.append(self.result_class(result)) except BlankFieldError: import ipdb ipdb.set_trace() pass return results_collection
def test_get_by_field_empty(self): results = ResultsCollection() # No items self.assertRaises(IndexError, results.get_by_field, 'name', 'Natalya Simonova')