TestDumpCommands.java

/*
  Copyright (c) 2013, 2014, Oracle and/or its affiliates. All rights reserved.

  The MySQL Connector/J is licensed under the terms of the GPLv2
  <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most MySQL Connectors.
  There are special exceptions to the terms and conditions of the GPLv2 as it is applied to
  this software, see the FLOSS License Exception
  <http://www.mysql.com/about/legal/licensing/foss-exception.html>.

  This program is free software; you can redistribute it and/or modify it under the terms
  of the GNU General Public License as published by the Free Software Foundation; version 2
  of the License.

  This program 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 for more details.

  You should have received a copy of the GNU General Public License along with this
  program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth
  Floor, Boston, MA 02110-1301  USA

 */

package testsuite.fabric;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Set;

import com.mysql.fabric.Server;
import com.mysql.fabric.ServerGroup;
import com.mysql.fabric.ShardMapping;
import com.mysql.fabric.proto.xmlrpc.XmlRpcClient;

import junit.framework.TestCase;

/**
 * Tests for `mysqlfabric dump.*'.
 * 
 * Depends on standard Fabric test setup (which as of yet is not defined).
 * Updates to this test setup will require changes to the tests.
 */
public class TestDumpCommands extends TestCase {
	private String fabricUrl;

	private XmlRpcClient client;

	public TestDumpCommands() {
		String hostname =  System.getProperty("com.mysql.fabric.testsuite.hostname");
		String port = System.getProperty("com.mysql.fabric.testsuite.port");
		this.fabricUrl = "http://" + hostname + ":" + port;
	}

	public void setUp() throws Exception {
		this.client = new XmlRpcClient(this.fabricUrl,
									   System.getProperty("com.mysql.fabric.testsuite.fabricUsername"),
									   System.getProperty("com.mysql.fabric.testsuite.fabricPassword"));
	}

	public static Comparator<Server> serverHostnamePortSorter = new Comparator<Server>() {
		public int compare(Server s1, Server s2) {
			int l;
			l = s1.getHostname().compareTo(s2.getHostname());
			if (l != 0)
				return l;

			l = ((Integer)s1.getPort()).compareTo(s2.getPort());
			return l;
		}
	};

	/**
	 * Test the Client.getServers() without a match pattern (all servers).
	 */
	public void testDumpServersAll() throws Exception {
		Set<ServerGroup> serverGroups = this.client.getServerGroups().getData();
		List<Server> servers = new ArrayList<Server>();
		for (ServerGroup g : serverGroups) {
			servers.addAll(g.getServers());
		}
		Collections.sort(servers, serverHostnamePortSorter);
		assertEquals(3, servers.size());

		assertEquals("fabric_test1_global", servers.get(0).getGroupName());
		assertEquals("fabric_test1_shard1", servers.get(1).getGroupName());
		assertEquals("fabric_test1_shard2", servers.get(2).getGroupName());

		
		String ghost =  System.getProperty("com.mysql.fabric.testsuite.global.host");
		String gport = System.getProperty("com.mysql.fabric.testsuite.global.port");
		String s1host =  System.getProperty("com.mysql.fabric.testsuite.shard1.host");
		String s1port = System.getProperty("com.mysql.fabric.testsuite.shard1.port");
		String s2host =  System.getProperty("com.mysql.fabric.testsuite.shard2.host");
		String s2port = System.getProperty("com.mysql.fabric.testsuite.shard2.port");
		
		
		assertEquals((ghost != null ? ghost : "127.0.0.1"), servers.get(0).getHostname());
		assertEquals((s1host != null ? s1host : "127.0.0.1"), servers.get(1).getHostname());
		assertEquals((s2host != null ? s2host : "127.0.0.1"), servers.get(2).getHostname());

		assertEquals((gport != null ? Integer.valueOf(gport) : 3401), servers.get(0).getPort());
		assertEquals((s1port != null ? Integer.valueOf(s1port) : 3402), servers.get(1).getPort());
		assertEquals((s2port != null ? Integer.valueOf(s2port) : 3403), servers.get(2).getPort());
	}

	/**
	 * Test the Client.getShardMaps() without a match pattern (all maps).
	 */
	public void testDumpShardMapsAll() throws Exception {
		Set<ShardMapping> shardMappings = this.client.getShardMappings().getData();
		assertEquals(1, shardMappings.size());
		ShardMapping shardMapping = shardMappings.iterator().next();

		assertEquals(1, shardMapping.getShardTables().size());
	}
}