Ada-C15 · CheezItMan · Aug 8, 2019 · Aug 8, 2019 · Aug 13, 2021 · Aug 13, 2021
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+__pycache__/
diff --git a/find_xml_file/__init__.py b/find_xml_file/__init__.py
diff --git a/find_xml_file/find_xml_file.py b/find_xml_file/find_xml_file.py
@@ -0,0 +1,14 @@
+import os
+
+
+def find_xml_file(folder):
+    files = os.listdir(folder)
+
+    xml_file = None
+
+    for filename in files:
+        if filename.endswith(".xml"):
+            xml_file = filename
+            break
+
+    return xml_file
diff --git a/test.sh b/test.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+pytest
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_data/example.xml b/tests/test_data/example.xml
diff --git a/tests/test_find_xml_file.py b/tests/test_find_xml_file.py
@@ -0,0 +1,10 @@
+from find_xml_file.find_xml_file import find_xml_file
+
+
+def test_no_files_returns_None():
+    assert find_xml_file(".") is None
+
+
+def test_finds_existing_xml_file():
+    PATH_TO_TEST_XML = "./test/test_data"
+    assert find_xml_file(PATH_TO_TEST_XML) == "example.xml"