[dts] [PATCH V1 3/5] tests/TestSuite_ddp_gtp_qregion.py: load the local dep module in scapy command

Xinfeng Zhao xinfengx.zhao at intel.com
Wed Sep 18 01:19:23 CEST 2019


Signed-off-by: Xinfeng Zhao <xinfengx.zhao at intel.com>
---
 tests/TestSuite_ddp_gtp_qregion.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/TestSuite_ddp_gtp_qregion.py b/tests/TestSuite_ddp_gtp_qregion.py
index 4963f5f..c79082a 100644
--- a/tests/TestSuite_ddp_gtp_qregion.py
+++ b/tests/TestSuite_ddp_gtp_qregion.py
@@ -2,12 +2,18 @@
 
 import time
 import re
+import os
 import sys
 import utils
-from scapy.all import *
+from scapy.all import Ether, IP, UDP, IPv6, IPv6
 from test_case import TestCase
 from pmd_output import PmdOutput
-from settings import get_nic_name
+from settings import get_nic_name, FOLDERS
+import random
+from gtp import *
+
+CWD = os.getcwd()
+DIR_DEPEND = os.path.join(CWD, FOLDERS['Depends'])
 
 
 class TestDdpGtpQregion(TestCase):
@@ -148,6 +154,8 @@ class TestDdpGtpQregion(TestCase):
         """
         pkts = self.gtp_pkts(flowtype, keyword, opt)
         for packet_type in pkts.keys():
+            self.tester.scapy_append("sys.path.append('%s')" % DIR_DEPEND)
+            self.tester.scapy_append("from gtp import *")
             self.tester.scapy_append(
                 'sendp([%s], iface="%s")'
                 % (pkts[packet_type], self.tester_intf))
@@ -210,6 +218,8 @@ class TestDdpGtpQregion(TestCase):
                     keyword = 'src_ipv6'
             pkts = self.gtp_pkts(flowtype, keyword, opt)
             for packet_type in pkts.keys():
+                self.tester.scapy_append("sys.path.append('%s')" % DIR_DEPEND)
+                self.tester.scapy_append("from gtp import *")
                 self.tester.scapy_append(
                     'sendp([%s], iface="%s")'
                     % (pkts[packet_type], self.tester_intf))
-- 
2.17.1



More information about the dts mailing list