1 files changed, 13 insertions, 16 deletions

diff --git a/yardstick/tests/unit/network_services/traffic_profile/test_rfc2544.py b/yardstick/tests/unit/network_services/traffic_profile/test_rfc2544.py
index 6d3558c7e..febcfe5da 100644
--- a/yardstick/tests/unit/network_services/traffic_profile/test_rfc2544.py
+++ b/yardstick/tests/unit/network_services/traffic_profile/test_rfc2544.py
@@ -255,31 +255,23 @@ class TestRFC2544Profile(base.BaseUnitTestCase):
         mock_get_framesize.return_value = '64B'
 
         samples = [
-            {'xe1': {'tx_throughput_fps': 110,
-                     'rx_throughput_fps': 101,
-                     'out_packets': 2100,
+            {'xe1': {'out_packets': 2100,
                      'in_packets': 2010,
                      'out_bytes': 134400,
                      'in_bytes': 128640,
                      'timestamp': datetime.datetime(2000, 1, 1, 1, 1, 1, 1)},
-             'xe2': {'tx_throughput_fps': 210,
-                     'rx_throughput_fps': 201,
-                     'out_packets': 4100,
+             'xe2': {'out_packets': 4100,
                      'in_packets': 4010,
                      'out_bytes': 262400,
                      'in_bytes': 256640,
                      'timestamp': datetime.datetime(2000, 1, 1, 1, 1, 1, 1)}},
-            {'xe1': {'tx_throughput_fps': 156,
-                     'rx_throughput_fps': 108,
-                     'out_packets': 2110,
+            {'xe1': {'out_packets': 2110,
                      'in_packets': 2040,
                      'out_bytes': 135040,
                      'in_bytes': 130560,
                      'latency': 'Latency1',
                      'timestamp': datetime.datetime(2000, 1, 1, 1, 1, 1, 31)},
-             'xe2': {'tx_throughput_fps': 253,
-                     'rx_throughput_fps': 215,
-                     'out_packets': 4150,
+             'xe2': {'out_packets': 4150,
                      'in_packets': 4010,
                      'out_bytes': 265600,
                      'in_bytes': 256640,
@@ -288,15 +280,20 @@ class TestRFC2544Profile(base.BaseUnitTestCase):
         ]
         completed, output = rfc2544_profile.get_drop_percentage(
             samples, 0, 0, False, 0.1)
-        expected = {'DropPercentage': 50.0,
-                    'Latency': {'xe1': 'Latency1', 'xe2': 'Latency2'},
+        expected = {'xe1': {'OutPackets': 10,
+                            'InPackets': 30,
+                            'OutBytes': 640,
+                            'InBytes': 1920},
+                    'xe2': {'OutPackets': 50,
+                            'InPackets': 0,
+                            'OutBytes': 3200,
+                            'InBytes': 0},
+                    'DropPercentage': 50.0,
                     'RxThroughput': 1000000.0,
                     'TxThroughput': 2000000.0,
                     'RxThroughputBps': 64000000.0,
                     'TxThroughputBps': 128000000.0,
-                    'CurrentDropPercentage': 50.0,
                     'Rate': 100.0,
-                    'Throughput': 1000000.0,
                     'Iteration': 1,
                     'PktSize': '64B',
                     'Status': 'Failure'}