1 files changed, 250 insertions, 0 deletions
diff --git a/test/com/sun/servicetag/Util.java b/test/com/sun/servicetag/Util.java
new file mode 100644
index 000000000..367486dba
--- /dev/null
+++ b/test/com/sun/servicetag/Util.java
@@ -0,0 +1,250 @@
+/*
+ * Copyright 2008 Sun Microsystems, Inc.  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.  Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/*
+ * @bug     6622366
+ * @summary Utility class used by other jtreg tests
+ */
+
+import com.sun.servicetag.RegistrationData;
+import com.sun.servicetag.ServiceTag;
+import com.sun.servicetag.Registry;
+
+import java.util.Set;
+import java.util.Date;
+import java.util.Map;
+import java.util.Properties;
+import java.util.TimeZone;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.io.*;
+
+public class Util {
+    public static ServiceTag newServiceTag(File f)
+            throws FileNotFoundException, IOException, NumberFormatException {
+        return newServiceTag(f, false /* with instance_urn */);
+    }
+
+    public static ServiceTag newServiceTag(File f, boolean noInstanceURN)
+            throws FileNotFoundException, IOException, NumberFormatException {
+        Properties props = new Properties();
+        FileReader reader = new FileReader(f);
+        try {
+            props.load(reader);
+        } finally {
+            reader.close();
+        }
+        if (noInstanceURN) {
+            return ServiceTag.newInstance(
+                            props.getProperty("product_name"),
+                            props.getProperty("product_version"),
+                            props.getProperty("product_urn"),
+                            props.getProperty("product_parent"),
+                            props.getProperty("product_parent_urn"),
+                            props.getProperty("product_defined_inst_id"),
+                            props.getProperty("product_vendor"),
+                            props.getProperty("platform_arch"),
+                            props.getProperty("container"),
+                            props.getProperty("source"));
+        } else {
+            return ServiceTag.newInstance(
+                            props.getProperty("instance_urn"),
+                            props.getProperty("product_name"),
+                            props.getProperty("product_version"),
+                            props.getProperty("product_urn"),
+                            props.getProperty("product_parent"),
+                            props.getProperty("product_parent_urn"),
+                            props.getProperty("product_defined_inst_id"),
+                            props.getProperty("product_vendor"),
+                            props.getProperty("platform_arch"),
+                            props.getProperty("container"),
+                            props.getProperty("source"));
+        }
+    }
+
+    public static boolean matches(ServiceTag st1, ServiceTag st2) {
+        if (!st1.getInstanceURN().equals(st2.getInstanceURN())) {
+            System.out.println("instance_urn: " + st1.getInstanceURN() +
+                " != " + st2.getInstanceURN());
+            return false;
+        }
+        return matchesNoInstanceUrn(st1, st2);
+    }
+
+    public static boolean matchesNoInstanceUrn(ServiceTag st1, ServiceTag st2) {
+        if (!st1.getProductName().equals(st2.getProductName())) {
+            System.out.println("product_name: " + st1.getProductName() +
+                " != " + st2.getProductName());
+            return false;
+        }
+
+        if (!st1.getProductVersion().equals(st2.getProductVersion())) {
+            System.out.println("product_version: " + st1.getProductVersion() +
+                " != " + st2.getProductVersion());
+            return false;
+        }
+        if (!st1.getProductURN().equals(st2.getProductURN())) {
+            System.out.println("product_urn: " + st1.getProductURN() +
+                " != " + st2.getProductURN());
+            return false;
+        }
+        if (!st1.getProductParentURN().equals(st2.getProductParentURN())) {
+            System.out.println("product_parent_urn: " + st1.getProductParentURN() +
+                " != " + st2.getProductParentURN());
+            return false;
+        }
+        if (!st1.getProductParent().equals(st2.getProductParent())) {
+            System.out.println("product_parent: " + st1.getProductParent() +
+                " != " + st2.getProductParent());
+            return false;
+        }
+        if (!st1.getProductDefinedInstanceID().equals(st2.getProductDefinedInstanceID())) {
+            System.out.println("product_defined_inst_id: " +
+                st1.getProductDefinedInstanceID() +
+                " != " + st2.getProductDefinedInstanceID());
+            return false;
+        }
+        if (!st1.getProductVendor().equals(st2.getProductVendor())) {
+            System.out.println("product_vendor: " + st1.getProductVendor() +
+                " != " + st2.getProductVendor());
+            return false;
+        }
+        if (!st1.getPlatformArch().equals(st2.getPlatformArch())) {
+            System.out.println("platform_arch: " + st1.getPlatformArch() +
+                " != " + st2.getPlatformArch());
+            return false;
+        }
+        if (!st1.getContainer().equals(st2.getContainer())) {
+            System.out.println("container: " + st1.getContainer() +
+                " != " + st2.getContainer());
+            return false;
+        }
+        if (!st1.getSource().equals(st2.getSource())) {
+            System.out.println("source: " + st1.getSource() +
+                " != " + st2.getSource());
+            return false;
+        }
+        return true;
+    }
+
+    public static void checkRegistrationData(String regFile,
+                                             Map<String, ServiceTag> stMap)
+            throws IOException {
+        BufferedInputStream in = new BufferedInputStream(new FileInputStream(regFile));
+        RegistrationData registration = RegistrationData.loadFromXML(in);
+        Set<ServiceTag> svcTags = registration.getServiceTags();
+        if (svcTags.size() != stMap.size()) {
+            throw new RuntimeException("Invalid service tag count= " +
+                svcTags.size() + " expected=" + stMap.size());
+        }
+        for (ServiceTag st : svcTags) {
+            ServiceTag st1 = stMap.get(st.getInstanceURN());
+            if (!matches(st, st1)) {
+                throw new RuntimeException("ServiceTag in the registry " +
+                    "does not match the one in the map");
+            }
+        }
+    }
+
+
+    /**
+     * Formats the Date into a timestamp string in YYYY-MM-dd HH:mm:ss GMT.
+     * @param timestamp Date
+     * @return a string representation of the timestamp in the YYYY-MM-dd HH:mm:ss GMT format.
+     */
+    static String formatTimestamp(Date timestamp) {
+        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z");
+        df.setTimeZone(TimeZone.getTimeZone("GMT"));
+        return df.format(timestamp);
+    }
+
+    /**
+     * Parses a timestamp string in YYYY-MM-dd HH:mm:ss GMT format.
+     * @param timestamp Timestamp in the YYYY-MM-dd HH:mm:ss GMT format.
+     * @return Date
+     */
+    static Date parseTimestamp(String timestamp) {
+        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z");
+        df.setTimeZone(TimeZone.getTimeZone("GMT"));
+        try {
+            return df.parse(timestamp);
+        } catch (ParseException e) {
+            // should not reach here
+            e.printStackTrace();
+            return new Date();
+        }
+    }
+
+    /**
+     * Returns the command simulating stclient behavior.
+     */
+    static String getSvcClientCommand(String stclientRegistry) {
+        String regDir = System.getProperty("test.classes");
+
+        StringBuilder sb = new StringBuilder();
+        // wrap each argument to the command with double quotes
+        sb.append("\"");
+        sb.append(System.getProperty("java.home"));
+        sb.append(File.separator).append("bin");
+        sb.append(File.separator).append("java");
+        sb.append("\"");
+        sb.append(" -cp ");
+        sb.append("\"").append(regDir).append("\"");
+        sb.append(" \"-Dstclient.registry.path=");
+        sb.append(stclientRegistry).append("\"");
+        sb.append(" SvcTagClient");
+        return sb.toString();
+    }
+
+    private static Registry registry = null;
+    /**
+     * Returns the Registry processed by SvcTagClient that simulates
+     * stclient.
+     */
+    static synchronized Registry getSvcTagClientRegistry() throws IOException {
+        if (registry != null) {
+            return registry;
+        }
+
+        // System.setProperty("servicetag.verbose", "true");
+        // enable the helper class
+        System.setProperty("servicetag.sthelper.supported", "true");
+
+        // clean up registry.xml
+        String regDir = System.getProperty("test.classes");
+        File registryFile = new File(regDir, "registry.xml");
+        if (registryFile.exists()) {
+            registryFile.delete();
+        }
+
+        String stclientCmd = Util.getSvcClientCommand(registryFile.getCanonicalPath());
+        System.out.println("stclient cmd: " + stclientCmd);
+        System.setProperty("servicetag.stclient.cmd", stclientCmd);
+
+        // get the Registry object after the system properties are set
+        registry = Registry.getSystemRegistry();
+        return registry;
+    }
+}