1 files changed, 53 insertions, 0 deletions
diff --git a/src/test/cpp/test_util.cc b/src/test/cpp/test_util.cc
new file mode 100644
index 0000000000..b50f74d79c
--- /dev/null
+++ b/src/test/cpp/test_util.cc
@@ -0,0 +1,53 @@
+// Copyright 2018 The Bazel Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+#include "src/test/cpp/test_util.h"
+
+#include "src/main/cpp/startup_options.h"
+#include "googletest/include/gtest/gtest.h"
+
+namespace blaze {
+
+void ExpectIsNullaryOption(const StartupOptions* options,
+                           const std::string& flag_name) {
+  EXPECT_TRUE(options->IsNullary("--" + flag_name));
+  EXPECT_TRUE(options->IsNullary("--no" + flag_name));
+
+  EXPECT_FALSE(options->IsNullary("--" + flag_name + "__invalid"));
+
+  EXPECT_DEATH(options->IsNullary("--" + flag_name + "=foo"),
+               ("In argument '--" + flag_name + "=foo': option '--" +
+                flag_name + "' does not take a value")
+                   .c_str());
+
+  EXPECT_DEATH(options->IsNullary("--no" + flag_name + "=foo"),
+               ("In argument '--no" + flag_name + "=foo': option '--no" +
+                flag_name + "' does not take a value")
+                   .c_str());
+
+  EXPECT_FALSE(options->IsUnary("--" + flag_name));
+  EXPECT_FALSE(options->IsUnary("--no" + flag_name));
+}
+
+void ExpectIsUnaryOption(const StartupOptions* options,
+                         const std::string& flag_name) {
+  EXPECT_TRUE(options->IsUnary("--" + flag_name));
+  EXPECT_TRUE(options->IsUnary("--" + flag_name + "="));
+  EXPECT_TRUE(options->IsUnary("--" + flag_name + "=foo"));
+
+  EXPECT_FALSE(options->IsUnary("--" + flag_name + "__invalid"));
+  EXPECT_FALSE(options->IsNullary("--" + flag_name));
+  EXPECT_FALSE(options->IsNullary("--no" + flag_name));
+}
+
+}  // namespace blaze