def _create_document(self, doc_type=None, party_code=None):
if not doc_type:
doc_type = self.default_doc_type
if not party_code:
party_code = self.default_party_code
new_file = _create_file()
document = Document(file=new_file,
bceid_user=self.user,
party_code=party_code,
doc_type=doc_type)
document.save()
return document
def test_storage_file_name_match(self, mock_redis_an, mock_redis_save):
mock_redis_an.return_value = 'file.txt'
mock_redis_save.return_value = 'file.txt'
file = SimpleUploadedFile('file.txt', b'this is some content')
test = Document()
test.file = file
test.doc_type = 'MC'
test.bceid_user = self.user
test.save()
self.assertTrue(mock_redis_save.called)
self.assertEqual(test.filename, test.file.name)
def test_storage_redis_delete(self, mock_redis_an, mock_redis_save,
mock_redis_delete):
mock_redis_an.return_value = '6061bebb-f2be-4a74-8757-c4063f6f6993_file_txt'
mock_redis_save.return_value = 'file.txt'
mock_redis_delete.return_value = True
file = SimpleUploadedFile('file.txt', b'this is some content')
test = Document()
test.file = file
test.doc_type = 'MC'
test.bceid_user = self.user
test.save()
test.delete()
self.assertTrue(mock_redis_delete.called)
def test_storage_redis_storage(self, mock_redis_an, mock_redis_save):
mock_redis_an.return_value = '6061bebb-f2be-4a74-8757-c4063f6f6993_file_txt'
mock_redis_save.return_value = 'file.txt'
file = SimpleUploadedFile('file.txt', b'this is some content')
test = Document()
test.file = file
test.doc_type = 'MC'
test.bceid_user = self.user
test.save()
self.assertTrue(mock_redis_save.called)
self.assertEqual(Document.objects.count(), 1)
test = Document.objects.get(id=test.id)
self.assertEqual(test.filename, 'file.txt')
self.assertNotEqual(test.file.name, 'file.txt')
def test_storage_connection_error(self, mock_redis):
mock_redis.side_effect = ConnectionError()
original_count = Document.objects.count()
connection_error = False
try:
file = SimpleUploadedFile('file.txt', b'this is some content')
test = Document()
test.file = file
test.doc_type = 'MC'
test.bceid_user = self.user
test.save()
except ConnectionError:
connection_error = True
self.assertTrue(connection_error)
self.assertEqual(Document.objects.count(), original_count)