def test_get_resource_types(self):
command = DeleteRedundantStaticCommand()
expected = {
RESOURCE_TYPES['RAW'], RESOURCE_TYPES['IMAGE'],
RESOURCE_TYPES['VIDEO']
}
self.assertEqual(expected, command.get_resource_types())
def test_get_needful_files_with_keep_unhashed_files_false(self):
command = DeleteRedundantStaticCommand()
command.keep_unhashed_files = False
expected_response = {
'static/tests/css/style.{}.css'.format(self.style_hash),
'static/tests/images/dummy-static-image.{}'.format(
self.image_hash) # removed jpg extension
}
self.assertEqual(command.get_needful_files(), expected_response)
def test_command_removes_not_needed_file(self):
command = DeleteRedundantStaticCommand()
uploaded_resources = [(RESOURCE_TYPES['RAW'], ['style1.css']),
(RESOURCE_TYPES['IMAGE'], []),
(RESOURCE_TYPES['VIDEO'], [])]
needful_files = {'style.css'}
with mock.patch.object(command,
'get_uploaded_resources',
return_value=uploaded_resources):
with mock.patch.object(command,
'get_needful_files',
return_value=needful_files):
files_to_remove = command.get_files_to_remove()
expected_response = {
RESOURCE_TYPES['RAW']: {'style1.css'},
RESOURCE_TYPES['IMAGE']: set(),
RESOURCE_TYPES['VIDEO']: set()
}
self.assertEqual(files_to_remove, expected_response)
def test_get_exclude_paths_returns_empty_tuple(self):
command = DeleteRedundantStaticCommand()
self.assertEqual(command.get_exclude_paths(), ())
def test_get_file_storage(self):
command = DeleteRedundantStaticCommand()
storage = command.get_file_storage(RESOURCE_TYPES['IMAGE'])
self.assertEqual(storage, command.storage)