1 files changed, 61 insertions, 0 deletions
diff --git a/unittests/DriverTests/DriverTest.h b/unittests/DriverTests/DriverTest.h
new file mode 100644
index 0000000000000..2349132ee2ce7
--- /dev/null
+++ b/unittests/DriverTests/DriverTest.h
@@ -0,0 +1,61 @@
+//===- lld/unittest/DriverTest.h ------------------------------------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "lld/Driver/Driver.h"
+#include "llvm/Support/raw_ostream.h"
+#include "gtest/gtest.h"
+#include <stdarg.h>
+
+namespace {
+
+using namespace llvm;
+using namespace lld;
+
+template<typename D, typename T>
+class ParserTest : public testing::Test {
+protected:
+
+  virtual const LinkingContext *linkingContext() = 0;
+
+  std::string &errorMessage() { return  _errorMessage; }
+
+  // Convenience method for getting number of input files.
+  int inputFileCount() {
+    return linkingContext()->getNodes().size();
+  }
+
+  // Convenience method for getting i'th input files name.
+  std::string inputFile(int index) {
+    Node &node = *linkingContext()->getNodes()[index];
+    if (node.kind() == Node::Kind::File)
+      return cast<FileNode>(&node)->getFile()->path();
+    llvm_unreachable("not handling other types of input files");
+  }
+
+  // For unit tests to call driver with various command lines.
+  bool parse(const char *args, ...) {
+    // Construct command line options from varargs.
+    std::vector<const char *> vec;
+    vec.push_back(args);
+    va_list ap;
+    va_start(ap, args);
+    while (const char *arg = va_arg(ap, const char *))
+      vec.push_back(arg);
+    va_end(ap);
+
+    // Call the parser.
+    raw_string_ostream os(_errorMessage);
+    return D::parse(vec.size(), &vec[0], _ctx, os);
+  }
+
+  T _ctx;
+  std::string _errorMessage;
+};
+
+}