@@ -90,7 +90,7 @@ def mock_wrapper(cls, test_case):
9090 attr .current_test = "" # Empty them to avoid side effects
9191 attr .current_test_case = "" # Empty them to avoid side effects
9292
93- return result
93+ return None
9494
9595 @functools .wraps (func )
9696 def real_wrapper (cls , test_case ):
@@ -249,7 +249,7 @@ def test_get_bgp_config(self, test_case):
249249 assert get_bgp_config == {} or len (get_bgp_config ) > 0
250250
251251 for bgp_group in get_bgp_config .values ():
252- assert helpers .test_model (models .BPGConfigGroupDict , bgp_group )
252+ assert helpers .test_model (models .BGPConfigGroupDict , bgp_group )
253253 for bgp_neighbor in bgp_group .get ("neighbors" , {}).values ():
254254 assert helpers .test_model (models .BGPConfigNeighborDict , bgp_neighbor )
255255
@@ -311,7 +311,9 @@ def test_get_ntp_peers(self, test_case):
311311
312312 for peer , peer_details in get_ntp_peers .items ():
313313 assert isinstance (peer , str )
314- assert helpers .test_model (models .NTPPeerDict , peer_details )
314+ assert helpers .test_model (
315+ models .NTPPeerDict , peer_details , allow_subset = True
316+ )
315317
316318 return get_ntp_peers
317319
@@ -323,7 +325,9 @@ def test_get_ntp_servers(self, test_case):
323325
324326 for server , server_details in get_ntp_servers .items ():
325327 assert isinstance (server , str )
326- assert helpers .test_model (models .NTPServerDict , server_details )
328+ assert helpers .test_model (
329+ models .NTPServerDict , server_details , allow_subset = True
330+ )
327331
328332 return get_ntp_servers
329333
0 commit comments