def test_swap_all_to_use_new_metadata_row(self):
"""
Test what happens when we want to set the 'New' row to the 'Current' row in a
normal scenario (e.g. when rows exist in all three import_states).
"""
original_new = factories.SDNFallbackMetadataFactory.create(
import_state="New")
original_current = factories.SDNFallbackMetadataFactory.create(
import_state="Current")
original_discard = factories.SDNFallbackMetadataFactory.create(
import_state="Discard")
SDNFallbackMetadata.swap_all_states()
self.assertEqual(len(SDNFallbackMetadata.objects.all()), 2)
former_new_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original_new.file_checksum)[0]
self.assertEqual(former_new_metadata.import_state, 'Current')
former_current_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original_current.file_checksum)[0]
self.assertEqual(former_current_metadata.import_state, 'Discard')
former_discard_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original_discard.file_checksum)
self.assertEqual(len(former_discard_metadata), 0)
def test_swap_all_rollback(self):
"""
Make sure that the rollback works if there are issues when swapping all of the rows.
"""
original_current = factories.SDNFallbackMetadataFactory.create(
import_state="Current")
original_discard = factories.SDNFallbackMetadataFactory.create(
import_state="Discard")
expected_logs = [
(self.LOGGER_NAME, 'WARNING',
"Expected a row in the 'Current' import_state after swapping, but there are none."
),
]
with self.assertRaises(SDNFallbackMetadata.DoesNotExist):
with LogCapture(self.LOGGER_NAME) as log:
SDNFallbackMetadata.swap_all_states()
log.check_present(*expected_logs)
self.assertEqual(len(SDNFallbackMetadata.objects.all()), 2)
former_current_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original_current.file_checksum)[0]
self.assertEqual(former_current_metadata.import_state, 'Current')
former_discard_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original_discard.file_checksum)[0]
self.assertEqual(former_discard_metadata.import_state, 'Discard')
def test_swap_discard_row(self):
"""Discard row gets deleted when swapping rows."""
factories.SDNFallbackMetadataFactory.create(import_state="Discard")
SDNFallbackMetadata.swap_all_states()
actual_rows = SDNFallbackMetadata.objects.all()
self.assertEqual(len(actual_rows), 0)
def test_swap_new_row(self):
"""Swap New row to Current row."""
factories.SDNFallbackMetadataFactory.create(import_state='New')
SDNFallbackMetadata.swap_all_states()
actual_rows = SDNFallbackMetadata.objects.all()
self.assertEqual(len(actual_rows), 1)
self.assertEqual(actual_rows[0].import_state, 'Current')
def test_swap_current_row(self):
"""Swap Current row to Discard row."""
original = factories.SDNFallbackMetadataFactory.create(import_state="Current")
# this is needed to bypass the requirement to always have a 'Current'
factories.SDNFallbackMetadataFactory.create(import_state="New")
SDNFallbackMetadata.swap_all_states()
actual_row = SDNFallbackMetadata.objects.filter(file_checksum=original.file_checksum)[0]
self.assertEqual(actual_row.import_state, 'Discard')
def test_swap_twice_one_row(self):
"""Swapping one row twice without adding a new file should result in an error."""
original = factories.SDNFallbackMetadataFactory.create(
import_state="New")
expected_logs = [
(self.LOGGER_NAME, 'WARNING',
"Expected a row in the 'Current' import_state after swapping, but there are none."
),
]
SDNFallbackMetadata.swap_all_states()
with self.assertRaises(SDNFallbackMetadata.DoesNotExist):
with LogCapture(self.LOGGER_NAME) as log:
SDNFallbackMetadata.swap_all_states()
log.check_present(*expected_logs)
self.assertEqual(len(SDNFallbackMetadata.objects.all()), 1)
former_new_metadata = SDNFallbackMetadata.objects.filter(
file_checksum=original.file_checksum)[0]
self.assertEqual(former_new_metadata.import_state, 'Current')
def populate_sdn_fallback_metadata(sdn_csv_string):
"""
Insert a new SDNFallbackMetadata entry if the new csv differs from the current one
Args:
sdn_csv_string (bytes): Bytes of the sdn csv
Returns:
sdn_fallback_metadata_entry (SDNFallbackMetadata): Instance of the current SDNFallbackMetadata class
or None if none exists
"""
file_checksum = hashlib.sha256(sdn_csv_string.encode('utf-8')).hexdigest()
metadata_entry = SDNFallbackMetadata.insert_new_sdn_fallback_metadata_entry(file_checksum)
return metadata_entry
def test_minimum_requirements(self):
"""Make sure the row is created correctly with the minimum dataset + defaults."""
new_metadata = SDNFallbackMetadata(
file_checksum="foobar",
download_timestamp=datetime.now(),
)
new_metadata.full_clean()
new_metadata.save()
self.assertEqual(len(SDNFallbackMetadata.objects.all()), 1)
actual_metadata = SDNFallbackMetadata.objects.all()[0]
self.assertEqual(actual_metadata.file_checksum, "foobar")
self.assertIsInstance(actual_metadata.download_timestamp, datetime)
self.assertEqual(actual_metadata.import_timestamp, None)
self.assertEqual(actual_metadata.import_state, 'New')
self.assertIsInstance(actual_metadata.created, datetime)
self.assertIsInstance(actual_metadata.modified, datetime)
def test_swap_all_non_existent_rows(self):
"""Swapping all shouldn't break / do anything if there are no existing rows."""
SDNFallbackMetadata.swap_all_states()
self.assertEqual(len(SDNFallbackMetadata.objects.all()), 0)
