def _assert_rows_in_ucr_table(self, config, rows): results = list( load_data_from_db(get_table_name(self.domain.name, config.table_id))) actual_rows = [{key: r[key] for key in rows[0]} for r in results] self.assertEqual(sorted(rows, key=lambda x: x['doc_id']), sorted(actual_rows, key=lambda x: x['doc_id']))
def _assert_rows_in_ucr_table(self, config, rows): results = list(load_data_from_db(get_table_name(self.domain.name, config.table_id))) if not rows: self.assertEqual(results, []) return actual_rows = [{key: r[key] for key in rows[0]} for r in results] self.assertEqual( sorted(rows, key=lambda x: x['doc_id']), sorted(actual_rows, key=lambda x: x['doc_id']) )
def test_bulk_save(self): docs = [] for i in range(10): docs.append({ "_id": str(i), "domain": self.domain, "doc_type": "CommCareCase", "name": 'doc_name_' + str(i) }) self.adapter.build_table() self.adapter.bulk_save(docs) results = list(load_data_from_db(get_table_name(self.domain, self.config.table_id))) self.assertEqual(len(results), 10)