def test_works_onerow_noextras(self): self.cur.fetchall.return_value = [[sentinel.uniqueid]] results = group_manager._process_query(self.db_conn, self.query) self.assertEqual(len(results), 1) self.assertEqual(len(results[0].keys()), 1) self.assertEqual(results[0]['uniqueid'], sentinel.uniqueid)
def test_blows_up_with_bad_extras(self): indiv_array = [[sentinel.uniqueid, sentinel.field1, sentinel.field2]] self.cur.fetchall.return_value = indiv_array self.extras.append('field3') with self.assertRaises(IndexError): results = group_manager._process_query(self.db_conn, self.query, self.extras)
def test_works_onerow_extras(self): indiv_array = [[sentinel.uniqueid, sentinel.field1, sentinel.field2]] self.cur.fetchall.return_value = indiv_array results = group_manager._process_query(self.db_conn, self.query, self.extras) self.assertEqual(len(results), 1) self.assertEqual(len(results[0]), 3) self.assertIs(results[0]['field1'], sentinel.field1) self.assertIs(results[0]['field2'], sentinel.field2)
def test_works_multirows_noextras(self): id_array = [[sentinel.uniqueid1], [sentinel.uniqueid2], [sentinel.uniqueid3]] self.cur.fetchall.return_value = id_array results = group_manager._process_query(self.db_conn, self.query) self.assertEqual(len(results), 3) self.assertEqual(len(results[0].keys()), 1) for i in xrange(0,3): self.assertIs(results[i]['uniqueid'], id_array[i][0])
def test_works_multirows_extras(self): test_array = [[sentinel.uniqueid1, sentinel.field1_1, sentinel.field2_1], [sentinel.uniqueid2, sentinel.field1_2, sentinel.field2_2], [sentinel.uniqueid3, sentinel.field1_3, sentinel.field2_3]] self.cur.fetchall.return_value = test_array results = group_manager._process_query(self.db_conn, self.query, self.extras) self.assertEqual(len(results),3) self.assertEqual(len(results[0]), 3) for i in xrange(0,3): self.assertIs(results[i]['uniqueid'], test_array[i][0]) self.assertIs(results[i]['field1'], test_array[i][1]) self.assertIs(results[i]['field2'], test_array[i][2])
def test_works_norows_extras(self): self.cur.fetchall.return_value = list() results = group_manager._process_query(self.db_conn, self.query, self.extras) self.assertEqual(len(results), 0)