def test_location_add_file_with_category(self) -> None:
temp_file_path = os.path.join(self.tmpdir,
".".join(["temp_file", "txt"]))
with open(temp_file_path, "wb") as tmp_file:
tmp_file.write(b"TEST DATA FILE")
add_file(self.client, temp_file_path, "LOCATION", self.location_1.id,
"test_category")
docs = get_location_documents(self.client, self.location_1)
for doc in docs:
self.assertEqual(doc.category, "test_category")
delete_document(self.client, doc)
def test_location_add_file(self) -> None:
temp_file_path = os.path.join(self.tmpdir, ".".join(["temp_file", "txt"]))
with open(temp_file_path, "wb") as tmp_file:
tmp_file.write(b"TEST DATA FILE")
add_file(
client=self.client,
local_file_path=temp_file_path,
entity_type="LOCATION",
entity_id=self.location_1.id,
)
docs = get_location_documents(client=self.client, location=self.location_1)
self.assertEqual(len(docs), 1)
for doc in docs:
delete_document(self.client, doc)
