Examples with IPPacket com.att.aro.core.packetreader.pojo.IPPacket used on opensource projects

Example 11 with IPPacket

use of com.att.aro.core.packetreader.pojo.IPPacket in project VideoOptimzer by attdevsupport.

the class SessionManagerImplTest method assembleSessionTest1.

@Test
public void assembleSessionTest1() {
    Date date = new Date();
    /*		InetAddress iAdr = null;
		InetAddress iAdr1 = null;
		try {
			iAdr = InetAddress.getByAddress(new byte[]{89,10,1,1});
			iAdr1 = InetAddress.getByAddress(new byte[]{89,10,1,1});
		} catch (UnknownHostException e) {
			e.printStackTrace();
		}*/
    InetAddress iAdr = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr.getAddress()).thenReturn(new byte[] { 89, 10, 1, 1 });
    Mockito.when(iAdr.getHostAddress()).thenReturn("@att");
    Mockito.when(iAdr.getHostName()).thenReturn("att");
    InetAddress iAdr1 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr1.getAddress()).thenReturn(new byte[] { 89, 10, 1, 1 });
    Mockito.when(iAdr1.getHostAddress()).thenReturn("@att");
    Mockito.when(iAdr1.getHostName()).thenReturn("att");
    Set<InetAddress> inetSet = new HashSet<InetAddress>();
    inetSet.add(iAdr);
    inetSet.add(iAdr1);
    IPPacket ipPack = Mockito.mock(IPPacket.class);
    Mockito.when(ipPack.getDestinationIPAddress()).thenReturn(iAdr1);
    Mockito.when(ipPack.getSourceIPAddress()).thenReturn(iAdr);
    Mockito.when(ipPack.getLen()).thenReturn(20);
    Mockito.when(ipPack.getPacketLength()).thenReturn(20);
    Mockito.when(ipPack.getPayloadLen()).thenReturn(10);
    DomainNameSystem dns = Mockito.mock(DomainNameSystem.class);
    Mockito.when(dns.getIpAddresses()).thenReturn(inetSet);
    Mockito.when(dns.getDomainName()).thenReturn("www.att.com");
    Mockito.when(dns.isResponse()).thenReturn(true);
    Mockito.when(dns.getCname()).thenReturn("ATT");
    Mockito.when(dns.getPacket()).thenReturn(ipPack);
    // UDP Packet Mock
    UDPPacket udpPacket = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket.isDNSPacket()).thenReturn(false);
    Mockito.when(udpPacket.getDns()).thenReturn(dns);
    Mockito.when(udpPacket.getSourcePort()).thenReturn(83);
    Mockito.when(udpPacket.getDestinationPort()).thenReturn(84);
    Mockito.when(udpPacket.getSourceIPAddress()).thenReturn(iAdr);
    Mockito.when(udpPacket.getDestinationIPAddress()).thenReturn(iAdr1);
    PacketInfo packetInfo1 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo1.getPacket()).thenReturn(udpPacket);
    Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo1.getPayloadLen()).thenReturn(20);
    Mockito.when(packetInfo1.getLen()).thenReturn(10);
    Mockito.when(packetInfo1.getAppName()).thenReturn("Test1");
    Mockito.when(packetInfo1.getTcpFlagString()).thenReturn("TestString");
    InetAddress iAdr2 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr2.getAddress()).thenReturn(new byte[] { 89, 10, 1, 1 });
    Mockito.when(iAdr2.getHostAddress()).thenReturn("@att");
    Mockito.when(iAdr2.getHostName()).thenReturn("att");
    /*		InetAddress iAdr2 = null;
		InetAddress iAdr3 = null;
		try {
			iAdr2 = InetAddress.getByAddress(new byte[]{89,10,1,1});
			iAdr3 = InetAddress.getByAddress(new byte[]{89,10,1,1});
		} catch (UnknownHostException e) {
			e.printStackTrace();
		}
*/
    TCPPacket tcpPacket = Mockito.mock(TCPPacket.class);
    Mockito.when(tcpPacket.getSourcePort()).thenReturn(81);
    Mockito.when(tcpPacket.getDestinationPort()).thenReturn(82);
    Mockito.when(tcpPacket.getDestinationIPAddress()).thenReturn(iAdr2);
    Mockito.when(tcpPacket.getSourceIPAddress()).thenReturn(iAdr2);
    Mockito.when(tcpPacket.getAckNumber()).thenReturn(0L);
    Mockito.when(tcpPacket.getLen()).thenReturn(50);
    Mockito.when(tcpPacket.isSYN()).thenReturn(true);
    Mockito.when(tcpPacket.isFIN()).thenReturn(true);
    Mockito.when(tcpPacket.isRST()).thenReturn(true);
    Mockito.when(tcpPacket.getSequenceNumber()).thenReturn(-3L);
    Mockito.when(tcpPacket.getTimeStamp()).thenReturn((double) date.getTime() - 3000);
    // Mockito.when(tcpPacket.getTimeStamp()).thenReturn((double)date.getTime()-10000);
    PacketInfo packetInfo2 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo2.getPacket()).thenReturn(tcpPacket);
    Mockito.when(packetInfo2.getDir()).thenReturn(PacketDirection.DOWNLINK);
    Mockito.when(packetInfo2.getTcpInfo()).thenReturn(TcpInfo.TCP_ESTABLISH);
    Mockito.when(packetInfo2.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo2.getLen()).thenReturn(15);
    Mockito.when(packetInfo2.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo2.getTcpFlagString()).thenReturn("Test2String");
    InetAddress iAdr3 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr3.getAddress()).thenReturn(new byte[] { 89, 10, 1, 1 });
    Mockito.when(iAdr3.getHostAddress()).thenReturn("@att");
    Mockito.when(iAdr3.getHostName()).thenReturn("att");
    TCPPacket tcpPacket2 = Mockito.mock(TCPPacket.class);
    Mockito.when(tcpPacket2.getSourcePort()).thenReturn(95);
    Mockito.when(tcpPacket2.getDestinationPort()).thenReturn(99);
    Mockito.when(tcpPacket2.getDestinationIPAddress()).thenReturn(iAdr3);
    Mockito.when(tcpPacket2.getSourceIPAddress()).thenReturn(iAdr3);
    Mockito.when(tcpPacket.getAckNumber()).thenReturn(0L);
    Mockito.when(tcpPacket2.isSYN()).thenReturn(true);
    Mockito.when(tcpPacket2.isFIN()).thenReturn(true);
    Mockito.when(tcpPacket2.isRST()).thenReturn(true);
    Mockito.when(tcpPacket.getSequenceNumber()).thenReturn(1L);
    Mockito.when(tcpPacket2.getTimeStamp()).thenReturn(10000d);
    PacketInfo packetInfo3 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo3.getPacket()).thenReturn(tcpPacket2);
    Mockito.when(packetInfo3.getDir()).thenReturn(PacketDirection.UPLINK);
    Mockito.when(packetInfo3.getTcpInfo()).thenReturn(TcpInfo.TCP_ACK_RECOVER);
    Mockito.when(packetInfo3.getPayloadLen()).thenReturn(30);
    Mockito.when(packetInfo3.getLen()).thenReturn(20);
    Mockito.when(packetInfo3.getAppName()).thenReturn("Test3");
    Mockito.when(packetInfo3.getTcpFlagString()).thenReturn("Test3String");
    /*		
		PacketInfo packetInfo4 = Mockito.mock(PacketInfo.class);
		Mockito.when(packetInfo4.getPacket()).thenReturn(tcpPacket2);
		Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.DOWNLINK);
		Mockito.when(packetInfo4.getTcpInfo()).thenReturn(TcpInfo.TCP_ACK);
		
		PacketInfo packetInfo5 = Mockito.mock(PacketInfo.class);
		Mockito.when(packetInfo5.getPacket()).thenReturn(tcpPacket2);
		Mockito.when(packetInfo5.getDir()).thenReturn(PacketDirection.DOWNLINK);
		Mockito.when(packetInfo5.getTcpInfo()).thenReturn(TcpInfo.TCP_DATA_RECOVER);
*/
    InetAddress iAdr4 = Mockito.mock(InetAddress.class);
    InetAddress iAdr5 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr4.getAddress()).thenReturn(new byte[] { 89, 10, 5, 1 });
    Mockito.when(iAdr4.getHostAddress()).thenReturn("@microfoft");
    Mockito.when(iAdr4.getHostName()).thenReturn("microfoft");
    Mockito.when(iAdr5.getAddress()).thenReturn(new byte[] { 72, 12, 23, 1 });
    Mockito.when(iAdr5.getHostAddress()).thenReturn("@apple");
    Mockito.when(iAdr5.getHostName()).thenReturn("apple");
    Set<InetAddress> inetSet1 = new HashSet<InetAddress>();
    inetSet1.add(iAdr4);
    inetSet1.add(iAdr5);
    IPPacket ipPack1 = Mockito.mock(IPPacket.class);
    Mockito.when(ipPack1.getDestinationIPAddress()).thenReturn(iAdr5);
    Mockito.when(ipPack1.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(ipPack1.getLen()).thenReturn(30);
    Mockito.when(ipPack1.getPacketLength()).thenReturn(40);
    Mockito.when(ipPack1.getPayloadLen()).thenReturn(40);
    DomainNameSystem dns1 = Mockito.mock(DomainNameSystem.class);
    Mockito.when(dns1.getIpAddresses()).thenReturn(inetSet);
    Mockito.when(dns1.getDomainName()).thenReturn("www.microsft.com");
    Mockito.when(dns1.isResponse()).thenReturn(false);
    Mockito.when(dns1.getCname()).thenReturn("microsoft");
    Mockito.when(dns1.getPacket()).thenReturn(ipPack1);
    // UDP Packet Mock
    UDPPacket udpPacket1 = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket1.isDNSPacket()).thenReturn(true);
    Mockito.when(udpPacket1.getDns()).thenReturn(dns1);
    Mockito.when(udpPacket1.getSourcePort()).thenReturn(90);
    Mockito.when(udpPacket1.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(udpPacket1.getDestinationPort()).thenReturn(91);
    Mockito.when(udpPacket1.getDestinationIPAddress()).thenReturn(iAdr5);
    PacketInfo packetInfo4 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo4.getPacket()).thenReturn(udpPacket);
    Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.DOWNLINK);
    // Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo4.getPayloadLen()).thenReturn(20);
    Mockito.when(packetInfo4.getLen()).thenReturn(10);
    Mockito.when(packetInfo4.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo4.getTcpFlagString()).thenReturn("TestString2");
    PacketInfo packetInfo5 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo5.getPacket()).thenReturn(tcpPacket);
    Mockito.when(packetInfo5.getDir()).thenReturn(PacketDirection.UPLINK);
    Mockito.when(packetInfo5.getTcpInfo()).thenReturn(TcpInfo.TCP_ESTABLISH);
    Mockito.when(packetInfo5.getPayloadLen()).thenReturn(20);
    Mockito.when(packetInfo5.getLen()).thenReturn(15);
    Mockito.when(packetInfo5.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo5.getTcpFlagString()).thenReturn("Test2String");
    // UDP Packet Mock
    UDPPacket udpPacket2 = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket2.isDNSPacket()).thenReturn(true);
    Mockito.when(udpPacket2.getDns()).thenReturn(dns1);
    Mockito.when(udpPacket2.getSourcePort()).thenReturn(90);
    Mockito.when(udpPacket2.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(udpPacket2.getDestinationPort()).thenReturn(91);
    Mockito.when(udpPacket2.getDestinationIPAddress()).thenReturn(iAdr5);
    PacketInfo packetInfo6 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo6.getPacket()).thenReturn(udpPacket);
    // Mockito.when(packetInfo6.getDir()).thenReturn(PacketDirection.DOWNLINK);
    Mockito.when(packetInfo6.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo6.getPayloadLen()).thenReturn(30);
    Mockito.when(packetInfo6.getLen()).thenReturn(10);
    Mockito.when(packetInfo6.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo6.getTcpFlagString()).thenReturn("TestString2");
    InetAddress iAdr6 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr6.getAddress()).thenReturn(new byte[] { 60, 100, 1, 1 });
    Mockito.when(iAdr6.getHostAddress()).thenReturn("@hari");
    Mockito.when(iAdr6.getHostName()).thenReturn("hari");
    TCPPacket tcpPacket6 = Mockito.mock(TCPPacket.class);
    Mockito.when(tcpPacket6.getSourcePort()).thenReturn(95);
    Mockito.when(tcpPacket6.getDestinationPort()).thenReturn(96);
    Mockito.when(tcpPacket6.getDestinationIPAddress()).thenReturn(iAdr6);
    Mockito.when(tcpPacket6.getSourceIPAddress()).thenReturn(iAdr6);
    Mockito.when(tcpPacket6.getAckNumber()).thenReturn(1L);
    Mockito.when(tcpPacket6.getLen()).thenReturn(50);
    Mockito.when(tcpPacket6.isSYN()).thenReturn(true);
    Mockito.when(tcpPacket6.isFIN()).thenReturn(true);
    Mockito.when(tcpPacket6.isRST()).thenReturn(false);
    Mockito.when(tcpPacket6.getSequenceNumber()).thenReturn(5L);
    Mockito.when(tcpPacket6.getTimeStamp()).thenReturn((double) date.getTime() - 20000);
    // Mockito.when(tcpPacket.getTimeStamp()).thenReturn((double)date.getTime()-10000);
    PacketInfo packetInfo7 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo7.getPacket()).thenReturn(tcpPacket);
    Mockito.when(packetInfo7.getDir()).thenReturn(PacketDirection.UNKNOWN);
    Mockito.when(packetInfo7.getTcpInfo()).thenReturn(TcpInfo.TCP_ACK_DUP);
    Mockito.when(packetInfo7.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo7.getLen()).thenReturn(15);
    Mockito.when(packetInfo7.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo7.getTcpFlagString()).thenReturn("Test2String");
    List<PacketInfo> packetsList = new ArrayList<PacketInfo>();
    // Adding UDP Packet to the list
    packetsList.add(packetInfo1);
    // Adding TCP Packet to the list
    packetsList.add(packetInfo2);
    packetsList.add(packetInfo3);
    packetsList.add(packetInfo4);
    packetsList.add(packetInfo5);
    packetsList.add(packetInfo6);
    packetsList.add(packetInfo7);
    List<Session> sessionsList = sessionMgr.processPacketsAndAssembleSessions(packetsList);
    assertEquals(5, sessionsList.size());
}

