tajo-common/src/test/java/org/apache/tajo/common/type/TestIPv4.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.tajo.common.type;

import org.junit.Test;
import org.apache.tajo.exception.InvalidAddressException;

import static org.junit.Assert.*;

public class TestIPv4 {

	@Test
	public final void testSet() throws InvalidAddressException {
		IPv4 ip = null;
	  ip = new IPv4("255.255.255.255");
		byte[] b = new byte[4];
		for (int i = 0; i < 4; i++) {
			b[i] = (byte)0xFF;
		}
		IPv4 ip2 = new IPv4(b);
		assertEquals(ip, ip2);
	}
	
	@Test
	public final void testEqual() throws InvalidAddressException {
		IPv4 ip1 = new IPv4("163.152.23.1");
		IPv4 ip2 = new IPv4("163.152.23.2");
		IPv4 ip3 = new IPv4("163.152.23.1");
		assertTrue(ip1.equals(ip3));
		assertFalse(ip1.equals(ip2));
		IPv4 ip4 = new IPv4("255.255.0.0");
		assertFalse(ip1.equals(ip4));
	}
	
	@Test
	public final void testAnd() throws InvalidAddressException {
		IPv4 ip1 = new IPv4("163.152.23.223");
		IPv4 ip2 = new IPv4("255.255.255.0");
		IPv4 ip3 = new IPv4("255.0.0.0");
		assertEquals(new IPv4("163.152.23.0"), ip1.and(ip2));
		assertFalse(ip1.and(ip2).equals(new IPv4("163.152.0.0")));
		assertTrue(ip1.and(ip3).equals(new IPv4("163.0.0.0")));
	}
	
	@Test
	public final void testMatchSubnet() throws InvalidAddressException {
		IPv4 ip1 = new IPv4("163.152.23.223");
		assertTrue(ip1.matchSubnet("163.152.23.0/1"));
		assertFalse(ip1.matchSubnet("163.152.23.0/28"));
	}
	
	@Test
	public final void testCompareTo() throws InvalidAddressException {
		IPv4 ip1 = new IPv4("163.152.23.1");
		IPv4 ip2 = new IPv4("163.152.23.2");
		IPv4 ip3 = new IPv4("177.234.123.12");
		IPv4 ip4 = new IPv4("177.234.123.12");
		assertTrue(ip1.compareTo(ip2) == -1);
		assertTrue(ip2.compareTo(ip1) == 1);
		assertTrue(ip3.compareTo(ip1) == 1);
		assertFalse(ip3.compareTo(ip2) != 1);
		assertTrue(ip4.compareTo(ip3) == 0);
	}
}