Skip to content

Commit f68d0b3

Browse files
committed
Improve loadbalancer test
* Update the save loadbalancer test to actually test some boolean config values
1 parent bb6be45 commit f68d0b3

2 files changed

Lines changed: 9 additions & 6 deletions

File tree

digitalocean/tests/data/loadbalancer/save.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,9 @@
7171
34153248,
7272
34153250
7373
],
74-
"redirect_http_to_https": false,
75-
"enable_proxy_protocol": false,
76-
"enable_backend_keepalive": false,
74+
"redirect_http_to_https": true,
75+
"enable_proxy_protocol": true,
76+
"enable_backend_keepalive": true,
7777
"vpc_uuid": "c33931f2-a26a-4e61-b85c-4e95a2ec431b"
7878
}
7979
}

digitalocean/tests/test_load_balancer.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,9 @@ def test_save(self):
256256
self.lb.sticky_sessions.cookie_ttl_seconds = 300
257257
self.lb.droplet_ids = [34153248, 34153250]
258258
self.lb.vpc_uuid = self.vpc_uuid
259+
self.lb.redirect_http_to_https = True
260+
self.lb.enable_proxy_protocol = True
261+
self.lb.enable_backend_keepalive = True
259262
res = self.lb.save()
260263

261264
lb = digitalocean.LoadBalancer(**res['load_balancer'])
@@ -292,9 +295,9 @@ def test_save(self):
292295
self.assertEqual(lb.sticky_sessions.cookie_ttl_seconds, 300)
293296
self.assertEqual(lb.droplet_ids, [34153248, 34153250])
294297
self.assertEqual(lb.tag, '')
295-
self.assertEqual(lb.redirect_http_to_https, False)
296-
self.assertEqual(lb.enable_proxy_protocol, False)
297-
self.assertEqual(lb.enable_backend_keepalive, False)
298+
self.assertEqual(lb.redirect_http_to_https, True)
299+
self.assertEqual(lb.enable_proxy_protocol, True)
300+
self.assertEqual(lb.enable_backend_keepalive, True)
298301
self.assertEqual(self.lb.vpc_uuid, self.vpc_uuid)
299302

300303
@responses.activate

0 commit comments

Comments
 (0)