Add organization.update_company_name

Fix unit-tests
This commit is contained in:
allegroai
2022-02-13 19:29:46 +02:00
parent cae38a365b
commit 604a38035b
3 changed files with 63 additions and 37 deletions

View File

@@ -2,9 +2,6 @@ from apiserver.tests.automated import TestService
class TestOrganization(TestService):
def setUp(self, version="2.12"):
super().setUp(version=version)
def test_get_user_companies(self):
company = self.api.organization.get_user_companies().companies[0]
self.assertEqual(len(company.owners), company.allocated)

View File

@@ -2,7 +2,7 @@ import math
from apiserver.tests.automated import TestService
class TestEntityOrdering(TestService):
class TestPagingAndScrolling(TestService):
name_prefix = f"Test paging "
def setUp(self, **kwargs):
@@ -13,7 +13,16 @@ class TestEntityOrdering(TestService):
tasks = [
self._temp_task(
name=f"{self.name_prefix}{i}",
hyperparams={"test": {"param": {"section": "test", "name": "param", "type": "str", "value": str(i)}}},
hyperparams={
"test": {
"param": {
"section": "test",
"name": "param",
"type": "str",
"value": str(i),
}
}
},
)
for i in range(18)
]
@@ -24,10 +33,7 @@ class TestEntityOrdering(TestService):
for page in range(0, math.ceil(len(self.task_ids) / page_size)):
start = page * page_size
expected_size = min(page_size, len(self.task_ids) - start)
tasks = self._get_tasks(
page=page,
page_size=page_size,
).tasks
tasks = self._get_tasks(page=page, page_size=page_size,).tasks
self.assertEqual(len(tasks), expected_size)
for i, t in enumerate(tasks):
self.assertEqual(t.name, f"{self.name_prefix}{start + i}")
@@ -38,10 +44,7 @@ class TestEntityOrdering(TestService):
for page in range(0, math.ceil(len(self.task_ids) / page_size)):
start = page * page_size
expected_size = min(page_size, len(self.task_ids) - start)
res = self._get_tasks(
size=page_size,
scroll_id=scroll_id,
)
res = self._get_tasks(size=page_size, scroll_id=scroll_id,)
self.assertTrue(res.scroll_id)
scroll_id = res.scroll_id
tasks = res.tasks
@@ -50,24 +53,19 @@ class TestEntityOrdering(TestService):
self.assertEqual(t.name, f"{self.name_prefix}{start + i}")
# no more data in this scroll
tasks = self._get_tasks(
size=page_size,
scroll_id=scroll_id,
).tasks
tasks = self._get_tasks(size=page_size, scroll_id=scroll_id,).tasks
self.assertFalse(tasks)
# refresh brings all
tasks = self._get_tasks(
size=page_size,
scroll_id=scroll_id,
refresh_scroll=True,
size=page_size, scroll_id=scroll_id, refresh_scroll=True,
).tasks
self.assertEqual([t.id for t in tasks], self.task_ids)
def _get_tasks(self, **page_params):
return self.api.tasks.get_all_ex(
name="^Test paging ",
order_by=["hyperparams.param"],
order_by=["hyperparams.test.param.value"],
**page_params,
)