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)
