Added response data to operational datastore, refactored data validation
[packetcable.git] / packetcable-policy-server / src / test / java / org / opendaylight / controller / packetcable / provider / PCMMServiceTest.java
index 73de401e0ce543239e9679fd6d22c4bb0041babf..4e3a9d49fa41f6bb2995c4a203fd83454292169f 100644 (file)
@@ -1,5 +1,20 @@
+/*
+ * (c) 2015 Cable Television Laboratories, Inc.  All rights reserved.
+ */
+
 package org.opendaylight.controller.packetcable.provider;
 
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.InetAddress;
+import java.net.Socket;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
@@ -8,38 +23,36 @@ import org.mockito.Mockito;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.ServiceClassName;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.ServiceFlowDirection;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.TosByte;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.TpProtocol;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.ccap.Ccaps;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.ccap.attributes.AmId;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.ccap.attributes.Connection;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.pcmm.qos.classifier.Classifier;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.pcmm.qos.gate.spec.GateSpec;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.pcmm.qos.gates.apps.subs.Gates;
-import org.opendaylight.yang.gen.v1.urn.packetcable.rev150327.pcmm.qos.traffic.profile.TrafficProfile;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.ServiceClassName;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.ServiceFlowDirection;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.TosByte;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.TpProtocol;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.ccap.attributes.AmId;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.ccap.attributes.Connection;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.ccaps.Ccap;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.pcmm.qos.classifier.Classifier;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.pcmm.qos.gate.spec.GateSpec;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.pcmm.qos.gates.apps.app.subscribers.subscriber.gates.Gate;
+import org.opendaylight.yang.gen.v1.urn.packetcable.rev151026.pcmm.qos.traffic.profile.TrafficProfile;
+import org.pcmm.PCMMPdpAgent;
+import org.pcmm.gates.IGateSpec.Direction;
 import org.pcmm.gates.IPCMMGate;
 import org.pcmm.rcd.IPCMMClient;
 import org.pcmm.rcd.impl.CMTS;
-import org.umu.cops.stack.*;
+import org.umu.cops.stack.COPSClientSI;
+import org.umu.cops.stack.COPSContext;
 import org.umu.cops.stack.COPSContext.RType;
+import org.umu.cops.stack.COPSData;
+import org.umu.cops.stack.COPSDecision;
 import org.umu.cops.stack.COPSDecision.Command;
 import org.umu.cops.stack.COPSDecision.DecisionFlag;
+import org.umu.cops.stack.COPSDecisionMsg;
+import org.umu.cops.stack.COPSHandle;
+import org.umu.cops.stack.COPSMsg;
+import org.umu.cops.stack.COPSMsgParser;
 import org.umu.cops.stack.COPSObjHeader.CNum;
 import org.umu.cops.stack.COPSObjHeader.CType;
 
