def test_reassign_all(self, mock_input):
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **self.options)
# Check that person_a has no entries and person_b has 3:
self.assertEqual(0, Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(3, Entry.objects.filter(speaker=self.person_b).count())
def test_reassign_all(self, mock_input):
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **self.options)
# Check that person_a has no entries and person_b has 3:
self.assertEqual(0,
Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(3,
Entry.objects.filter(speaker=self.person_b).count())
def test_reassign_with_from_date(self, mock_input):
options = self.options.copy()
options['date_from'] = '2012-01-01'
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **options)
# Check that person_a has 1 entry and person_b has 2:
self.assertEqual(1, Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(2, Entry.objects.filter(speaker=self.person_b).count())
def test_reassign_with_from_date(self, mock_input):
options = self.options.copy()
options['date_from'] = '2012-01-01'
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **options)
# Check that person_a has 1 entry and person_b has 2:
self.assertEqual(1,
Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(2,
Entry.objects.filter(speaker=self.person_b).count())
def test_reassign_all_with_slugs(self, mock_input):
options = {
'person_to': self.person_b.slug,
'person_from': self.person_a.slug,
'quiet': True
}
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **options)
# Check that person_a has no entries and person_b has 3:
self.assertEqual(0, Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(3, Entry.objects.filter(speaker=self.person_b).count())
def test_reassign_all_with_slugs(self, mock_input):
options = {
'person_to': self.person_b.slug,
'person_from': self.person_a.slug,
'quiet': True
}
with no_stdout_or_stderr():
call_command('hansard_reattribute_entries', **options)
# Check that person_a has no entries and person_b has 3:
self.assertEqual(0,
Entry.objects.filter(speaker=self.person_a).count())
self.assertEqual(3,
Entry.objects.filter(speaker=self.person_b).count())
