def test_get_success_content_disposition(self):
self.arquivo = ArquivoFactory(
arquivo__filename='test_get_success_invalid_uuid.txt')
response = self.client.get(self.__get_arquivo_url(self.arquivo.uuid))
self.assertEqual(
'attachment; filename=test_get_success_invalid_uuid.txt',
response.get('Content-Disposition'))
def setUp(self, mock_fieldfile_save):
self.data = {
'arquivo': SimpleUploadedFile('data.txt', 'content data'.encode())
}
self.arquivo = ArquivoFactory()
self.serializer = ArquivoSerializer(instance=self.arquivo)
def test_str(self, mock_storage):
name = 'darth-vader.txt'
arquivo = ArquivoFactory(arquivo__filename=name)
self.assertEqual(name, str(arquivo))
mock_storage.assert_called_once()
def test_get_success_content(self):
self.arquivo = ArquivoFactory(arquivo__data='abcdef')
response = self.client.get(self.__get_arquivo_url(self.arquivo.uuid))
self.assertEqual(b'abcdef', response.getvalue())
def test_get_success_content_length(self):
self.arquivo = ArquivoFactory(arquivo__data='abc')
response = self.client.get(self.__get_arquivo_url(self.arquivo.uuid))
self.assertEqual('3', response.get('Content-Length'))
def test_get_404_file_not_exist_on_disk(self, mock_fieldfile_save):
arquivo = ArquivoFactory(
arquivo__filename='test_get_404_file_not_exist_on_disk.txt')
response = self.client.get(self.__get_arquivo_url(arquivo.uuid))
self.assertEqual(404, response.status_code)
mock_fieldfile_save.assert_called_once()
def test_get_status_code_200(self):
self.arquivo = ArquivoFactory()
response = self.client.get(self.__get_arquivo_url(self.arquivo.uuid))
self.assertEqual(200, response.status_code)
def test_get_success_result_keys(self, mock_fieldfile_save):
arquivo = ArquivoFactory()
response = self.client.get(self.__get_arquivo_url(arquivo.uuid))
self.assertCountEqual(response.data.keys(),
['uuid', 'created_at', 'updated_at', 'arquivo'])
mock_fieldfile_save.assert_called_once()
def test_get_status_code_200(self, mock_fieldfile_save):
arquivo = ArquivoFactory()
response = self.client.get(self.__get_arquivo_url(arquivo.uuid))
self.assertEqual(200, response.status_code)
mock_fieldfile_save.assert_called_once()