Example 12 with IPPacket

use of com.att.aro.core.packetreader.pojo.IPPacket in project VideoOptimzer by attdevsupport.

the class SessionManagerImplTest method assembleSessionTest.

@Test
public void assembleSessionTest() {
    Date date = new Date();
    InetAddress iAdr = null;
    InetAddress iAdr1 = null;
    try {
        iAdr = InetAddress.getByAddress(new byte[] { 89, 10, 1, 1 });
        iAdr1 = InetAddress.getByAddress(new byte[] { 89, 10, 1, 1 });
    } catch (UnknownHostException e) {
        e.printStackTrace();
    }
    // Mockito.when(iAdr.getAddress()).thenReturn(new byte[]{89,10,1,1});
    // Mockito.when(iAdr.getHostAddress()).thenReturn("@att");
    // Mockito.when(iAdr.getHostName()).thenReturn("att");
    // Mockito.when(iAdr1.getAddress()).thenReturn(new byte[]{89,10,1,1});
    // Mockito.when(iAdr1.getHostAddress()).thenReturn("@att");
    // Mockito.when(iAdr1.getHostName()).thenReturn("att");
    Set<InetAddress> inetSet = new HashSet<InetAddress>();
    inetSet.add(iAdr);
    inetSet.add(iAdr1);
    IPPacket ipPack = Mockito.mock(IPPacket.class);
    Mockito.when(ipPack.getDestinationIPAddress()).thenReturn(iAdr1);
    Mockito.when(ipPack.getSourceIPAddress()).thenReturn(iAdr);
    Mockito.when(ipPack.getLen()).thenReturn(20);
    Mockito.when(ipPack.getPacketLength()).thenReturn(20);
    Mockito.when(ipPack.getPayloadLen()).thenReturn(20);
    DomainNameSystem dns = Mockito.mock(DomainNameSystem.class);
    Mockito.when(dns.getIpAddresses()).thenReturn(inetSet);
    Mockito.when(dns.getDomainName()).thenReturn("www.att.com");
    Mockito.when(dns.isResponse()).thenReturn(true);
    Mockito.when(dns.getCname()).thenReturn("ATT");
    Mockito.when(dns.getPacket()).thenReturn(ipPack);
    // UDP Packet Mock
    UDPPacket udpPacket = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket.isDNSPacket()).thenReturn(true);
    Mockito.when(udpPacket.getDns()).thenReturn(dns);
    Mockito.when(udpPacket.getSourcePort()).thenReturn(83);
    Mockito.when(udpPacket.getDestinationPort()).thenReturn(84);
    Mockito.when(udpPacket.getSourceIPAddress()).thenReturn(iAdr);
    Mockito.when(udpPacket.getDestinationIPAddress()).thenReturn(iAdr1);
    PacketInfo packetInfo1 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo1.getPacket()).thenReturn(udpPacket);
    Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo1.getLen()).thenReturn(10);
    Mockito.when(packetInfo1.getAppName()).thenReturn("Test1");
    Mockito.when(packetInfo1.getTcpFlagString()).thenReturn("TestString");
    /*		InetAddress iAdr2 = Mockito.mock(InetAddress.class);
		Mockito.when(iAdr2.getAddress()).thenReturn(new byte[]{89,10,1,1});
		Mockito.when(iAdr2.getHostAddress()).thenReturn("@att");
		Mockito.when(iAdr2.getHostName()).thenReturn("att");*/
    InetAddress iAdr2 = null;
    InetAddress iAdr3 = null;
    try {
        iAdr2 = InetAddress.getByAddress(new byte[] { 89, 10, 1, 1 });
        iAdr3 = InetAddress.getByAddress(new byte[] { 89, 10, 1, 1 });
    } catch (UnknownHostException e) {
        e.printStackTrace();
    }
    TCPPacket tcpPacket = Mockito.mock(TCPPacket.class);
    Mockito.when(tcpPacket.getSourcePort()).thenReturn(81);
    Mockito.when(tcpPacket.getDestinationPort()).thenReturn(82);
    Mockito.when(tcpPacket.getDestinationIPAddress()).thenReturn(iAdr2);
    Mockito.when(tcpPacket.getSourceIPAddress()).thenReturn(iAdr2);
    Mockito.when(tcpPacket.getAckNumber()).thenReturn(0L);
    Mockito.when(tcpPacket.getLen()).thenReturn(50);
    Mockito.when(tcpPacket.isSYN()).thenReturn(true);
    Mockito.when(tcpPacket.isFIN()).thenReturn(true);
    Mockito.when(tcpPacket.isRST()).thenReturn(true);
    Mockito.when(tcpPacket.getSequenceNumber()).thenReturn(20L);
    Mockito.when(tcpPacket.getTimeStamp()).thenReturn((double) date.getTime() - 3000);
    // Mockito.when(tcpPacket.getTimeStamp()).thenReturn((double)date.getTime()-10000);
    PacketInfo packetInfo2 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo2.getPacket()).thenReturn(tcpPacket);
    Mockito.when(packetInfo2.getDir()).thenReturn(PacketDirection.UPLINK);
    Mockito.when(packetInfo2.getTcpInfo()).thenReturn(TcpInfo.TCP_ESTABLISH);
    Mockito.when(packetInfo2.getPayloadLen()).thenReturn(20);
    Mockito.when(packetInfo2.getLen()).thenReturn(15);
    Mockito.when(packetInfo2.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo2.getTcpFlagString()).thenReturn("Test2String");
    /*		InetAddress iAdr3 = Mockito.mock(InetAddress.class);
		Mockito.when(iAdr3.getAddress()).thenReturn(new byte[]{89,10,1,1});
		Mockito.when(iAdr3.getHostAddress()).thenReturn("@att");
		Mockito.when(iAdr3.getHostName()).thenReturn("att");
*/
    TCPPacket tcpPacket2 = Mockito.mock(TCPPacket.class);
    Mockito.when(tcpPacket2.getSourcePort()).thenReturn(95);
    Mockito.when(tcpPacket2.getDestinationPort()).thenReturn(99);
    Mockito.when(tcpPacket2.getDestinationIPAddress()).thenReturn(iAdr3);
    Mockito.when(tcpPacket2.getSourceIPAddress()).thenReturn(iAdr3);
    Mockito.when(tcpPacket.getAckNumber()).thenReturn(0L);
    Mockito.when(tcpPacket2.isSYN()).thenReturn(true);
    Mockito.when(tcpPacket2.isFIN()).thenReturn(true);
    Mockito.when(tcpPacket2.isRST()).thenReturn(true);
    Mockito.when(tcpPacket.getSequenceNumber()).thenReturn(1L);
    Mockito.when(tcpPacket2.getTimeStamp()).thenReturn(10000d);
    PacketInfo packetInfo3 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo3.getPacket()).thenReturn(tcpPacket2);
    Mockito.when(packetInfo3.getDir()).thenReturn(PacketDirection.DOWNLINK);
    Mockito.when(packetInfo3.getTcpInfo()).thenReturn(TcpInfo.TCP_ACK_RECOVER);
    Mockito.when(packetInfo3.getPayloadLen()).thenReturn(30);
    Mockito.when(packetInfo3.getLen()).thenReturn(20);
    Mockito.when(packetInfo3.getAppName()).thenReturn("Test3");
    Mockito.when(packetInfo3.getTcpFlagString()).thenReturn("Test3String");
    /*		
		PacketInfo packetInfo4 = Mockito.mock(PacketInfo.class);
		Mockito.when(packetInfo4.getPacket()).thenReturn(tcpPacket2);
		Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.DOWNLINK);
		Mockito.when(packetInfo4.getTcpInfo()).thenReturn(TcpInfo.TCP_ACK);
		
		PacketInfo packetInfo5 = Mockito.mock(PacketInfo.class);
		Mockito.when(packetInfo5.getPacket()).thenReturn(tcpPacket2);
		Mockito.when(packetInfo5.getDir()).thenReturn(PacketDirection.DOWNLINK);
		Mockito.when(packetInfo5.getTcpInfo()).thenReturn(TcpInfo.TCP_DATA_RECOVER);
*/
    InetAddress iAdr4 = Mockito.mock(InetAddress.class);
    InetAddress iAdr5 = Mockito.mock(InetAddress.class);
    Mockito.when(iAdr4.getAddress()).thenReturn(new byte[] { 89, 10, 5, 1 });
    Mockito.when(iAdr4.getHostAddress()).thenReturn("@microfoft");
    Mockito.when(iAdr4.getHostName()).thenReturn("microfoft");
    Mockito.when(iAdr5.getAddress()).thenReturn(new byte[] { 72, 12, 23, 1 });
    Mockito.when(iAdr5.getHostAddress()).thenReturn("@apple");
    Mockito.when(iAdr5.getHostName()).thenReturn("apple");
    Set<InetAddress> inetSet1 = new HashSet<InetAddress>();
    inetSet1.add(iAdr4);
    inetSet1.add(iAdr5);
    IPPacket ipPack1 = Mockito.mock(IPPacket.class);
    Mockito.when(ipPack1.getDestinationIPAddress()).thenReturn(iAdr5);
    Mockito.when(ipPack1.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(ipPack1.getLen()).thenReturn(30);
    Mockito.when(ipPack1.getPacketLength()).thenReturn(40);
    Mockito.when(ipPack1.getPayloadLen()).thenReturn(40);
    DomainNameSystem dns1 = Mockito.mock(DomainNameSystem.class);
    Mockito.when(dns1.getIpAddresses()).thenReturn(inetSet);
    Mockito.when(dns1.getDomainName()).thenReturn("www.att.com");
    Mockito.when(dns1.isResponse()).thenReturn(false);
    Mockito.when(dns1.getCname()).thenReturn("ATT");
    Mockito.when(dns1.getPacket()).thenReturn(ipPack1);
    // UDP Packet Mock
    UDPPacket udpPacket1 = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket1.isDNSPacket()).thenReturn(true);
    Mockito.when(udpPacket1.getDns()).thenReturn(dns1);
    Mockito.when(udpPacket1.getSourcePort()).thenReturn(90);
    Mockito.when(udpPacket1.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(udpPacket1.getDestinationPort()).thenReturn(91);
    Mockito.when(udpPacket1.getDestinationIPAddress()).thenReturn(iAdr5);
    PacketInfo packetInfo4 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo4.getPacket()).thenReturn(udpPacket);
    Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.DOWNLINK);
    // Mockito.when(packetInfo4.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo4.getPayloadLen()).thenReturn(30);
    Mockito.when(packetInfo4.getLen()).thenReturn(10);
    Mockito.when(packetInfo4.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo4.getTcpFlagString()).thenReturn("TestString2");
    PacketInfo packetInfo5 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo5.getPacket()).thenReturn(tcpPacket);
    Mockito.when(packetInfo5.getDir()).thenReturn(PacketDirection.DOWNLINK);
    Mockito.when(packetInfo5.getTcpInfo()).thenReturn(TcpInfo.TCP_ESTABLISH);
    Mockito.when(packetInfo5.getPayloadLen()).thenReturn(20);
    Mockito.when(packetInfo5.getLen()).thenReturn(15);
    Mockito.when(packetInfo5.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo5.getTcpFlagString()).thenReturn("Test2String");
    // UDP Packet Mock
    UDPPacket udpPacket2 = Mockito.mock(UDPPacket.class);
    Mockito.when(udpPacket2.isDNSPacket()).thenReturn(true);
    Mockito.when(udpPacket2.getDns()).thenReturn(dns1);
    Mockito.when(udpPacket2.getSourcePort()).thenReturn(90);
    Mockito.when(udpPacket2.getSourceIPAddress()).thenReturn(iAdr4);
    Mockito.when(udpPacket2.getDestinationPort()).thenReturn(91);
    Mockito.when(udpPacket2.getDestinationIPAddress()).thenReturn(iAdr5);
    PacketInfo packetInfo6 = Mockito.mock(PacketInfo.class);
    Mockito.when(packetInfo6.getPacket()).thenReturn(udpPacket);
    // Mockito.when(packetInfo6.getDir()).thenReturn(PacketDirection.DOWNLINK);
    Mockito.when(packetInfo6.getDir()).thenReturn(PacketDirection.UNKNOWN);
    // Mockito.when(packetInfo1.getDir()).thenReturn(PacketDirection.UPLINK);
    // Mockito.when(packetInfo1.getPayloadLen()).thenReturn(10);
    Mockito.when(packetInfo6.getPayloadLen()).thenReturn(30);
    Mockito.when(packetInfo6.getLen()).thenReturn(10);
    Mockito.when(packetInfo6.getAppName()).thenReturn("Test2");
    Mockito.when(packetInfo6.getTcpFlagString()).thenReturn("TestString2");
    List<PacketInfo> packetsList = new ArrayList<PacketInfo>();
    // Adding UDP Packet to the list
    packetsList.add(packetInfo1);
    // Adding TCP Packet to the list
    packetsList.add(packetInfo2);
    packetsList.add(packetInfo3);
    packetsList.add(packetInfo4);
    packetsList.add(packetInfo5);
    packetsList.add(packetInfo6);
    List<Session> sessionsList = sessionMgr.processPacketsAndAssembleSessions(packetsList);
    assertEquals(5, sessionsList.size());
}

