Skip to content

Commit ac50e04

Browse files
committed
Modified the test file to remove click commands and do the REDIS-DB updates for MCLAG config commands.
1 parent 4e53afc commit ac50e04

File tree

1 file changed

+42
-23
lines changed

1 file changed

+42
-23
lines changed

tests/test_mclag_cfg.py

+42-23
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,37 @@ def check_table_doesnt_exists(db, table, key):
3535
return True, error_info
3636

3737

38-
38+
def create_mclag_domain(dvs, domain_id, source_ip, peer_ip, peer_link):
39+
tbl = swsscommon.Table(dvs.cdb, "MCLAG_DOMAIN")
40+
fvs = swsscommon.FieldValuePairs([("source_ip", source_ip),
41+
("peer_ip", peer_ip),
42+
("peer_link", peer_link)])
43+
tbl.set(domain_id, fvs)
44+
time.sleep(1)
45+
46+
def remove_mclag_domain(dvs, domain_id):
47+
tbl = swsscommon.Table(dvs.cdb, "MCLAG_DOMAIN")
48+
tbl._del(domain_id)
49+
time.sleep(1)
50+
51+
def add_mclag_domain_field(dvs, domain_id, field, value):
52+
tbl = swsscommon.Table(dvs.cdb, "MCLAG_DOMAIN")
53+
fvs = swsscommon.FieldValuePairs([(field, value)])
54+
tbl.set(domain_id, fvs)
55+
time.sleep(1)
56+
57+
def create_mclag_interface(dvs, domain_id, mclag_interface):
58+
tbl = swsscommon.Table(dvs.cdb, "MCLAG_INTERFACE")
59+
fvs = swsscommon.FieldValuePairs([("if_type", "PortChannel")])
60+
key_string = domain_id + "|" + mclag_interface
61+
tbl.set(key_string, fvs)
62+
time.sleep(1)
63+
64+
def remove_mclag_interface(dvs, domain_id, mclag_interface):
65+
tbl = swsscommon.Table(dvs.cdb, "MCLAG_INTERFACE")
66+
key_string = domain_id + "|" + mclag_interface
67+
tbl._del(key_string)
68+
time.sleep(1)
3969

4070
# Test MCLAG Configs
4171
class TestMclagConfig(object):
@@ -72,8 +102,7 @@ def test_mclag_cfg_domain_add(self, dvs, testlog):
72102
delete_table_keys(self.cfg_db, self.CFG_MCLAG_DOMAIN_TABLE)
73103
delete_table_keys(self.cfg_db, self.CFG_MCLAG_INTERFACE_TABLE)
74104

75-
cmd_string ="config mclag add {} {} {} {}".format(self.MCLAG_DOMAIN_ID, self.MCLAG_SRC_IP, self.MCLAG_PEER_IP, self.MCLAG_PEER_LINK)
76-
dvs.runcmd(cmd_string)
105+
create_mclag_domain(dvs, self.MCLAG_DOMAIN_ID, self.MCLAG_SRC_IP, self.MCLAG_PEER_IP, self.MCLAG_PEER_LINK)
77106
time.sleep(2)
78107

79108
#check whether domain cfg table contents are same as configured values
@@ -90,8 +119,7 @@ def test_mclag_cfg_domain_add(self, dvs, testlog):
90119
@pytest.mark.dev_sanity
91120
def test_mclag_cfg_domain_add_2nd(self, dvs, testlog):
92121
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
93-
cmd_string ="config mclag add {} {} {} {}".format(self.MCLAG_DOMAIN_2, self.MCLAG_SRC_IP, self.MCLAG_PEER_IP, self.MCLAG_PEER_LINK)
94-
dvs.runcmd(cmd_string)
122+
create_mclag_domain(dvs, self.MCLAG_DOMAIN_2, self.MCLAG_SRC_IP, self.MCLAG_PEER_IP, self.MCLAG_PEER_LINK)
95123
time.sleep(2)
96124

97125
#check whether second domain config is not added to config db
@@ -105,8 +133,7 @@ def test_mclag_cfg_domain_add_2nd(self, dvs, testlog):
105133
@pytest.mark.dev_sanity
106134
def test_mclag_cfg_intf_add(self, dvs, testlog):
107135
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
108-
cmd_string ="config mclag member add {} {}".format(self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE1)
109-
dvs.runcmd(cmd_string)
136+
create_mclag_interface(dvs, self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE1)
110137
time.sleep(2)
111138

