Got rid of get_vm_one_queue(). Separated its uses to many small attributes.

Signed-off-by: Ido Barnea <ibarnea@cisco.com>

1 files changed, 3 insertions, 1 deletions

diff --git a/src/pre_test.h b/src/pre_test.h
index 14b444cf..88616440 100644
--- a/src/pre_test.h
+++ b/src/pre_test.h
@@ -83,11 +83,12 @@ class CPretestOnePortInfo {
 
 class CPretest {
  public:
-    CPretest(uint16_t max_ports) {
+    CPretest(uint16_t max_ports, uint16_t queues_per_port) {
         m_max_ports = max_ports;
         for (int i =0; i < max_ports; i++) {
             m_port_info[i].set_port_id(i);
         }
+        m_num_q = queues_per_port;
     }
     void add_ip(uint16_t port, uint32_t ip, uint16_t vlan, MacAddress src_mac);
     void add_ip(uint16_t port, uint32_t ip, MacAddress src_mac);
@@ -115,6 +116,7 @@ class CPretest {
  private:
     CPretestOnePortInfo m_port_info[TREX_MAX_PORTS];
     uint16_t m_max_ports;
+    uint16_t m_num_q;
 };
 
 #endif