From 31a4ef5afc56c5d6ff6870da71a954e4bcbee057 Mon Sep 17 00:00:00 2001 From: Ido Barnea Date: Tue, 22 Nov 2016 18:03:52 +0200 Subject: client config gtest Signed-off-by: Ido Barnea --- src/trex_client_config.h | 30 +++++++++++++++++++++++------- 1 file changed, 23 insertions(+), 7 deletions(-) (limited to 'src/trex_client_config.h') diff --git a/src/trex_client_config.h b/src/trex_client_config.h index dbdfd12d..257d354f 100644 --- a/src/trex_client_config.h +++ b/src/trex_client_config.h @@ -190,9 +190,9 @@ class ClientCfgBase { public: virtual void dump (FILE *fd) const { - fprintf(fd, "initiator:\n"); + fprintf(fd, " initiator :\n"); m_initiator.dump(fd); - fprintf(fd, "responder:\n"); + fprintf(fd, " responder :\n"); m_responder.dump(fd); } virtual void update(uint32_t index, const ClientCfgExt *cfg); @@ -205,9 +205,9 @@ public: class ClientCfgExt : public ClientCfgBase { public: virtual void dump (FILE *fd) const { - fprintf(fd, "initiator:\n"); + fprintf(fd, " initiator:\n"); m_initiator.dump(fd); - fprintf(fd, "responder:\n"); + fprintf(fd, " responder:\n"); m_responder.dump(fd); } @@ -254,7 +254,7 @@ class ClientCfgCompactEntry { * */ class ClientCfgEntry { - + friend class basic_client_cfg_test1_Test; public: ClientCfgEntry() { @@ -297,6 +297,14 @@ public: uint32_t m_count; private: + void set_params(uint32_t start, uint32_t end, uint32_t count) { // for tests + m_ip_start = start; + m_ip_end = end; + m_count = count; + } + void set_cfg(const ClientCfgExt &cfg) { + m_cfg = cfg; + } uint32_t m_iterator; }; @@ -305,7 +313,8 @@ private: * */ class ClientCfgDB { -public: + friend class basic_client_cfg_test1_Test; + public: ClientCfgDB() { m_is_empty = true; @@ -314,6 +323,10 @@ public: m_tg = NULL; } + ~ClientCfgDB() { + m_groups.clear(); + } + void dump(FILE *fd) ; /** @@ -349,7 +362,10 @@ public: private: void parse_single_group(YAMLParserWrapper &parser, const YAML::Node &node); void parse_dir(YAMLParserWrapper &parser, const YAML::Node &node, ClientCfgDirExt &dir); - + void set_vlan(bool val) {m_under_vlan = val;} // for tests + void add_group(uint32_t ip, ClientCfgEntry cfg) { // for tests + m_groups.insert(std::make_pair(ip, cfg)); + } /** * verify the YAML file loaded in valid * -- cgit 1.2.3-korg