add additional detail to e2e test

open-traffic-generator · Jan 8, 2021 · dfdd89d · dfdd89d
1 parent 5a4dd23
commit dfdd89d
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/snappi/tests/conftest.py b/snappi/tests/conftest.py
@@ -93,6 +93,12 @@ def get_config():
                     status=200)
 
 
+@app.route('/control/transmit', methods=['POST'])
+def set_transmit_state():
+    global CONFIG
+    return Response(status=200)
+
+
 @app.route('/results/port', methods=['POST'])
 def get_port_metrics():
     import snappi

diff --git a/snappi/tests/test_e2e_port_flow_config.py b/snappi/tests/test_e2e_port_flow_config.py
@@ -44,6 +44,23 @@ def test_e2e_port_flow_config():
     api.set_config(config)
     print(api.get_config())
 
+    # start transmit
+    transmit_state = api.transmit_state()
+    transmit_state.state = 'start'
+    api.set_transmit_state(transmit_state)
+
+    # get port metrics
+    port_metrics_request = api.port_metrics_request()
+    port_metrics = api.get_port_metrics(port_metrics_request)
+    for metric in port_metrics:
+        print(metric)
+
+    # get flow metrics
+    flow_metrics_request = api.flow_metrics_request()
+    flow_metrics = api.get_flow_metrics(flow_metrics_request)
+    for metric in flow_metrics:
+        print(metric)
+
 
 if __name__ == '__main__':
     pytest.main(['-vv', '-s', __file__])