112139
#check whether mclag interface config is reflected
@@ -118,9 +145,7 @@ def test_mclag_cfg_intf_add(self, dvs, testlog):
118145
@pytest.mark.dev_sanity
119146
def test_mclag_cfg_intf_remove_and_add(self, dvs, testlog):
120147
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
121-
122-
cmd_string ="config mclag member del {} {}".format(self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE1)
123-
dvs.runcmd(cmd_string)
148+
remove_mclag_interface(dvs, self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE1)
124149
time.sleep(2)
125150

126151
#check whether mclag interface is removed
@@ -129,8 +154,7 @@ def test_mclag_cfg_intf_remove_and_add(self, dvs, testlog):
129154
assert ok,error_info
130155

131156
#add different mclag interface
132-
cmd_string ="config mclag member del {} {}".format(self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE2)
133-
dvs.runcmd(cmd_string)
157+
create_mclag_interface(dvs, self.MCLAG_DOMAIN_ID, self.MCLAG_INTERFACE2)
134158
time.sleep(2)
135159

136160
#check whether new mclag interface is added
@@ -144,8 +168,8 @@ def test_mclag_cfg_session_timeout_valid_values(self, dvs, testlog):
144168
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
145169

146170
for value in self.MCLAG_SESS_TMOUT_VALID_LIST:
147-
cmd_string ="config mclag session-timeout {} {}".format(self.MCLAG_DOMAIN_ID, value)
148-
dvs.runcmd(cmd_string)
171+
add_mclag_domain_field(dvs, self.MCLAG_DOMAIN_ID, "session_timeout", value)
172+
149173
time.sleep(2)
150174

151175
#check whether domain cfg table contents are same as configured values
@@ -165,10 +189,8 @@ def test_mclag_cfg_ka_valid_values(self, dvs, testlog):
165189
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
166190

167191
for value in self.MCLAG_KA_VALID_LIST:
168-
cmd_string ="config mclag keepalive-interval {} {}".format(self.MCLAG_DOMAIN_ID, value)
169-
dvs.runcmd(cmd_string)
192+
add_mclag_domain_field(dvs, self.MCLAG_DOMAIN_ID, "keepalive_interval", value)
170193
time.sleep(2)
171-
172194
#check whether domain cfg table contents are same as configured values
173195
ok,error_info = dvs.all_table_entry_has(self.cfg_db, self.CFG_MCLAG_DOMAIN_TABLE, self.MCLAG_DOMAIN_ID,
174196
[
@@ -188,8 +210,7 @@ def test_mclag_cfg_ka_invalid_values(self, dvs, testlog):
188210
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
189211

190212
for value in self.MCLAG_KA_INVALID_LIST:
191-
cmd_string ="config mclag keepalive-interval {} {}".format(self.MCLAG_DOMAIN_ID, value)
192-
dvs.runcmd(cmd_string)
213+
add_mclag_domain_field(dvs, self.MCLAG_DOMAIN_ID, "keepalive_interval", value)
193214
time.sleep(2)
194215

195216
#check whether domain cfg table contents are same as configured values
@@ -206,8 +227,7 @@ def test_mclag_cfg_session_invalid_values(self, dvs, testlog):
206227
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
207228

208229
for value in self.MCLAG_SESS_TMOUT_INVALID_LIST:
209-
cmd_string ="config mclag session-timeout {} {}".format(self.MCLAG_DOMAIN_ID, value)
210-
dvs.runcmd(cmd_string)
230+
add_mclag_domain_field(dvs, self.MCLAG_DOMAIN_ID, "session_timeout", value)
211231
time.sleep(2)
212232

213233
#check whether domain cfg table contents are same as configured values
@@ -223,8 +243,7 @@ def test_mclag_cfg_session_invalid_values(self, dvs, testlog):
223243
def test_mclag_cfg_domain_del(self, dvs, testlog):
224244
self.cfg_db = swsscommon.DBConnector(swsscommon.CONFIG_DB, dvs.redis_sock, 0)
225245

226-
cmd_string ="config mclag del {}".format(self.MCLAG_DOMAIN_ID)
227-
dvs.runcmd(cmd_string)
246+
remove_mclag_domain(dvs, self.MCLAG_DOMAIN_ID)
228247
time.sleep(2)
229248

230249
#check whether domain cfg table contents are same as configured values

0 commit comments

Comments
 (0)