MITLibraries
diff --git a/‎Pipfile.lock‎
Lines changed: 48 additions & 16 deletions b/‎Pipfile.lock‎
Lines changed: 48 additions & 16 deletions
diff --git a/‎tests/test_models.py‎
Lines changed: 15 additions & 3 deletions b/‎tests/test_models.py‎
Lines changed: 15 additions & 3 deletions
@@ -1,3 +1,5 @@
+import os
+
 import attr
 import pytest
 import requests_mock
@@ -60,6 +62,7 @@ def test_filtered_item_search(client):
 
 
 def test__pop_inst(client):
+    """Test _pop_inst function."""
     class_type = models.Collection
     rec_obj = {'name': 'Test title', 'type': 'collection', 'items': []}
     rec_obj = client._pop_inst(class_type, rec_obj)
@@ -68,18 +71,27 @@ def test__pop_inst(client):
 
 
 def test__build_uuid_list(client):
+    """Test _build_uuid_list function."""
     rec_obj = {'items': [{'uuid': '1234'}]}
     children = 'items'
     child_list = client._build_uuid_list(rec_obj, children)
     assert '1234' in child_list
 
 
-# def test_build_file_list_local():
-#     file_list = models.build_file_list_local(directory, file_extension)
-#     assert False
+def test_build_file_list_local():
+    """Test filtered_item_search function."""
+    file_extension = 'pdf'
+    directory = 'test_temp'
+    os.mkdir(directory)
+    open(f'{directory}/999.pdf', 'w')
+    file_list = models.build_file_list_local(directory, file_extension)
+    os.remove(f'{directory}/999.pdf')
+    os.rmdir(directory)
+    assert '999.pdf' in file_list
 
 
 def test_build_file_list_remote():
+    """Test build_file_list_remote function."""
     content = '<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>'
     content += '<head><title>Index of /pdf</title></head><body><h1>Index of /'
     content += 'pdf</h1><table><tr><th>Name</th><th>Last modified</th><th>'