UtilsForTest.java

// Copyright 2012 Citrix Systems, Inc. Licensed under the
// Apache License, Version 2.0 (the "License"); you may not use this
// file except in compliance with the License.  Citrix Systems, Inc.
// reserves all rights not expressly granted by the License.
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// 
// Automatically generated by addcopyright.py at 04/03/2012
package com.cloud.test.utils;

import java.io.InputStream;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import com.cloud.utils.encoding.Base64;
import com.cloud.utils.exception.CloudRuntimeException;

public class UtilsForTest {
	
	private static DocumentBuilderFactory factory = DocumentBuilderFactory
	.newInstance();
	
	public static boolean verifyTags (Map<String, String> params) {
		boolean result = true;
		for (String value : params.keySet()) {
			if (params.get(value) == null) {
				result=false;
			}
		}
		return result;
	}
	
	public static boolean verifyTagValues (Map<String, String> params, Map<String, String> pattern) {
		boolean result = true;
		
		if (pattern != null) {
			for (String value : pattern.keySet()) {
				if (!pattern.get(value).equals(params.get(value))) {
					result=false;
					System.out.println("Tag " + value + " has " + params.get(value) + " while expected value is: " + pattern.get(value));
				}
			}
		}
		return result;
	}
	
	
	public static Map<String, String> parseXML(InputStream is,
			String[] tagNames) {
		Map<String, String> returnValues = new HashMap<String, String>();
		try {
			DocumentBuilder docBuilder = factory.newDocumentBuilder();
			Document doc = docBuilder.parse(is);
			Element rootElement = doc.getDocumentElement();

			for (int i = 0; i < tagNames.length; i++) {
				NodeList targetNodes = rootElement
						.getElementsByTagName(tagNames[i]);
				if (targetNodes.getLength() <= 0) {
					System.out.println("no " + tagNames[i]
							+ " tag in the response");
					returnValues.put(tagNames[i], null);
				} else {
					returnValues.put(tagNames[i], targetNodes.item(0)
							.getTextContent());
				}
			}
		} catch (Exception ex) {
			System.out.println("error processing XML");
			ex.printStackTrace();
		}
		return returnValues;
	}
	
	
	public static ArrayList<HashMap<String, String>> parseMulXML (InputStream is, String[] tagNames){
		ArrayList<HashMap<String, String>> returnValues = new ArrayList<HashMap <String, String>>();
		
		try {
			DocumentBuilder docBuilder = factory.newDocumentBuilder();
			Document doc = docBuilder.parse(is);
			Element rootElement = doc.getDocumentElement();
			for (int i = 0; i < tagNames.length; i++) {
				NodeList targetNodes = rootElement
						.getElementsByTagName(tagNames[i]);
				if (targetNodes.getLength() <= 0) {
					System.out.println("no " + tagNames[i]
							+ " tag in XML response...returning null");
				} else {
					for (int j = 0; j < targetNodes.getLength(); j++) {
						HashMap<String, String> valueList = new HashMap<String,String> ();
						Node node = targetNodes.item(j);
						//parse child nodes
						NodeList child = node.getChildNodes();
						for (int c=0; c<node.getChildNodes().getLength(); c++){
							child.item(c).getNodeName();
							valueList.put(child.item(c).getNodeName(), child.item(c).getTextContent());
						}
						returnValues.add(valueList);
					}
					
				}
			}
		} catch (Exception ex) {
			System.out.println(ex);
		}
		
		return returnValues;
	}
	
	
	public static String createMD5String(String password) {
		MessageDigest md5;
		try {
			md5 = MessageDigest.getInstance("MD5");
		} catch (NoSuchAlgorithmException e) {
			throw new CloudRuntimeException("Error", e);
		}

		md5.reset();
		BigInteger pwInt = new BigInteger(1, md5.digest(password.getBytes()));

		// make sure our MD5 hash value is 32 digits long...
		StringBuffer sb = new StringBuffer();
		String pwStr = pwInt.toString(16);
		int padding = 32 - pwStr.length();
		for (int i = 0; i < padding; i++) {
			sb.append('0');
		}
		sb.append(pwStr);
		return sb.toString();
	}
	
	
	
	
	
	
	
	public static Map<String, String> getSingleValueFromXML(InputStream is,
			String[] tagNames) {
		Map<String, String> returnValues = new HashMap<String, String>();
		try {
			DocumentBuilder docBuilder = factory.newDocumentBuilder();
			Document doc = docBuilder.parse(is);
			Element rootElement = doc.getDocumentElement();

			for (int i = 0; i < tagNames.length; i++) {
				NodeList targetNodes = rootElement
						.getElementsByTagName(tagNames[i]);
				if (targetNodes.getLength() <= 0) {
					System.out.println("no " + tagNames[i]
							+ " tag in XML response...returning null");
				} else {
					returnValues.put(tagNames[i], targetNodes.item(0)
							.getTextContent());
				}
			}
		} catch (Exception ex) {
			System.out.println("error processing XML");
			ex.printStackTrace();
		}
		return returnValues;
	}
	
	
	public static Map<String, List<String>> getMultipleValuesFromXML(
			InputStream is, String[] tagNames) {
		Map<String, List<String>> returnValues = new HashMap<String, List<String>>();
		try {
			DocumentBuilder docBuilder = factory.newDocumentBuilder();
			Document doc = docBuilder.parse(is);
			Element rootElement = doc.getDocumentElement();
			for (int i = 0; i < tagNames.length; i++) {
				NodeList targetNodes = rootElement
						.getElementsByTagName(tagNames[i]);
				if (targetNodes.getLength() <= 0) {
					System.out.println("no " + tagNames[i]
							+ " tag in XML response...returning null");
				} else {
					List<String> valueList = new ArrayList<String>();
					for (int j = 0; j < targetNodes.getLength(); j++) {
						Node node = targetNodes.item(j);
						valueList.add(node.getTextContent());
					}
					returnValues.put(tagNames[i], valueList);
				}
			}
		} catch (Exception ex) {
			System.out.println(ex);
		}
		return returnValues;
	}
	
	
	
	public static String signRequest(String request, String key) {
		try {
			Mac mac = Mac.getInstance("HmacSHA1");
			SecretKeySpec keySpec = new SecretKeySpec(key.getBytes(),
					"HmacSHA1");
			mac.init(keySpec);
			mac.update(request.getBytes());
			byte[] encryptedBytes = mac.doFinal();
			//System.out.println("HmacSHA1 hash: " + encryptedBytes);
			return Base64.encodeBytes(encryptedBytes);
		} catch (Exception ex) {
			System.out.println("unable to sign request");
			ex.printStackTrace();
		}
		return null;
	}
	
}