-import java.io.ByteArrayInputStream;
-import java.io.ByteArrayOutputStream;
-import java.io.InputStream;
-import java.io.OutputStream;
-import java.net.InetAddress;
-import java.net.Socket;
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.Map;
-import java.util.Set;
-
 /**
  * Tests the PCMMService's ability to connect to a CMTS. Gate additions will not properly work as there is currently
  * not any other means to receive acknowledgements. This functionality must be tested by the PCMMService's client
@@ -47,86 +60,207 @@ import java.util.Set;
  */
 public class PCMMServiceTest {
 
+    private final static String ccapId = "ccap-1";
+    private final static String gatePath = "testGatePath";
+
+    /**
+     * Denotes whether or not a real CMTS is being tested against.
+     * Ensure the checked-in value is always false else tests will most likely fail.
+     */
+    private final static boolean realCmts = false;
+
+
+    // The test objects/values to use that will be instantiated in @Before
+
     /**
      * The mock CMTS running on localhost with a dynamic port assigned.
      */
     private CMTS icmts;
 
     /**
-     * Represents the IP address of the CM the gate should be set against.
+     * The IP address object for the CMTS to test against
      */
-    private byte[] cmAddr = new byte[4];
+    private Ipv4Address cmtsAddr;
+
+    /**
+     * The gate classifier's srcIp value, any valid IP should work.
+     */
+    private Ipv4Address srcAddr;
+
+    /**
+     * The gate classifier's dstIp value, any valid IP should work.
+     */
+    private Ipv4Address dstAddr;
+
+    /**
+     * Defines the CMTS to add to the PCMMService
+     */
+    private Ccap ccap;
+
+    /**
+     * The class under test
+     */
+    private PCMMService service;
+
+    /**
+     * The cable modem IP address to which a gate should be set
+     */
+    private InetAddress cmAddrInet;
+    private InetAddress invalidCmAddrInet;
 
     @Before
-    public void setup() {
-        icmts = new CMTS();
-        icmts.startServer();
-
-        cmAddr[0] = 10;
-        cmAddr[1] = 32;
-        cmAddr[2] = 110;
-        cmAddr[3] = (byte)180;
+    public void setup() throws IOException {
+        srcAddr = new Ipv4Address("10.10.10.0");
+        dstAddr = new Ipv4Address("10.32.99.99");
+
+        if (realCmts) {
+            cmAddrInet = InetAddress.getByAddress(new byte[] {10, 32, 110, (byte)172});
+            invalidCmAddrInet = InetAddress.getByAddress(new byte[] {99, 99, 99, 99});
+
+            // Use me when testing against a CMTS or emulator not running in the same JVM
+            cmtsAddr = new Ipv4Address("10.32.10.3");
+            ccap = makeCcapObj(PCMMPdpAgent.WELL_KNOWN_PDP_PORT, cmtsAddr, ccapId);
+        } else {
+            cmAddrInet = InetAddress.getByAddress(new byte[] {10, 32, 110, (byte)180});
+            invalidCmAddrInet = InetAddress.getByAddress(new byte[] {99, 99, 99, 99});
+
+            // Use me for automated testing and the CMTS emulator running in the same JVM
+            cmtsAddr = new Ipv4Address("127.0.0.1");
+
+            final Set<String> upGate = new HashSet<>();
+            upGate.add("extrm_up");
+            final Set<String> dnGate = new HashSet<>();
+            dnGate.add("extrm_dn");
+            final Map<Direction, Set<String>> gates = new HashMap<>();
+            gates.put(Direction.UPSTREAM, upGate);
+            gates.put(Direction.DOWNSTREAM, dnGate);
+
+            final Map<String, Boolean> cmStatus = new HashMap<>();
+            cmStatus.put(cmAddrInet.getHostAddress(), true);
+            cmStatus.put(invalidCmAddrInet.getHostAddress(), false);
+            icmts = new CMTS(gates, cmStatus);
+            icmts.startServer();
+
+            ccap = makeCcapObj(icmts.getPort(), cmtsAddr, ccapId);
+        }
+
+        service = new PCMMService(IPCMMClient.CLIENT_TYPE, ccap);
     }
 
     @After
     public void tearDown() {
         if (icmts != null) icmts.stopServer();
+        service.disconect();
     }
 
     @Test
     public void testAddCcap() {
+        connectToCmts(service);
+    }
 
-        // TODO - comment out for testing
-        //        final Ipv4Address cmtsAddr = new Ipv4Address("10.32.10.3");
-//        final Ccaps ccap = makeCcapsObj(PCMMPdpAgent.WELL_KNOWN_PDP_PORT, cmtsAddr.getValue(), "ccap-1");
-
-        // TODO - Use this instead for automated testing
-        final Ipv4Address cmtsAddr = new Ipv4Address("127.0.0.1");
-        final Ccaps ccap = makeCcapsObj(icmts.getPort(), cmtsAddr.getValue(), "ccap-1");
-
-        final PCMMService service = new PCMMService(IPCMMClient.CLIENT_TYPE, ccap);
+    @Test
+    public void testAddInvalidCcapBadPort() {
+        final int port;
+        if (icmts != null) port = icmts.getPort() + 1;
+        else port = PCMMPdpAgent.WELL_KNOWN_PDP_PORT + 1;
+        ccap = makeCcapObj(port, cmtsAddr, ccapId);
+        service = new PCMMService(IPCMMClient.CLIENT_TYPE, ccap);
         final String message = service.addCcap();
         Assert.assertNotNull(message);
-        Assert.assertTrue(message, message.startsWith("200"));
-        Assert.assertNotNull(service.ccapClient.pcmmPdp.getClientHandle());
-        service.disconect();
+        final String expectedMsg = "404 Not Found - CCAP " + ccapId + " failed to connect @ " + cmtsAddr.getValue()
+                + ':' + port + " - ";
+        Assert.assertTrue(expectedMsg, message.startsWith(expectedMsg));
     }
 
     @Test
-    public void testAddAndRemoveUpGate() throws Exception {
+    public void testAddValidUpGateTwice() throws Exception {
+        connectToCmts(service);
+        final String expectedMsg1 = "200 OK - sendGateSet for " + ccapId + '/' + gatePath + " returned GateId";
+        addAndValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet, gatePath,
+                expectedMsg1);
 
-        // TODO - Use this block to test against a real CMTS
-//        final Ipv4Address cmtsAddr = new Ipv4Address("10.32.10.3");
-//        final Ccaps ccap = makeCcapsObj(PCMMPdpAgent.WELL_KNOWN_PDP_PORT, cmtsAddr.getValue(), "ccap-1");
+        final String expectedMsg2 = "404 Not Found - sendGateSet for " + ccapId + '/' + gatePath + " already exists";
+        addAndValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet, gatePath,
+                expectedMsg2);
 
-        // TODO - Use this block for automated testing
-        final Ipv4Address cmtsAddr = new Ipv4Address("127.0.0.1");
-        final Ccaps ccap = makeCcapsObj(icmts.getPort(), cmtsAddr.getValue(), "ccap-1");
+        Assert.assertTrue(deleteGate(service, gatePath));
+    }
 
-        final PCMMService service = new PCMMService(IPCMMClient.CLIENT_TYPE, ccap);
-        service.addCcap();
+    @Test
+    public void testAddTwoValidUpGates() throws Exception {
+        connectToCmts(service);
 
-        final Gates gate = makeGateObj("extrm_up", cmtsAddr, ServiceFlowDirection.Us);
-        final String gatePath = "testGatePath";
+        final String gatePath1 = "gatePath1";
+        final String expectedMsg1 = "200 OK - sendGateSet for " + ccapId + '/' + gatePath1 + " returned GateId";
+        addAndValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet, gatePath1,
+                expectedMsg1);
 
-        // Add gate
-        final String msg = service.sendGateSet(gatePath, InetAddress.getByAddress(cmAddr), gate,
-                ServiceFlowDirection.Us);
-        Assert.assertTrue(msg, msg.startsWith("200"));
+        final String gatePath2 = "gatePath2";
+        final String expectedMsg2 = "200 OK - sendGateSet for " + ccapId + '/' + gatePath2 + " returned GateId";
+        addAndValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet, gatePath2,
+                expectedMsg2);
 
