commit ead771cb7ac8d8aacc71b2e6d62d465b1090ebd3
author Bruno Cardoso Lopes <bruno.cardoso@gmail.com> Fri May 13 21:31:32 2016 +0000
committer Bruno Cardoso Lopes <bruno.cardoso@gmail.com> Fri May 13 21:31:32 2016 +0000
tree 3e82535f023e9e95aecb2d78cab1ca5212f7d0fa
parent 862eaf8a12d62898469e1169c1bb5d1fa7b1e30b

[Support][Unittests] Add unittest for recursive_directory_iterator::level()

llvm-svn: 269488

llvm/unittests/Support/Path.cpp

diff --git a/llvm/unittests/Support/Path.cpp b/llvm/unittests/Support/Path.cpp
index fbeabe0..70bbf24 100644
--- a/llvm/unittests/Support/Path.cpp
+++ b/llvm/unittests/Support/Path.cpp
@@ -716,6 +716,20 @@
   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0/za1"));
   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0"));
   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive"));
+
+  // Test recursive_directory_iterator level()
+  ASSERT_NO_ERROR(
+      fs::create_directories(Twine(TestDirectory) + "/reclevel/a/b/c"));
+  fs::recursive_directory_iterator I(Twine(TestDirectory) + "/reclevel", ec), E;
+  for (int l = 0; I != E; I.increment(ec), ++l) {
+    ASSERT_NO_ERROR(ec);
+    EXPECT_EQ(I.level(), l);
+  }
+  EXPECT_EQ(I, E);
+  ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b/c"));
+  ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b"));
+  ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a"));
+  ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel"));
 }
 
 const char archive[] = "!<arch>\x0A";