def test_migrate_version(self, mock_test_container, mock_prod_container):
mock_test_object = mock.Mock()
mock_test_container.get_object.return_value = mock_test_object
version = FileVersionFactory()
version.location['container'] = TEST_CONTAINER_NAME
version.save()
migrate_version(version)
mock_test_container.get_object.assert_called_with(version.location['object'])
mock_test_object.copy.assert_called_with(mock_prod_container)
version.reload()
assert_equal(version.location['container'], PROD_CONTAINER_NAME)
assert_equal(len(get_targets()), 0)
def test_migrate_version(self, mock_test_container, mock_prod_container):
mock_test_object = mock.Mock()
mock_test_container.get_object.return_value = mock_test_object
version = FileVersionFactory()
version.location['container'] = TEST_CONTAINER_NAME
version.save()
migrate_version(version)
mock_test_container.get_object.assert_called_with(
version.location['object'])
mock_test_object.copy.assert_called_with(mock_prod_container)
version.reload()
assert_equal(version.location['container'], PROD_CONTAINER_NAME)
assert_equal(len(get_targets()), 0)
def test_ensure_glacier_exists(self, mock_vault, mock_download):
version = FileVersionFactory()
version.metadata['archive'] = 'foo'
version.save()
ensure_glacier(version, dry_run=False)
assert_false(mock_vault.upload_archive.called)
def test_ensure_glacier_exists(self, mock_vault, mock_download):
version = FileVersionFactory()
version.metadata['archive'] = 'foo'
version.save()
ensure_glacier(version, dry_run=False)
assert_false(mock_vault.upload_archive.called)
