fix txt test

weni-ai · Mar 4, 2024 · 72b8b29 · 72b8b29
1 parent 616412e
commit 72b8b29
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)