/*

 * 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;

    }

}