[test] Support more markdown file extensions

tim-gromeyer · Nov 25, 2023 · 73f783e · 73f783e
1 parent 346fd6c
commit 73f783e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/main.cpp b/tests/main.cpp
@@ -173,9 +173,13 @@ int main(int argc, const char **argv) {
   // List to store all markdown files in this dir
   vector<string> files;
 
+  static vector<string> markdownExtensions = {".md", ".markdown", ".mkd"};
+
   // Find the files
   for (const auto &p : fs::recursive_directory_iterator(DIR)) {
-    if (p.path().extension() == ".md" && p.path().parent_path() == DIR)
+    if (std::find(markdownExtensions.begin(), markdownExtensions.end(),
+                  p.path().extension()) != markdownExtensions.end() &&
+        p.path().parent_path() == DIR)
       files.emplace_back(p.path().string());
   }