uima-ducc-cli/src/test/java/org/apache/uima/ducc/cli/test/MachineFactsTest.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 org.apache.uima.ducc.cli.test;

import static org.junit.Assert.*;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;

import org.apache.uima.ducc.cli.ws.json.MachineFacts;
import org.apache.uima.ducc.cli.ws.json.MachineFactsList;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

import com.google.gson.Gson;

public class MachineFactsTest {
	
	private long shareSize = 15;
	
	private Random random = new Random();
	
	private String[] statusSet = { "defined", "down", "up" };
	private String[] memorySet = { "", "24", "31", "39", "43", "47", "79" };
	
	private String createStatus() {
		int index = random.nextInt(statusSet.length);
		String status = statusSet[index];
		return status;
	}
	
	private String createIP() {
		String ip = "";
		int type = random.nextInt(10);
		if(type > 2) {
			int index = random.nextInt(256);
			ip = "192.168.5."+index;
		}
		return ip;
	}
	
	private String createName() {
		String name = "node1"+random.nextInt(801);
		return name;
	}
	
	private String createMemory() {
		int index = random.nextInt(memorySet.length);
		String memory = memorySet[index];
		return memory;
	}
	
	private String createSwap() {
		String swap = "";
		int type = random.nextInt(10);
		if(type > 2) {
			int index = random.nextInt(10);
			swap = ""+index;
		}
		return swap;
	}
	
	private List<String> createAliens() {
		List<String> aliens = new ArrayList<String>();
		int count = random.nextInt(10);
		for(int i=0;i<count;i++) {
			aliens.add(""+random.nextInt(99999));
		}
		return aliens;
	}
	
	private String createSharesTotal(String memory) {
		String sharesTotal = "";
		try {
			Integer integer = new Integer(memory);
			long shares = integer / shareSize;
			sharesTotal = ""+shares;
		}
		catch(Exception e) {
		}
		return sharesTotal;
	}
	
	private String createSharesInuse(String total) {
		String sharesInuse = "";
		try {
			Integer integer = new Integer(total);
			long shares = random.nextInt(integer+1);
			sharesInuse = ""+shares;
		}
		catch(Exception e) {
		}
		return sharesInuse;
	}
	
	private String createHeartbeat() {
		String heartbeat = "";
		int type = random.nextInt(100);
		int beat = random.nextInt(60);
		if(type > 5) {
			if(type > 15) {
				heartbeat = ""+beat;
			}
			else {
				heartbeat = ""+random.nextInt(600)+60;
			}
		}
		return heartbeat;
	}
	
	private MachineFacts createMachineFacts() {
		String status = createStatus();
		String ip = createIP();
		String name = createName();
		String reserve = createMemory();
		String memory = createMemory();
		String swap = createSwap();
		List<String> aliens = createAliens();
		String sharesTotal = createSharesTotal(memory);
		String sharesInuse = createSharesInuse(sharesTotal);
		String heartbeat = createHeartbeat();
		MachineFacts machineFacts = new MachineFacts(status,ip,name, reserve, memory, swap, aliens, sharesTotal, sharesInuse, heartbeat);
		return machineFacts;
	}
	
	private MachineFactsList createMachineFactsList() {
		MachineFactsList machineFactsList = new MachineFactsList();
		for(int i=0; i<10; i++) {
			machineFactsList.add(createMachineFacts());
		}
		return machineFactsList;
	}
	
	private boolean compare(MachineFacts m1, MachineFacts m2) {
		boolean retVal = false;
		try {
			if(true
			&& m1.heartbeat.equals(m2.heartbeat) 
			&& m1.ip.equals(m2.ip) 
			&& m1.memory.equals(m2.memory) 
			&& m1.name.equals(m2.name) 
			&& m1.reserve.equals(m2.reserve) 
			&& m1.sharesInuse.equals(m2.sharesInuse) 
			&& m1.sharesTotal.equals(m2.sharesTotal)
			&& m1.status.equals(m2.status) 
			&& m1.swap.equals(m2.swap) 
			) {
				retVal = true;
			}
		}
		catch(Exception e) {
		}
		return retVal;
	}
	
	@BeforeClass
	public static void setUpBeforeClass() throws Exception {
	}

	@AfterClass
	public static void tearDownAfterClass() throws Exception {
	}

	@Before
	public void setUp() throws Exception {
	}

	@After
	public void tearDown() throws Exception {
	}

	@Test
	public void testMachineFacts() {
		MachineFactsTest machineFactsTest = new MachineFactsTest();
		MachineFactsList machineFactsList = machineFactsTest.createMachineFactsList();
		Gson gson = new Gson();
		String jSon = gson.toJson(machineFactsList);
		MachineFactsList reconstituted = gson.fromJson(jSon, MachineFactsList.class);
		for(int i=0; i<machineFactsList.size(); i++) {
			MachineFacts m1 = machineFactsList.get(i);
			MachineFacts m2 = reconstituted.get(i);
			if(!compare(m1,m2)) {
				fail("missing "+"name="+m1.name);
			}
		}
	}

}