Example 13 with IPPacket

use of com.att.aro.core.packetreader.pojo.IPPacket in project VideoOptimzer by attdevsupport.

the class TraceDataReaderImplTest method readTraceDir_.

@Test
public void readTraceDir_() throws IOException {
    String[] time = { "1410212153 1410213352", "272927100", "1410213352.550" };
    String[] appId = { "5", "5", "13", "-127" };
    traceDataReaderImpl.setFileReader(filereader);
    when(filereader.directoryExist(any(String.class))).thenReturn(true);
    when(filereader.fileExist(any(String.class))).thenReturn(true);
    when(filereader.readAllLine(any(String.class))).thenReturn(time);
    when(filereader.readAllLine(Util.getCurrentRunningDir() + Util.FILE_SEPARATOR + "appid")).thenReturn(appId);
    when(filereader.fileExist(Util.getCurrentRunningDir() + Util.FILE_SEPARATOR + "traffic1.cap")).thenReturn(false);
    Map<String, List<ScheduledAlarmInfo>> scheduledAlarms = new HashMap<String, List<ScheduledAlarmInfo>>();
    List<AlarmAnalysisInfo> statistics = new ArrayList<AlarmAnalysisInfo>();
    AlarmAnalysisResult alarmResult = mock(AlarmAnalysisResult.class);
    when(alarmResult.getScheduledAlarms()).thenReturn(scheduledAlarms);
    when(alarmResult.getStatistics()).thenReturn(statistics);
    when(alarmanalysisinfoparser.parse(any(String.class), any(String.class), any(String.class), any(double.class), any(double.class), any(Date.class))).thenReturn(alarmResult);
    // Crypto
    traceDataReaderImpl.setCrypto(crypto);
    when(crypto.readSSLKeys(any(String.class))).thenReturn(1);
    Mockito.doAnswer(new Answer<Object>() {

        public Object answer(InvocationOnMock invocation) {
            byte b = 3;
            short s = 1;
            InetAddress address1 = null;
            InetAddress address2 = null;
            try {
                address2 = InetAddress.getByName("78.46.84.177");
                address1 = InetAddress.getByName("78.46.84.171");
            } catch (UnknownHostException e) {
                e.printStackTrace();
            }
            Date date1 = new Date();
            IPPacket ippack01 = mock(IPPacket.class);
            when(ippack01.getIPVersion()).thenReturn(b);
            when(ippack01.getFragmentOffset()).thenReturn(s);
            when(ippack01.getSourceIPAddress()).thenReturn(address1);
            when(ippack01.getDestinationIPAddress()).thenReturn(address2);
            when(ippack01.getTimeStamp()).thenReturn((double) date1.getTime());
            // pretend jpcap lib
            traceDataReaderImpl.packetArrived("com.google.android.youtube", ippack01);
            return null;
        }
    }).when(packetreader).readPacket(any(String.class), any(IPacketListener.class));
    AppInfo app = new AppInfo();
    Map<String, String> appMap = new HashMap<String, String>();
    appMap.put("flipboard.app", "2.3.8");
    appMap.put("com.att.android.arodatacollector", "3.1.1.6");
    appMap.put("com.google.android.youtube", "4.0.23");
    app.setAppVersionMap(appMap);
    List<String> appInfos = new ArrayList<String>();
    appInfos.add("flipboard.app");
    appInfos.add("com.att.android.arodatacollector");
    appInfos.add("com.google.android.youtube");
    app.setAppInfos(appInfos);
    when(appinforeader.readData(any(String.class))).thenReturn(app);
    DeviceDetail device = new DeviceDetail();
    device.setTotalLines(8);
    device.setScreenSize("720*1280");
    when(devicedetailreader.readData(any(String.class))).thenReturn(device);
    CollectOptions cOptions = new CollectOptions();
    when(collectOptionsReader.readData(any(String.class))).thenReturn(cOptions);
    NetworkTypeObject obj = new NetworkTypeObject();
    when(networktypereader.readData(any(String.class), any(double.class), any(double.class))).thenReturn(obj);
    TraceDirectoryResult result = traceDataReaderImpl.readTraceDirectory(Util.getCurrentRunningDir());
    assertSame(3, result.getAppIds().size());
}

