added pg dependent tests

Author: AyanKoche

vertx-pg-client/src/test/java/io/vertx/pgclient/data/CidrCodecTest.java

@@ -0,0 +1,116 @@
+package io.vertx.pgclient.data;
+
+import io.vertx.ext.unit.TestContext;
+import io.vertx.pgclient.PgConnection;
+import io.vertx.sqlclient.*;
+import org.junit.Test;
+
+import java.net.Inet4Address;
+import java.net.Inet6Address;
+import java.net.InetAddress;
+import java.util.function.BiFunction;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertThrows;
+
+public class CidrCodecTest extends DataTypeTestBase{
+
+  @Test
+  public void testValidIPv4() throws Exception {
+    InetAddress address = InetAddress.getByName("192.168.1.1");
+    Cidr cidr = new Cidr();
+    cidr.setAddress(address);
+    cidr.setNetmask(24);
+    assertEquals(address, cidr.getAddress());
+    assertEquals(Integer.valueOf(24), cidr.getNetmask());
+  }
+
+  @Test
+  public void testValidIPv6() throws Exception {
+    InetAddress address = InetAddress.getByName("fe80::f03c:91ff:feae:e944");
+    Cidr cidr = new Cidr();
+    cidr.setAddress(address);
+    cidr.setNetmask(64);
+    assertEquals(address, cidr.getAddress());
+    assertEquals(Integer.valueOf(64), cidr.getNetmask());
+  }
+
+  @Test
+  public void testInvalidNetmaskIPv4() throws Exception {
+    InetAddress address = InetAddress.getByName("192.168.1.1");
+    Cidr cidr = new Cidr();
+    cidr.setAddress(address);
+    assertThrows(IllegalArgumentException.class, () -> cidr.setNetmask(33));
+  }
+
+  @Test
+  public void testInvalidNetmaskIPv6() throws Exception {
+    InetAddress address = InetAddress.getByName("fe80::f03c:91ff:feae:e944");
+    Cidr cidr = new Cidr();
+    cidr.setAddress(address);
+    assertThrows(IllegalArgumentException.class, () -> cidr.setNetmask(129));
+  }
+
+  @Test
+  public void testBinaryDecodeCIDR(TestContext ctx) throws Exception {
+    testDecodeCIDR(ctx, SqlClient::preparedQuery);
+  }
+
+  private void testDecodeCIDR(TestContext ctx, BiFunction<SqlClient, String, Query<RowSet<Row>>> a) throws Exception {
+    InetAddress addr1 = Inet4Address.getByName("128.0.0.0");
+    InetAddress addr2 = Inet6Address.getByName("2001:0db8:1234:0000:0000:0000:0000:0000");
+    PgConnection.connect(vertx, options).onComplete(ctx.asyncAssertSuccess(conn -> {
+      a.apply(conn, "SELECT " +
+          "'128.0.0.0'::CIDR," +
+          "'128.0.0.0/4'::CIDR," +
+          "'2001:0db8:1234:0000:0000:0000:0000:0000'::CIDR," +
+          "'2001:0db8:1234:0000:0000:0000:0000:0000/56'::CIDR")
+        .execute()
+        .onComplete(ctx.asyncAssertSuccess(rows -> {
+          ctx.assertEquals(1, rows.size());
+          Row row = rows.iterator().next();
+          Cidr v1 = (Cidr) row.getValue(0);
+          Cidr v2 = (Cidr) row.getValue(1);
+          Cidr v3 = (Cidr) row.getValue(2);
+          Cidr v4 = (Cidr) row.getValue(3);
+          ctx.assertEquals(addr1, v1.getAddress());
+          ctx.assertEquals(32,v1.getNetmask());
+          ctx.assertEquals(addr1, v2.getAddress());
+          ctx.assertEquals(4, v2.getNetmask());
+          ctx.assertEquals(addr2, v3.getAddress());
+          ctx.assertEquals(128, v3.getNetmask());
+          ctx.assertEquals(addr2, v4.getAddress());
+          ctx.assertEquals(56, v4.getNetmask());
+        }));
+    }));
+  }
+
+  @Test
+  public void testBinaryEncodeCIDR(TestContext ctx) throws Exception {
+    InetAddress addr1 = Inet4Address.getByName("128.0.0.0");
+    InetAddress addr2 = Inet6Address.getByName("2001:0db8:1234:0000:0000:0000:0000:0000");
+    PgConnection.connect(vertx, options).onComplete(ctx.asyncAssertSuccess(conn -> {
+      conn
+        .preparedQuery("SELECT ($1::CIDR)::VARCHAR, ($2::CIDR)::VARCHAR, ($3::CIDR)::VARCHAR, ($4::CIDR)::VARCHAR")
+        .execute(Tuple.of(
+          new Cidr().setAddress(addr1),
+          new Cidr().setAddress(addr1).setNetmask(4),
+          new Cidr().setAddress(addr2),
+          new Cidr().setAddress(addr2).setNetmask(56)
+        ))
+        .onComplete(ctx.asyncAssertSuccess(rows -> {
+          ctx.assertEquals(1, rows.size());
+          Row row = rows.iterator().next();
+          String v1 = row.getString(0);
+          String v2 = row.getString(1);
+          String v3 = row.getString(2);
+          String v4 = row.getString(3);
+          ctx.assertEquals("128.0.0.0/32", v1);
+          ctx.assertEquals("128.0.0.0/4", v2);
+          ctx.assertEquals("2001:db8:1234::/128", v3);
+          ctx.assertEquals("2001:db8:1234::/56", v4);
+        }));
+    }));
+  }
+
+}