@@ -132,6 +132,41 @@ def test_delete_node(self):
132132 m .call_url , "/nodebalancers/123456/configs/65432/nodes/54321"
133133 )
134134
135+
136+ class NodeBalancerTest (ClientBaseCase ):
137+ def test_update (self ):
138+ """
139+ Test that you can update a NodeBalancer.
140+ """
141+ nb = NodeBalancer (self .client , 123456 )
142+ nb .label = "updated-label"
143+ nb .client_conn_throttle = 7
144+ nb .tags = ["foo" , "bar" ]
145+
146+ with self .mock_put ("nodebalancers/123456" ) as m :
147+ nb .save ()
148+ self .assertEqual (m .call_url , "/nodebalancers/123456" )
149+ self .assertEqual (
150+ m .call_data ,
151+ {
152+ "label" : "updated-label" ,
153+ "client_conn_throttle" : 7 ,
154+ "tags" : ["foo" , "bar" ],
155+ },
156+ )
157+
158+ def test_firewalls (self ):
159+ """
160+ Test that you can get the firewalls for the requested NodeBalancer.
161+ """
162+ nb = NodeBalancer (self .client , 12345 )
163+ firewalls_url = "/nodebalancers/12345/firewalls"
164+
165+ with self .mock_get (firewalls_url ) as m :
166+ result = nb .firewalls ()
167+ self .assertEqual (m .call_url , firewalls_url )
168+ self .assertEqual (len (result ), 1 )
169+
135170 def test_config_rebuild (self ):
136171 """
137172 Test that you can rebuild the cofig of a node balancer.
@@ -193,15 +228,3 @@ def test_statistics(self):
193228 "linode.com - balancer12345 (12345) - day (5 min avg)" ,
194229 )
195230 self .assertEqual (m .call_url , statistics_url )
196-
197- def test_firewalls (self ):
198- """
199- Test that you can get the firewalls for the requested NodeBalancer.
200- """
201- nb = NodeBalancer (self .client , 12345 )
202- firewalls_url = "/nodebalancers/12345/firewalls"
203-
204- with self .mock_get (firewalls_url ) as m :
205- result = nb .firewalls ()
206- self .assertEqual (m .call_url , firewalls_url )
207- self .assertEqual (len (result ), 1 )
0 commit comments