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

Search in sources :

Example 11 with DomainNameSystem

use of com.att.aro.core.packetreader.pojo.DomainNameSystem 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());
}
Also used : DomainNameSystem(com.att.aro.core.packetreader.pojo.DomainNameSystem) ArrayList(java.util.ArrayList) UDPPacket(com.att.aro.core.packetreader.pojo.UDPPacket) Date(java.util.Date) TCPPacket(com.att.aro.core.packetreader.pojo.TCPPacket) PacketInfo(com.att.aro.core.packetanalysis.pojo.PacketInfo) InetAddress(java.net.InetAddress) IPPacket(com.att.aro.core.packetreader.pojo.IPPacket) HashSet(java.util.HashSet) Session(com.att.aro.core.packetanalysis.pojo.Session) Test(org.junit.Test) BaseTest(com.att.aro.core.BaseTest)

Example 12 with DomainNameSystem

use of com.att.aro.core.packetreader.pojo.DomainNameSystem 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());
}
Also used : DomainNameSystem(com.att.aro.core.packetreader.pojo.DomainNameSystem) UnknownHostException(java.net.UnknownHostException) ArrayList(java.util.ArrayList) UDPPacket(com.att.aro.core.packetreader.pojo.UDPPacket) Date(java.util.Date) TCPPacket(com.att.aro.core.packetreader.pojo.TCPPacket) PacketInfo(com.att.aro.core.packetanalysis.pojo.PacketInfo) InetAddress(java.net.InetAddress) IPPacket(com.att.aro.core.packetreader.pojo.IPPacket) HashSet(java.util.HashSet) Session(com.att.aro.core.packetanalysis.pojo.Session) Test(org.junit.Test) BaseTest(com.att.aro.core.BaseTest)

Aggregations

DomainNameSystem (com.att.aro.core.packetreader.pojo.DomainNameSystem)12 UDPPacket (com.att.aro.core.packetreader.pojo.UDPPacket)10 InetAddress (java.net.InetAddress)10 BaseTest (com.att.aro.core.BaseTest)8 PacketInfo (com.att.aro.core.packetanalysis.pojo.PacketInfo)8 TCPPacket (com.att.aro.core.packetreader.pojo.TCPPacket)8 ArrayList (java.util.ArrayList)8 HashSet (java.util.HashSet)8 Test (org.junit.Test)8 Session (com.att.aro.core.packetanalysis.pojo.Session)7 BurstCollectionAnalysisData (com.att.aro.core.packetanalysis.pojo.BurstCollectionAnalysisData)4 RrcStateRange (com.att.aro.core.packetanalysis.pojo.RrcStateRange)4 CpuActivity (com.att.aro.core.peripheral.pojo.CpuActivity)4 HashMap (java.util.HashMap)4 UserEvent (com.att.aro.core.peripheral.pojo.UserEvent)3 ProfileLTE (com.att.aro.core.configuration.pojo.ProfileLTE)2 IPPacket (com.att.aro.core.packetreader.pojo.IPPacket)2 Inet4Address (java.net.Inet4Address)2 UnknownHostException (java.net.UnknownHostException)2 Date (java.util.Date)2
About Me
UseOf

Java Tutorials :

Simple Java RabbitMQ Example with Spring
Simple Springboot JPA Eclipeslink Example
Simple SpringBoot JPA Hibernate Example
Jackson JSON @JsonIgnore and @JsonIgnoreProperties Examples
Java Infinite recursion (StackOverflowError) Jackson solutions
Simple Java Jackson Serialize Objects to JSON and convert them back To Object
ElasticSearch 5 - Upload files using Java API in binary field Example
Java JAXB marshall unmarshall Examples from String and Stream
Java 8 forEach List and Map examples
ElasticSearch 5 Java API SearchRequestBuilder examples
Java ElasticSearch 5 examples with node index and mapping - running local
Convert String to int or Integer Java 8
Java 9 in action - JShell - Ubuntu
Java - removing invalid characters from a file name
Hello world!? or Hello Tutorial