1 files changed, 52 insertions, 0 deletions
diff --git a/src/language_matchers.h b/src/language_matchers.h
new file mode 100644
index 0000000..b230358
--- /dev/null
+++ b/src/language_matchers.h
@@ -0,0 +1,52 @@
+// Copyright 2021 Benjamin Barenblat
+//
+// 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
+//
+//     https://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.
+
+// gMock matchers for Terms.
+
+#ifndef EC_SRC_LANGUAGE_MATCHERS_H_
+#define EC_SRC_LANGUAGE_MATCHERS_H_
+
+#include <gmock/gmock.h>
+
+#include "src/language.h"
+#include "third_party/abseil/absl/strings/str_cat.h"
+
+namespace ec {
+
+MATCHER_P(PointsToGroundTerm, d,
+          absl::StrCat("points to GroundTerm(", d, ")")) {
+  using ::testing::DoubleEq;
+  using ::testing::Pointee;
+  using ::testing::Property;
+  using ::testing::WhenDynamicCastTo;
+
+  return ExplainMatchResult(Pointee(WhenDynamicCastTo<const GroundTerm&>(
+                                Property(&GroundTerm::value, DoubleEq(d)))),
+                            arg, result_listener);
+}
+
+MATCHER_P(PointsToSymbolTerm, s,
+          absl::StrCat("points to SymbolTerm(", s, ")")) {
+  using ::testing::Eq;
+  using ::testing::Pointee;
+  using ::testing::WhenDynamicCastTo;
+
+  return ExplainMatchResult(
+      Pointee(WhenDynamicCastTo<const SymbolTerm&>(Eq(SymbolTerm(s)))), arg,
+      result_listener);
+}
+
+}  // namespace ec
+
+#endif  // EC_SRC_LANGUAGE_MATCHERS_H_