-        // TODO - add validation to the PCMMGateReq contained within the map
-        Assert.assertEquals(1, service.gateRequests.size());
+        Assert.assertTrue(deleteGate(service, gatePath1));
+        Assert.assertTrue(deleteGate(service, gatePath2));
+    }
 
-        // Remove gate
-        service.sendGateDelete(gatePath);
+    @Test
+    public void testAddValidDownGateTwice() throws Exception {
+        connectToCmts(service);
+        final String expectedMsg1 = "200 OK - sendGateSet for " + ccapId + '/' + gatePath + " returned GateId";
+        addAndValidateGate(service, "extrm_dn", srcAddr, dstAddr, ServiceFlowDirection.Ds, cmAddrInet, gatePath,
+                expectedMsg1);
 
-        // Wait up to 1 sec for response to be processed
-        final long start = System.currentTimeMillis();
-        while (1000 < System.currentTimeMillis() - start) {
-            if (service.gateRequests.size() == 0) break;
-        }
-        Assert.assertEquals(0, service.gateRequests.size());
-        service.disconect();
+        final String expectedMsg2 = "404 Not Found - sendGateSet for " + ccapId + '/' + gatePath + " already exists";
+        addAndValidateGate(service, "extrm_dn", srcAddr, dstAddr, ServiceFlowDirection.Ds, cmAddrInet, gatePath,
+                expectedMsg2);
+
+        Assert.assertTrue(deleteGate(service, gatePath));
+    }
+
+    @Test
+    public void testDeleteNonExistentGate() throws Exception {
+        connectToCmts(service);
+        Assert.assertFalse(deleteGate(service, gatePath));
+    }
+
+    @Test
+    public void testAddAndRemoveValidUpGate() throws Exception {
+        final String expectedMsgStart = "200 OK - sendGateSet for " + ccapId + '/' + gatePath + " returned GateId";
+        addRemoveValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet, gatePath,
+                expectedMsgStart);
+    }
+
+    @Test
+    public void testAddAndRemoveValidDownGate() throws Exception {
+        final String expectedMsgStart = "200 OK - sendGateSet for " + ccapId + '/' + gatePath + " returned GateId";
+        addRemoveValidateGate(service, "extrm_dn", srcAddr, dstAddr, ServiceFlowDirection.Ds, cmAddrInet, gatePath,
+                expectedMsgStart);
+    }
+
+    @Test
+    public void testAddAndRemoveInvalidCmAddrUpGate() throws Exception {
+        // TODO - fix cmts emulator
+        final String expectedMsgStart = "404 Not Found - sendGateSet for " + ccapId + '/' + gatePath
+                + " returned error - Error Code: 13 Error Subcode : 0  Invalid SubscriberID";
+        addRemoveValidateGate(service, "extrm_up", srcAddr, dstAddr, ServiceFlowDirection.Us, invalidCmAddrInet,
+                gatePath, expectedMsgStart);
+    }
+
+    @Test
+    public void testAddInvalidScnUpGate() throws Exception {
+        final String expectedMsgStart = "404 Not Found - sendGateSet for " + ccapId + '/' + gatePath
+                + " returned error - Error Code: 11 Error Subcode : 0  Undefined Service Class Name";
+        addRemoveValidateGate(service, "extrm_up_invalid", srcAddr, dstAddr, ServiceFlowDirection.Us, cmAddrInet,
+                gatePath, expectedMsgStart);
+    }
+
+    @Test
+    public void testAddInvalidScnDownGate() throws Exception {
+        final String expectedMsgStart = "404 Not Found - sendGateSet for " + ccapId + '/' + gatePath
+                + " returned error - Error Code: 11 Error Subcode : 0  Undefined Service Class Name";
+        addRemoveValidateGate(service, "extrm_dn_invalid", srcAddr, dstAddr, ServiceFlowDirection.Ds, cmAddrInet,
+                gatePath, expectedMsgStart);
     }
 
     /**
@@ -138,11 +272,10 @@ public class PCMMServiceTest {
     @Test
     public void testGateRequestDecisionMsg() throws Exception {
         final Socket socket = new MockSocket();
-        final Ccaps ccap = makeCcapsObj(icmts.getPort(), "127.0.0.1", "ccap-1");
-        final Ipv4Address cmtsAddr = new Ipv4Address("127.0.0.1");
-        final Gates gate = makeGateObj("extrm_up", cmtsAddr, ServiceFlowDirection.Us);
-        final IPCMMGate gateReq = makeGateRequest(ccap, gate, InetAddress.getByName("localhost"),
-                ServiceFlowDirection.Us);
+
+        final ServiceFlowDirection direction = ServiceFlowDirection.Us;
+        final Gate gate = makeGateObj("extrm_up", cmtsAddr, direction, new Ipv4Address("127.0.0.1"));
+        final IPCMMGate gateReq = makeGateRequest(ccap, gate, InetAddress.getByName("localhost"), direction);
         final byte[] data = gateReq.getData();
 
         final Set<COPSDecision> decisionSet = new HashSet<>();
@@ -161,14 +294,91 @@ public class PCMMServiceTest {
     }
 
     /**
-     * Creates a mock Ccaps object that can be used for connecting to a CMTS
+     * Attempts to create a gate against a CMTS, validates the results then attempts to delete it.
+     * @param service - the service used to connect to a CMTS for issuing requests
+     * @param scnName - the service class name (aka. gate name)
+     * @param srcAddr - the address to the CMTS subnet?
+     * @param dstAddr - the destination address
+     * @param direction - the gate direction
+     * @param cmAddrInet - the address to the cable modem to which the gate will be assigned
+     * @param gatePath - the path to the gate
+     * @param expGateSetMsgStart - the expected start of the gate set return message to be validated against
+     */
+    private void addRemoveValidateGate(final PCMMService service, final String scnName, final Ipv4Address srcAddr,
+                                       final Ipv4Address dstAddr, final ServiceFlowDirection direction,
+                                       final InetAddress cmAddrInet, final String gatePath,
+                                       final String expGateSetMsgStart) {
+        connectToCmts(service);
+        addAndValidateGate(service, scnName, srcAddr, dstAddr, direction, cmAddrInet, gatePath, expGateSetMsgStart);
+        deleteGate(service, gatePath);
+    }
+
+    private void connectToCmts(final PCMMService service) {
+        final String message = service.addCcap();
+        Assert.assertNotNull(message);
+        final String expectedMsg = "200 OK - CCAP " + ccapId + " connected @ "
+                + ccap.getConnection().getIpAddress().getIpv4Address().getValue()
+                + ":" + ccap.getConnection().getPort().getValue();
+        Assert.assertEquals(expectedMsg, message);
+        Assert.assertNotNull(service.ccapClient.pcmmPdp.getClientHandle());
+    }
+
+    /**
+     * Attempts to create a gate against a CMTS and validates the results.
+     * @param service - the service used to connect to a CMTS for issuing requests
+     * @param scnName - the service class name (aka. gate name)
+     * @param srcAddr - the address to the CMTS subnet?
+     * @param dstAddr - the destination address
+     * @param direction - the gate direction
+     * @param cmAddrInet - the address to the cable modem to which the gate will be assigned
+     * @param gatePath - the path to the gate
+     * @param expGateSetMsgStart - the expected start of the gate set return message to be validated against
+     */
+    private void addAndValidateGate(final PCMMService service, final String scnName, final Ipv4Address srcAddr,
+                                    final Ipv4Address dstAddr, final ServiceFlowDirection direction,
+                                    final InetAddress cmAddrInet, final String gatePath,
+                                    final String expGateSetMsgStart) {
+        final Gate gate = makeGateObj(scnName, srcAddr, direction, dstAddr);
+
+//        final String gateSetMsg = service.sendGateSet(gatePath, cmAddrInet, gate, direction);
+//        Assert.assertNotNull(gateSetMsg);
+//        Assert.assertTrue(gateSetMsg, gateSetMsg.startsWith(expGateSetMsgStart));
+
+        // TODO update this method for the new GateSetStatus object
+        PCMMService.GateSetStatus status = service.sendGateSet(gatePath, cmAddrInet, gate, direction);
+        Assert.assertNotNull(status);
+        Assert.assertTrue(status.getMessage().startsWith(expGateSetMsgStart));
+
+        // TODO - add validation to the PCMMGateReq contained within the map
+        Assert.assertNotNull(service.gateRequests.get(gatePath));
+    }
+
+    /**
+     * Attempts to delete a gate
+     * @param service - the service used to connect to a CMTS for issuing requests
+     * @param gatePath - the path to the gate
+     */
+    private boolean deleteGate(final PCMMService service, final String gatePath) {
+        final boolean out = service.sendGateDelete(gatePath);
+
+        // Wait up to 1 sec for response to be processed
+        final long start = System.currentTimeMillis();
+        while (1000 < System.currentTimeMillis() - start) {
+            if (service.gateRequests.size() == 0) break;
+        }
+        Assert.assertNull(service.gateRequests.get(gatePath));
+        return out;
+    }
+
+    /**
+     * Creates a mock Ccap object that can be used for connecting to a CMTS
      * @param inPort - the CMTS port number
      * @param ipAddr - the CMTS IPv4 address string
      * @param ccapId - the ID of the CCAP
-     * @return - the mock Ccaps object
+     * @return - the mock Ccap object
      */
