|
2 | 2 | from test.unit.base import ClientBaseCase |
3 | 3 |
|
4 | 4 | from linode_api4 import LongviewSubscription |
| 5 | +from linode_api4.objects.beta import BetaProgram |
5 | 6 | from linode_api4.objects.linode import Instance |
6 | 7 | from linode_api4.objects.networking import IPAddress |
7 | 8 | from linode_api4.objects.object_storage import ( |
@@ -411,6 +412,48 @@ def test_payments(self): |
411 | 412 | self.assertEqual(payment.date, datetime(2015, 1, 1, 5, 1, 2)) |
412 | 413 | self.assertEqual(payment.usd, 1000) |
413 | 414 |
|
| 415 | + def test_enrolled_betas(self): |
| 416 | + """ |
| 417 | + Tests that enrolled beta programs can be retrieved |
| 418 | + """ |
| 419 | + enrolled_betas = self.client.account.enrolled_betas() |
| 420 | + |
| 421 | + self.assertEqual(len(enrolled_betas), 1) |
| 422 | + beta = enrolled_betas[0] |
| 423 | + |
| 424 | + self.assertEqual(beta.id, "cool") |
| 425 | + self.assertEqual(beta.enrolled, datetime(2018, 1, 2, 3, 4, 5)) |
| 426 | + self.assertEqual(beta.started, datetime(2018, 1, 2, 3, 4, 5)) |
| 427 | + self.assertEqual(beta.ended, datetime(2018, 1, 2, 3, 4, 5)) |
| 428 | + |
| 429 | + def test_join_beta_program(self): |
| 430 | + """ |
| 431 | + Tests that user can join a beta program |
| 432 | + """ |
| 433 | + join_beta_url = "/account/betas" |
| 434 | + with self.mock_post({}) as m: |
| 435 | + self.client.account.join_beta_program("cool_beta") |
| 436 | + self.assertEqual( |
| 437 | + m.call_data, |
| 438 | + { |
| 439 | + "id": "cool_beta", |
| 440 | + }, |
| 441 | + ) |
| 442 | + self.assertEqual(m.call_url, join_beta_url) |
| 443 | + |
| 444 | + # Test that user can join a beta program with an BetaProgram object |
| 445 | + with self.mock_post({}) as m: |
| 446 | + self.client.account.join_beta_program( |
| 447 | + BetaProgram(self.client, "cool_beta") |
| 448 | + ) |
| 449 | + self.assertEqual( |
| 450 | + m.call_data, |
| 451 | + { |
| 452 | + "id": "cool_beta", |
| 453 | + }, |
| 454 | + ) |
| 455 | + self.assertEqual(m.call_url, join_beta_url) |
| 456 | + |
414 | 457 |
|
415 | 458 | class BetaProgramGroupTest(ClientBaseCase): |
416 | 459 | """ |
|
0 commit comments