1 files changed, 37 insertions, 0 deletions
diff --git a/unittests/ADT/PostOrderIteratorTest.cpp b/unittests/ADT/PostOrderIteratorTest.cpp
new file mode 100644
index 0000000..1da1078
--- /dev/null
+++ b/unittests/ADT/PostOrderIteratorTest.cpp
@@ -0,0 +1,37 @@
+//===- PostOrderIteratorTest.cpp - PostOrderIterator unit tests -----------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+#include "gtest/gtest.h"
+#include "llvm/ADT/PostOrderIterator.h"
+#include "llvm/IR/BasicBlock.h"
+#include "llvm/IR/CFG.h"
+using namespace llvm;
+
+namespace {
+
+// Whether we're able to compile
+TEST(PostOrderIteratorTest, Compiles) {
+  typedef SmallPtrSet<void *, 4> ExtSetTy;
+
+  // Tests that template specializations are kept up to date
+  void *Null = nullptr;
+  po_iterator_storage<std::set<void *>, false> PIS;
+  PIS.insertEdge(Null, Null);
+  ExtSetTy Ext;
+  po_iterator_storage<ExtSetTy, true> PISExt(Ext);
+  PIS.insertEdge(Null, Null);
+
+  // Test above, but going through po_iterator (which inherits from template
+  // base)
+  BasicBlock *NullBB = nullptr;
+  auto PI = po_end(NullBB);
+  PI.insertEdge(NullBB, NullBB);
+  auto PIExt = po_ext_end(NullBB, Ext);
+  PIExt.insertEdge(NullBB, NullBB);
+}
+}