vmware-base/test/com/cloud/hypervisor/vmware/util/TestVmwareUtil.java

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with 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.
package com.cloud.hypervisor.vmware.util;
import java.io.File;

import junit.framework.Assert;

import org.apache.log4j.Logger;

import com.cloud.hypervisor.vmware.resource.SshHelper;
import com.cloud.hypervisor.vmware.resource.VmwareContextFactory;
import com.cloud.utils.Pair;
import com.cloud.utils.testcase.Log4jEnabledTestCase;
import com.vmware.vim25.ManagedObjectReference;

public class TestVmwareUtil extends Log4jEnabledTestCase {
    private static final Logger s_logger = Logger.getLogger(TestVmwareUtil.class);

	public void testContextCreation() {
		try {
			VmwareContext context = VmwareContextFactory.create("vsphere-1.lab.vmops.com", "Administrator", "Suite219");
			Assert.assertTrue(true);
			context.close();
		} catch(Exception e) {
			s_logger.error("Unexpected exception : ", e);
		}
	}
	
	public void testSearchIndex() {
		try {
			VmwareContext context = VmwareContextFactory.create("vsphere-1.lab.vmops.com", "Administrator", "Suite219");
			Assert.assertTrue(true);
			
			ManagedObjectReference morHost = context.getService().findByDnsName(context.getServiceContent().getSearchIndex(), 
				null, "esxhost-1.lab.vmops.com", false);
			Assert.assertTrue(morHost.getType().equalsIgnoreCase("HostSystem"));
			
			morHost = context.getService().findByIp(context.getServiceContent().getSearchIndex(), 
				null, "192.168.1.168", false);
			Assert.assertTrue(morHost == null);
			context.close();
		} catch(Exception e) {
			s_logger.error("Unexpected exception : ", e);
		}
	}
	
	public void testVmxFileDownload() {
		try {
			VmwareContext context = VmwareContextFactory.create("vsphere-1.lab.vmops.com", "Administrator", "Suite219");
			byte[] vmxContent = context.getResourceContent("https://vsphere-1.lab.vmops.com/folder/ServerRoom-Fedora32/ServerRoom-Fedora32.vmx?dcPath=cupertino&dsName=NFS%20datastore");
			System.out.print(new String(vmxContent));
			context.close();
		} catch(Exception e) {
			s_logger.error("Unexpected exception : ", e);
		}
	}
	
	public void testVmxFileParser() {
		String[] tokens = "[NFS datastore] Fedora-clone-test/Fedora-clone-test.vmx".split("\\[|\\]|/");
		
		for(String str : tokens) {
			System.out.println("Token " + str);
		}
	}
	
	public void testSsh() {
		try {
			File file = new File("c:\\temp\\id_rsa.kelven");
			if(!file.exists()) {
				System.out.println("key file does not exist!");
			}
			
			Pair<Boolean, String> result = SshHelper.sshExecute("192.168.1.107", 22, "kelven", file, null, "ls -al");
			System.out.println("Result: " + result.second());
		} catch(Exception e) {
			s_logger.error("Unexpected exception : ", e);
		}
	}
	
	public void testScp() {
		try {
			File file = new File("c:\\temp\\id_rsa.kelven");
			if(!file.exists()) {
				System.out.println("key file does not exist!");
			}
			
			SshHelper.scpTo("192.168.1.107", 22, "kelven", file, null, "~", "Hello, world".getBytes(), 
				"hello.txt", null);
		} catch(Exception e) {
			s_logger.error("Unexpected exception : ", e);
		}
	}
}