NovaBryan/EVE_Dump
Log | Files | 📦 下载最新源码 (.zip) | Refs | README

test_shouldDropPacket.py (1830B)

      1 #Embedded file name: C:\BuildAgent\work\20d73f1b24eed06e\eve\release\V24.01\carbon\common\script\net\test\test_shouldDropPacket.py
      2 from unittest import TestCase
      3 import mock
      4 from carbon.common.lib.const import ADDRESS_TYPE_CLIENT, ADDRESS_TYPE_BROADCAST, ADDRESS_TYPE_NODE
      5 from carbon.common.script.net.packetthreshold import PacketThreshold
      6 
      7 class Message(object):
      8 
      9     def __init__(self, destination = 'client', destinationIdType = ''):
     10         self.destination = mock.Mock(addressType={'client': ADDRESS_TYPE_CLIENT,
     11          'broadcast': ADDRESS_TYPE_BROADCAST,
     12          'server': ADDRESS_TYPE_NODE}[destination], idtype=destinationIdType)
     13 
     14 
     15 class TestShouldDropPacket(TestCase):
     16 
     17     def test_cases(self):
     18         self.assertTrue(self.ShouldDropPacket('123', 'tcp:packet:machoNet', destination='client', source='server'))
     19         self.assertTrue(self.ShouldDropPacket('123', 'tcp:packet:client', destination='client', source='proxy'))
     20         self.assertTrue(self.ShouldDropPacket('123', 'tcp:packet:machoNet', destination='broadcast', source='server'))
     21         self.assertFalse(self.ShouldDropPacket('123', 'tcp:packet:machoNet', destination='broadcast', source='proxy'))
     22         self.assertFalse(self.ShouldDropPacket('123', 'tcp:packet:machoNet', destination='broadcast', source='server', idType='nodeID'))
     23         self.assertTrue(self.ShouldDropPacket('123', 'ip:packet:server', destination='server', source='client'))
     24         self.assertFalse(self.ShouldDropPacket('1', 'tcp:packet:machoNet', destination='client', source='server'))
     25 
     26     def ShouldDropPacket(self, payload, transportName, destination, source, idType = ''):
     27         dropSize = 2
     28         message = Message(destination=destination, destinationIdType=idType)
     29         return PacketThreshold(transportName, dropSize, source).ShouldDropPacket(payload, message)