test/jdk/com/sun/jndi/dns/AttributeTests/GetRRsBase.java
changeset 51192 499b873761d8
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/com/sun/jndi/dns/AttributeTests/GetRRsBase.java	Wed Jul 25 11:03:07 2018 +0800
@@ -0,0 +1,89 @@
+/*
+ * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+import javax.naming.Context;
+import javax.naming.NamingException;
+import javax.naming.directory.Attributes;
+import javax.naming.directory.InitialDirContext;
+
+public abstract class GetRRsBase extends DNSTestBase {
+    public static final int ROOT_LIMIT = 5; // point at which to use root ctx
+    private String[] keys;
+    private String[][] numAttrs;
+    private String[][] attrs;
+
+    public GetRRsBase() {
+        // set default test data
+        setKeys(new String[] { "host1", "", "ipv6", "sun", "_ftp._Tcp",
+                "1.4.2.1.in-addr.arpa", });
+        setNumAttrs(new String[][] { { "IN 1", "IN 15", "IN 13", "IN 16" },
+                { "IN 2", "IN 6" }, { "IN 28" }, { "IN 5" }, { "IN 33" },
+                { "IN 12" }, });
+        setAttrs(
+                new String[][] { { "A", "MX", "HINFO", "TXT" }, { "NS", "SOA" },
+                        { "AAAA" }, { "CNAME" }, { "SRV" }, { "PTR" }, });
+    }
+
+    public void initContext() throws Exception {
+        setContext(new InitialDirContext(env()));
+    }
+
+    public void switchToRootUrl() throws NamingException {
+        DNSTestUtils.cleanup(context());
+        env().put(Context.PROVIDER_URL, DNSTestUtils.getRootUrl(env()));
+        setContext(new InitialDirContext(env()));
+    }
+
+    public void verifyAttributes(Attributes attrs, String[] mandatory) {
+        DNSTestUtils.verifySchema(attrs, mandatory, new String[] {});
+    }
+
+    public Attributes getAttributes(String name, String[] attrIds)
+            throws Exception {
+        return context().getAttributes(name, attrIds);
+    }
+
+    public void setKeys(String[] keys) {
+        this.keys = keys;
+    }
+
+    public String[] getKeys() {
+        return keys;
+    }
+
+    public void setNumAttrs(String[][] numAttrs) {
+        this.numAttrs = numAttrs;
+    }
+
+    public String[][] getNumAttrs() {
+        return numAttrs;
+    }
+
+    public void setAttrs(String[][] attrs) {
+        this.attrs = attrs;
+    }
+
+    public String[][] getAttrs() {
+        return attrs;
+    }
+}