-    private Ccaps makeCcapsObj(final int inPort, final String ipAddr, final String ccapId) {
-        final Ccaps ccap = Mockito.mock(Ccaps.class);
+    private Ccap makeCcapObj(final int inPort, final Ipv4Address ipAddr, final String ccapId) {
+        final Ccap ccap = Mockito.mock(Ccap.class);
         final Connection conn = Mockito.mock(Connection.class);
         Mockito.when(ccap.getConnection()).thenReturn(conn);
         final PortNumber port = Mockito.mock(PortNumber.class);
@@ -177,8 +387,7 @@ public class PCMMServiceTest {
 
         final IpAddress addr = Mockito.mock(IpAddress.class);
         Mockito.when(conn.getIpAddress()).thenReturn(addr);
-        final Ipv4Address ipv4 = new Ipv4Address(ipAddr);
-        Mockito.when(addr.getIpv4Address()).thenReturn(ipv4);
+        Mockito.when(addr.getIpv4Address()).thenReturn(ipAddr);
 
         Mockito.when(ccap.getCcapId()).thenReturn(ccapId);
         final AmId amid = Mockito.mock(AmId.class);
@@ -190,13 +399,14 @@ public class PCMMServiceTest {
     }
 
     /**
-     * Creates a mock Gates object
+     * Creates a mock Gate object
      * @param scnValue - the service class name defined on the CMTS
      * @param dstAddr - the CM address this gate should be set against
      * @return - the gate request
      */
-    private Gates makeGateObj(final String scnValue, final Ipv4Address dstAddr, final ServiceFlowDirection direction) {
-        final Gates gate = Mockito.mock(Gates.class);
+    private Gate makeGateObj(final String scnValue, final Ipv4Address srcAddr, final ServiceFlowDirection direction,
+                              final Ipv4Address dstAddr) {
+        final Gate gate = Mockito.mock(Gate.class);
         final GateSpec gateSpec = Mockito.mock(GateSpec.class);
         Mockito.when(gate.getGateSpec()).thenReturn(gateSpec);
         Mockito.when(gateSpec.getDirection()).thenReturn(direction);
@@ -218,11 +428,12 @@ public class PCMMServiceTest {
         Mockito.when(classifier.getDstPort()).thenReturn(dstPort);
         final TpProtocol protocol = new TpProtocol(0);
         Mockito.when(classifier.getProtocol()).thenReturn(protocol);
-        final Ipv4Address srcAddr = new Ipv4Address("127.0.0.1");
         Mockito.when(classifier.getSrcIp()).thenReturn(srcAddr);
         final PortNumber srcPort = new PortNumber(1234);
         Mockito.when(classifier.getSrcPort()).thenReturn(srcPort);
-        final TosByte tosByte = new TosByte((short)160);
+
+        // TODO - Can this value be any other value than 0 or 1 (See TosByte enumeration)
+        final TosByte tosByte = new TosByte((short)0);
         Mockito.when(classifier.getTosByte()).thenReturn(tosByte);
         final TosByte tosMask = new TosByte((short)224);
         Mockito.when(classifier.getTosMask()).thenReturn(tosMask);
@@ -234,31 +445,31 @@ public class PCMMServiceTest {
         return gate;
     }
 
-    private IPCMMGate makeGateRequest(final Ccaps ccap, final Gates gateReq, final InetAddress addrSubId,
+    private IPCMMGate makeGateRequest(final Ccap ccap, final Gate gateReq, final InetAddress addrSubId,
                                      final ServiceFlowDirection direction) {
         final PCMMGateReqBuilder gateBuilder = new PCMMGateReqBuilder();
-        gateBuilder.build(ccap.getAmId());
-        gateBuilder.build(addrSubId);
+        gateBuilder.setAmId(ccap.getAmId());
+        gateBuilder.setSubscriberId(addrSubId);
         // force gateSpec.Direction to align with SCN direction
         final ServiceClassName scn = gateReq.getTrafficProfile().getServiceClassName();
         if (scn != null) {
-            gateBuilder.build(gateReq.getGateSpec(), direction);
+            gateBuilder.setGateSpec(gateReq.getGateSpec(), direction);
         } else {
             // not an SCN gate
-            gateBuilder.build(gateReq.getGateSpec(), null);
+            gateBuilder.setGateSpec(gateReq.getGateSpec(), null);
         }
-        gateBuilder.build(gateReq.getTrafficProfile());
+        gateBuilder.setTrafficProfile(gateReq.getTrafficProfile());
 
         // pick a classifier type (only one for now)
         if (gateReq.getClassifier() != null) {
-            gateBuilder.build(gateReq.getClassifier());
+            gateBuilder.setClassifier(gateReq.getClassifier());
         } else if (gateReq.getExtClassifier() != null) {
-            gateBuilder.build(gateReq.getExtClassifier());
+            gateBuilder.setExtClassifier(gateReq.getExtClassifier());
         } else if (gateReq.getIpv6Classifier() != null) {
-            gateBuilder.build(gateReq.getIpv6Classifier());
+            gateBuilder.setIpv6Classifier(gateReq.getIpv6Classifier());
         }
         // assemble the final gate request
-        return gateBuilder.getGateReq();
+        return gateBuilder.build();
     }
 
     private class MockSocket extends Socket {