def testSetContentTypeFromFile(self):
"""Tests that content type is correctly determined for symlinks."""
if system_util.IS_WINDOWS:
return unittest.skip('use_magicfile features not available on Windows')
surprise_html = b'<html><body>And you thought I was just text!</body></html>'
temp_dir_path = self.CreateTempDir()
txt_file_path = self.CreateTempFile(tmpdir=temp_dir_path,
contents=surprise_html,
file_name='html_in_disguise.txt')
link_name = 'link_to_realfile' # Notice no file extension was supplied.
os.symlink(txt_file_path, temp_dir_path + os.path.sep + link_name)
# Content-type of a symlink should be obtained from the link's target.
dst_obj_metadata_mock = mock.MagicMock(contentType=None)
src_url_stub = mock.MagicMock(object_name=temp_dir_path + os.path.sep +
link_name,
**{
'IsFileUrl.return_value': True,
'IsStream.return_value': False,
'IsFifo.return_value': False
})
# The file command should detect HTML in the real file.
with SetBotoConfigForTest([('GSUtil', 'use_magicfile', 'True')]):
_SetContentTypeFromFile(src_url_stub, dst_obj_metadata_mock)
self.assertEqual('text/html; charset=us-ascii',
dst_obj_metadata_mock.contentType)
dst_obj_metadata_mock = mock.MagicMock(contentType=None)
# The mimetypes module should guess based on the real file's extension.
with SetBotoConfigForTest([('GSUtil', 'use_magicfile', 'False')]):
_SetContentTypeFromFile(src_url_stub, dst_obj_metadata_mock)
self.assertEqual('text/plain', dst_obj_metadata_mock.contentType)
def testSetsContentTypesForCommonFileExtensionsCorrectly(self):
extension_rules = copy_helper.COMMON_EXTENSION_RULES.items()
for extension, expected_content_type in extension_rules:
dst_obj_metadata_mock = mock.MagicMock(contentType=None)
src_url_stub = mock.MagicMock(object_name='file.' + extension)
src_url_stub.IsFileUrl.return_value = True
src_url_stub.IsStream.return_value = False
src_url_stub.IsFifo.return_value = False
_SetContentTypeFromFile(src_url_stub, dst_obj_metadata_mock)
self.assertEqual(expected_content_type, dst_obj_metadata_mock.contentType)