Example 14 with IPPacket

use of com.att.aro.core.packetreader.pojo.IPPacket in project VideoOptimzer by attdevsupport.

the class TraceDataReaderImplTest method readTraceDir_checkExternalVideoAndTime.

@Test
public void readTraceDir_checkExternalVideoAndTime() throws IOException {
    String[] time = { "1410212153.578 1410213352.550", "272927100", "1410213352.550" };
    String[] appId = {};
    traceDataReaderImpl.setFileReader(filereader);
    when(filereader.directoryExist(any(String.class))).thenReturn(true);
    when(filereader.readAllLine(any(String.class))).thenReturn(time);
    when(filereader.readAllLine(any(String.class))).thenReturn(appId);
    when(filereader.readAllLine(Util.getCurrentRunningDir() + Util.FILE_SEPARATOR + "exVideo_time")).thenReturn(time);
    when(filereader.fileExist(any(String.class))).thenReturn(true);
    when(filereader.fileExist(Util.getCurrentRunningDir() + Util.FILE_SEPARATOR + "traffic1.cap")).thenReturn(false);
    when(filereader.fileExist(Util.getCurrentRunningDir() + Util.FILE_SEPARATOR + "time")).thenReturn(false);
    Map<String, List<ScheduledAlarmInfo>> scheduledAlarms = new HashMap<String, List<ScheduledAlarmInfo>>();
    List<AlarmAnalysisInfo> statistics = new ArrayList<AlarmAnalysisInfo>();
    AlarmAnalysisResult alarmResult = mock(AlarmAnalysisResult.class);
    when(alarmResult.getScheduledAlarms()).thenReturn(scheduledAlarms);
    when(alarmResult.getStatistics()).thenReturn(statistics);
    when(alarmanalysisinfoparser.parse(any(String.class), any(String.class), any(String.class), any(double.class), any(double.class), any(Date.class))).thenReturn(alarmResult);
    Mockito.doAnswer(new Answer<Object>() {

        public Object answer(InvocationOnMock invocation) {
            byte b = 3;
            short s = 1;
            InetAddress address1 = null;
            InetAddress address2 = null;
            try {
                address2 = InetAddress.getByName("78.46.84.177");
                address1 = InetAddress.getByName("78.46.84.171");
            } catch (UnknownHostException e) {
                e.printStackTrace();
            }
            Date date1 = new Date();
            IPPacket ippack01 = mock(IPPacket.class);
            when(ippack01.getIPVersion()).thenReturn(b);
            when(ippack01.getFragmentOffset()).thenReturn(s);
            when(ippack01.getSourceIPAddress()).thenReturn(address1);
            when(ippack01.getDestinationIPAddress()).thenReturn(address2);
            when(ippack01.getTimeStamp()).thenReturn((double) date1.getTime());
            // pretend jpcap lib
            traceDataReaderImpl.packetArrived("com.google.android.youtube", ippack01);
            return null;
        }
    }).when(packetreader).readPacket(any(String.class), any(IPacketListener.class));
    AppInfo app = new AppInfo();
    Map<String, String> appMap = new HashMap<String, String>();
    appMap.put("flipboard.app", "2.3.8");
    appMap.put("com.att.android.arodatacollector", "3.1.1.6");
    appMap.put("com.google.android.youtube", "4.0.23");
    app.setAppVersionMap(appMap);
    List<String> appInfos = new ArrayList<String>();
    appInfos.add("flipboard.app");
    appInfos.add("com.att.android.arodatacollector");
    appInfos.add("com.google.android.youtube");
    app.setAppInfos(appInfos);
    when(appinforeader.readData(any(String.class))).thenReturn(app);
    DeviceDetail device = new DeviceDetail();
    when(devicedetailreader.readData(any(String.class))).thenReturn(device);
    NetworkTypeObject obj = new NetworkTypeObject();
    when(networktypereader.readData(any(String.class), any(double.class), any(double.class))).thenReturn(obj);
    TraceDirectoryResult result = traceDataReaderImpl.readTraceDirectory(Util.getCurrentRunningDir());
    assertSame(0, result.getAppIds().size());
}