async def setUp(self):
self.backend_mock = Mock(spec=Orchestrator)
self.agents_service = AgentsService()
self.account = Account(**ACCOUNT_DEV_DICT)
self.user = User(**USER_WITH_MULTIPLE_ACCOUNTS_DICT)
async def test_calls_backend_with_right_parameters_add_user(self):
account = Account(**ACCOUNT_DEV_DICT)
user = User(**USER_WITH_NO_ACCOUNTS_DICT)
backend = CoroutineMock(add_user=CoroutineMock())
await AccountsService.add_user_to_account(user, account, backend)
backend.add_user.assert_awaited_with(user, account)
async def test_user_has_permission_permission_ok(self):
user = User(**USER_WITH_MULTIPLE_ACCOUNTS_DICT)
account = Account(**ACCOUNT_DEV_DICT)
self.assertTrue(await account.user_has_permission(user))
async def test_get_accounts_from_user_user_has_no_accounts(self):
user = User(**USER_WITH_NO_ACCOUNTS_DICT)
accounts = await self.backend.get_accounts_from_user(user)
self.assertCountEqual([], accounts)
async def test_get_user_by_id_user_exists(self):
user = await self.backend.get_user_by_id(USER_WITH_ONE_ACCOUNT_ID)
self.assertIsNotNone(user)
self.assertEqual(User(**USER_WITH_ONE_ACCOUNT_DICT), user)
async def test_get_job_by_id_job_not_found(self):
job_id = "job-not-found"
user = User(**USER_WITH_MULTIPLE_ACCOUNTS_DICT)
account = Account(**ACCOUNT_DEV_DICT)
job = await self.backend.get_job_by_id(job_id, user, account)
self.assertIsNone(job)
async def setUp(self):
await super(JobsEndpointTestCase, self).setUp()
self.client = await self.aiohttp_client(app)
self.user = User(**USER_WITH_MULTIPLE_ACCOUNTS_DICT)
self.account = Account(**ACCOUNT_DEV_DICT)