3wordchant/capsul-flask
1
0
Fork 1
Code Issues Pull Requests 1 Releases Wiki Activity

trying to get tests to pass with hub_model=capsulflask

Browse Source
This commit is contained in:
forest
2021-07-27 12:48:48 -05:00
committed by 3wc
parent 5e682cc705
commit 2348191990
6 changed files with 24 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
capsulflask/tests/test_console.py
View File

@ -6,7 +6,7 @@ from flask import url_for
from capsulflask.db import get_model
from capsulflask.tests_base import BaseTestCase
from capsulflask.hub_model import MockHub
from capsulflask.spoke_model import MockSpoke
class ConsoleTests(BaseTestCase):
@ -65,9 +65,11 @@ class ConsoleTests(BaseTestCase):
get_model().create_payment_session('fake', 'test', 'test@example.com', 20)
get_model().consume_payment_session('fake', 'test', 20)
with patch.object(MockHub, 'capacity_avaliable', return_value=False) as mock_method:
with patch.object(MockSpoke, 'capacity_avaliable', return_value=False) as mock_method:
response = client.post(url_for("console.create"), data=data)
self.assert200(response)
mock_method.assert_called()
capacity_message = \
@ -112,6 +114,14 @@ class ConsoleTests(BaseTestCase):
response = client.post(url_for("console.create"), data=data)
self.assertEqual(
len(get_model().list_all_operations()),
1
)
vms = get_model().list_vms_for_account('test@example.com')
self.assertEqual(
len(vms),
@ -125,6 +135,7 @@ class ConsoleTests(BaseTestCase):
url_for("console.index") + f'?created={vm_id}'
)
def test_keys_loads(self):
self._login('test@example.com')
with self.client as client: