@@ -3958,6 +3958,7 @@ def test_get_outbound_type(self):
39583958 self .models ,
39593959 decorator_mode = DecoratorMode .CREATE ,
39603960 )
3961+ self .create_attach_agentpool_context (ctx1 )
39613962 outbound_type_1 = ctx1 ._get_outbound_type (False , False , None )
39623963 expect_outbound_type_1 = CONST_OUTBOUND_TYPE_MANAGED_NAT_GATEWAY
39633964 self .assertEqual (outbound_type_1 ,expect_outbound_type_1 )
@@ -3968,6 +3969,7 @@ def test_get_outbound_type(self):
39683969 self .models ,
39693970 decorator_mode = DecoratorMode .CREATE ,
39703971 )
3972+ self .create_attach_agentpool_context (ctx2 )
39713973 outbound_type_2 = ctx2 ._get_outbound_type (False , False , None )
39723974 expect_outbound_type_2 = CONST_OUTBOUND_TYPE_LOAD_BALANCER
39733975 self .assertEqual (outbound_type_2 ,expect_outbound_type_2 )
@@ -3978,6 +3980,7 @@ def test_get_outbound_type(self):
39783980 self .models ,
39793981 decorator_mode = DecoratorMode .CREATE ,
39803982 )
3983+ self .create_attach_agentpool_context (ctx3 )
39813984 outbound_type_3 = ctx3 ._get_outbound_type (False , False , None )
39823985 expect_outbound_type_3 = CONST_OUTBOUND_TYPE_LOAD_BALANCER
39833986 self .assertEqual (outbound_type_3 ,expect_outbound_type_3 )
0 commit comments