def test_revision_generator_notices(self, init, merge_changes): init.return_value = None b = Builder() # Don't need parameters as init's been mocked out aaaa = {'document_number': 'aaaa', 'effective_on': '2012-12-12', 'publication_date': '2011-11-11', 'changes': []} bbbb = {'document_number': 'bbbb', 'effective_on': '2012-12-12', 'publication_date': '2011-11-12', 'changes': []} cccc = {'document_number': 'cccc', 'effective_on': '2013-01-01', 'publication_date': '2012-01-01', 'changes': []} b.notices = [aaaa, bbbb, cccc] b.eff_notices = {'2012-12-12': [aaaa, bbbb], '2013-01-01': [cccc]} b.doc_number = 'aaaa' b.checkpointer = NullCheckpointer() tree = Node(label=['1111']) version_list = [] notice_lists = [] for notice, _, _, notices in b.revision_generator(tree): version_list.append(notice['document_number']) notice_lists.append(notices) self.assertEqual(['bbbb', 'cccc'], version_list) self.assertEqual(2, len(notice_lists)) self.assertEqual(2, len(notice_lists[0])) self.assertTrue(aaaa in notice_lists[0]) self.assertTrue(bbbb in notice_lists[0]) self.assertEqual(3, len(notice_lists[1])) self.assertTrue(aaaa in notice_lists[1]) self.assertTrue(bbbb in notice_lists[1]) self.assertTrue(cccc in notice_lists[1])
def test_revision_generator_notices(self, init, merge_changes): init.return_value = None b = Builder() # Don't need parameters as init's been mocked out aaaa = {'document_number': 'aaaa', 'effective_on': '2012-12-12', 'publication_date': '2011-11-11', 'changes': []} bbbb = {'document_number': 'bbbb', 'effective_on': '2012-12-12', 'publication_date': '2011-11-12', 'changes': []} cccc = {'document_number': 'cccc', 'effective_on': '2013-01-01', 'publication_date': '2012-01-01', 'changes': []} b.notices = [aaaa, bbbb, cccc] b.eff_notices = {'2012-12-12': [aaaa, bbbb], '2013-01-01': [cccc]} b.doc_number = 'aaaa' tree = Node(label=['1111']) version_list = [] notice_lists = [] for notice, _, _, notices in b.revision_generator(tree): version_list.append(notice['document_number']) notice_lists.append(notices) self.assertEqual(['bbbb', 'cccc'], version_list) self.assertEqual(2, len(notice_lists)) self.assertEqual(2, len(notice_lists[0])) self.assertTrue(aaaa in notice_lists[0]) self.assertTrue(bbbb in notice_lists[0]) self.assertEqual(3, len(notice_lists[1])) self.assertTrue(aaaa in notice_lists[1]) self.assertTrue(bbbb in notice_lists[1]) self.assertTrue(cccc in notice_lists[1])
def test_changes_in_sequence_skips(self, init, merge_changes): """Skips over notices which occurred _before_ our starting point""" init.return_value = None b = Builder() # Don't need parameters as init's been mocked out aaaa = {'document_number': 'aaaa', 'effective_on': '2012-12-12', 'publication_date': '2011-11-11', 'changes': []} bbbb = {'document_number': 'bbbb', 'effective_on': '2012-12-12', 'publication_date': '2011-11-12', 'changes': []} cccc = {'document_number': 'cccc', 'effective_on': '2013-01-01', 'publication_date': '2012-01-01', 'changes': []} b.eff_notices = {'2012-12-12': [aaaa, bbbb], '2013-01-01': [cccc]} b.doc_number = bbbb['document_number'] changes = list(b.changes_in_sequence()) self.assertEqual(len(changes), 1) self.assertEqual(cccc['document_number'], changes[0][0]) self.assertEqual(cccc['document_number'], merge_changes.call_args[0][0])