diff --git a/app/tests/test_document_loader.py b/app/tests/test_document_loader.py index 262e325..37d0862 100644 --- a/app/tests/test_document_loader.py +++ b/app/tests/test_document_loader.py @@ -160,6 +160,7 @@ def test_docx_loader_cls(self): split_pages: List[Document] = docx_loader.load_and_split_text(self.text_splitter) self.assertEqual(list, type(split_pages)) + @mock.patch.dict(os.environ, {"AWS_STORAGE_BUCKET_NAME": "file-path"}) def test_txt_loader_cls(self): file_path = f'{self.path}/{self.file_name}.txt' docx_loader = TxtLoader(file_path)