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)