def _assert_duplicates_are_deleted(self, object_type):
# Create a user, device and two duplicate reports
user = Dummy.create_user()
device = Dummy.create_device(user)
report_1 = Dummy.create_report(object_type, device)
Dummy.create_report(object_type, device)
# Assert that 2 instances have been created
self.assertEqual(object_type.objects.count(), 2)
# Run the migration
logger = logging.getLogger("crashreports")
with self.assertLogs(logger, "DEBUG") as logging_watcher:
self.migrate_to_dest()
# Assert the correct message is logged
self.assertTrue({
"INFO:crashreports.migrations."
"0006_add_unique_constraints_and_drop_duplicates:"
"Found 1 {} instances that have duplicates. "
"These will be removed.".format(object_type.__name__),
"DEBUG:crashreports.migrations"
".0006_add_unique_constraints_and_drop_duplicates:Removing "
"duplicates: {}".format(
str({
"device": device.id,
"date": report_1.date,
"min_id": report_1.id,
"num_duplicates": 2,
})),
}.issubset(set(logging_watcher.output)))
# Assert that only one instance is left in the database
self.assertEqual(object_type.objects.count(), 1)
def test_change_of_date_field_type(self):
"""Test that the 'date' field of heartbeats is changed to a date."""
# Create a user, device and a heartbeat
user = Dummy.create_user()
device = Dummy.create_device(user)
heartbeat_timestamp = datetime(2015,
12,
15,
1,
23,
45,
tzinfo=pytz.utc)
heartbeat = Dummy.create_report(HeartBeat,
device,
date=heartbeat_timestamp)
# Assert that the date is of type datetime
self.assertIsInstance(heartbeat.date, datetime)
# Run the migration
self.migrate_to_dest()
# Assert that the date is now of type date and has the correct value
heartbeat = HeartBeat.objects.first()
self.assertIsInstance(heartbeat.date, date)
self.assertEqual(heartbeat.date, heartbeat_timestamp.date())
def test_logfile_deletion(self):
"""Test deletion of logfile instances."""
# Create a user, device and crashreport with logfile
device = Dummy.create_device(Dummy.create_user())
crashreport = Dummy.create_report(Crashreport, device)
logfile, logfile_path = Dummy.create_log_file_with_actual_file(
crashreport)
# Assert that the crashreport and logfile have been created
self.assertEqual(Crashreport.objects.count(), 1)
self.assertEqual(LogFile.objects.count(), 1)
self.assertTrue(os.path.isfile(logfile_path))
# Delete the logfile
response = self.fp_staff_client.delete(
reverse(self.POST_LOGFILE_URL, args=[logfile.id]))
self.assertEqual(status.HTTP_204_NO_CONTENT, response.status_code)
# Assert that the logfile has been deleted
self.assertEqual(LogFile.objects.count(), 0)
self.assertFalse(os.path.isfile(logfile_path))
def _assert_duplicate_entries_can_not_be_created(self, object_type):
# Create a user, device and a report
user = Dummy.create_user()
device = Dummy.create_device(user)
Dummy.create_report(object_type, device)
# Assert creating a duplicate report fails
logger = logging.getLogger("crashreports")
with self.assertLogs(logger, "DEBUG") as logging_watcher:
report = Dummy.create_report(object_type, device)
self.assertEqual(
logging_watcher.output,
[
"DEBUG:crashreports.models:"
"Duplicate {} received and dropped: {}".format(
object_type.__name__, str(model_to_dict(report))
)
],
)
self.assertEqual(object_type.objects.count(), 1)