test/com/sun/servicetag/SystemRegistryTest.java

/*
 * Copyright (c) 2008, 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.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * 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.
 */

/*
 * @test
 * @bug     6622366
 * @summary Basic Test for registry class
 *          by replacing stclient with SvcTagClient utility
 * @author  Mandy Chung
 *
 * @run build SvcTagClient SystemRegistryTest Util
 * @run main SystemRegistryTest
 */

import com.sun.servicetag.*;
import java.io.*;
import java.util.*;

public class SystemRegistryTest {
    private static String registryDir = System.getProperty("test.classes");
    private static String servicetagDir = System.getProperty("test.src");
    private static List<ServiceTag> list = new ArrayList<ServiceTag>();
    private static String[] files = new String[] {
                                        "servicetag1.properties",
                                        "servicetag2.properties",
                                        "servicetag3.properties"
                                    };

    private static Registry registry;
    public static void main(String[] argv) throws Exception {
        registry = Util.getSvcTagClientRegistry();

        for (String filename : files) {
            File f = new File(servicetagDir, filename);
            ServiceTag svcTag = Util.newServiceTag(f);
            ServiceTag st = registry.addServiceTag(svcTag);
            list.add(st);
            System.out.println(st);
        }

        testDuplicate(list.get(0));
        testNotFound();

        // remove a service tag
        String urn = list.get(0).getInstanceURN();
        ServiceTag svcTag = registry.removeServiceTag(urn);
        if (!Util.matches(svcTag, list.get(0))) {
           throw new RuntimeException(urn +
               " deleted but does not match.");
        }

        // get a service tag
        svcTag = list.get(1);
        urn = svcTag.getInstanceURN();
        ServiceTag st = registry.getServiceTag(urn);
        if (!Util.matches(svcTag, st)) {
           throw new RuntimeException(urn +
               " returned from getServiceTag but does not match.");
        }
        // update the service tag
        registry.updateServiceTag(urn, "My new defined ID");
        st = registry.getServiceTag(urn);
        if (Util.matches(svcTag, st)) {
           throw new RuntimeException(urn +
               " updated but expected to be different.");
        }

        if (!st.getProductDefinedInstanceID().equals("My new defined ID")) {
            throw new RuntimeException("Invalid product_defined_instance_id " +
                st.getProductDefinedInstanceID());
        }
        if (st.getInstallerUID() != -1) {
            throw new RuntimeException("Invalid installer_uid " +
                st.getInstallerUID());
        }
        if (st.getTimestamp().equals(svcTag.getTimestamp())) {
            throw new RuntimeException("Timestamp " +
                st.getTimestamp() + " == " + svcTag.getTimestamp());
        }

    }
    private static void testDuplicate(ServiceTag st) throws IOException {
        boolean dup = false;
        try {
           registry.addServiceTag(st);
        } catch (IllegalArgumentException e) {
           dup = true;
        }
        if (!dup) {
           throw new RuntimeException(st.getInstanceURN() +
               " added successfully but expected to be a duplicated.");
        }
    }

    private static void testNotFound() throws Exception {
        String instanceURN = "urn:st:721cf98a-f4d7-6231-bb1d-f2f5aa903ef7";
        ServiceTag svctag = registry.removeServiceTag(instanceURN);
        if (svctag != null) {
           throw new RuntimeException(instanceURN +
               " exists but expected not found");
        }

        svctag = registry.updateServiceTag(instanceURN, "testing");
        if (svctag != null) {
           throw new RuntimeException(instanceURN +
               " exists but expected not found");
        }
    }
}