def test_get_doc_counts_per_doc_type(self):
doc_types = self.migration.doc_types
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.source_db, doc_types), {
'CommCareUser': 2,
'CommCareUser-Deleted': 0,
'Group': 1,
'Group-Deleted': 1
})
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.target_db, doc_types), {
'CommCareUser': 0,
'CommCareUser-Deleted': 0,
'Group': 0,
'Group-Deleted': 0
})
self.migration.phase_1_bulk_migrate()
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.source_db, doc_types), {
'CommCareUser': 2,
'CommCareUser-Deleted': 0,
'Group': 1,
'Group-Deleted': 1
})
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.target_db, doc_types), {
'CommCareUser': 2,
'CommCareUser-Deleted': 0,
'Group': 1,
'Group-Deleted': 1
})
def get_doc_counts(self):
source_counts = get_doc_counts_per_doc_type(self.source_db,
self.doc_types)
target_counts = get_doc_counts_per_doc_type(self.target_db,
self.doc_types)
return [(self.source_db, source_counts),
(self.target_db, target_counts)]
def test_get_doc_counts_per_doc_type(self):
doc_types = self.migration.doc_types
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.source_db, doc_types),
{'CommCareUser': 2, 'CommCareUser-Deleted': 0, 'Group': 1, 'Group-Deleted': 1})
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.target_db, doc_types),
{'CommCareUser': 0, 'CommCareUser-Deleted': 0, 'Group': 0, 'Group-Deleted': 0})
self.migration.phase_1_bulk_migrate()
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.source_db, doc_types),
{'CommCareUser': 2, 'CommCareUser-Deleted': 0, 'Group': 1, 'Group-Deleted': 1})
self.assertEqual(
get_doc_counts_per_doc_type(self.migration.target_db, doc_types),
{'CommCareUser': 2, 'CommCareUser-Deleted': 0, 'Group': 1, 'Group-Deleted': 1})
def get_doc_counts(self):
source_counts = get_doc_counts_per_doc_type(self.source_db, self.doc_types)
target_counts = get_doc_counts_per_doc_type(self.target_db, self.doc_types)
return [(self.source_db, source_counts), (self.target_db, target_counts)]
