BUG-4329: switch to javaparser

[controller.git] / opendaylight / config / yang-jmx-generator-plugin / src / test / java / org / opendaylight / controller / config / yangjmxgenerator / plugin / AbstractVerifier.java

diff --git a/opendaylight/config/yang-jmx-generator-plugin/src/test/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/AbstractVerifier.java b/opendaylight/config/yang-jmx-generator-plugin/src/test/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/AbstractVerifier.java
new file mode 100644 (file)
index 0000000..d9dd5d0
--- /dev/null
+++ b/opendaylight/config/yang-jmx-generator-plugin/src/test/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/AbstractVerifier.java
@@ -0,0 +1,43 @@
+/*
+ * Copyright (c) 2016 Pantheon Technologies s.r.o. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.config.yangjmxgenerator.plugin;
+
+import static org.junit.Assert.assertEquals;
+import com.github.javaparser.ast.PackageDeclaration;
+import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
+import com.github.javaparser.ast.visitor.VoidVisitorAdapter;
+import com.google.common.base.Preconditions;
+
+abstract class AbstractVerifier extends VoidVisitorAdapter<Void> {
+    private final String expectedPackageName;
+    private final String expectedType;
+    private String packageName;
+    private String type;
+
+    AbstractVerifier(final String expectedPackageName, final String fileName) {
+        this.expectedPackageName = Preconditions.checkNotNull(expectedPackageName);
+        this.expectedType = fileName.substring(0, fileName.length() - 5);
+    }
+
+    @Override
+    public void visit(final ClassOrInterfaceDeclaration n, final Void arg) {
+        type = n.getName();
+        super.visit(n, arg);
+    }
+
+    @Override
+    public final void visit(final PackageDeclaration n, final Void arg) {
+        packageName = n.getName().toString();
+        super.visit(n, arg);
+    }
+
+    void verify() {
+        assertEquals(expectedPackageName, packageName);
+        assertEquals(expectedType, type);
+    }
+}