def role(vcr_cassette, client, unique):
"""There is a valid role in the system."""
from datadog_api_client.v2.exceptions import ApiException
from datadog_api_client.v2.model import role_create_request
from datadog_api_client.v2.model import role_create_data
from datadog_api_client.v2.model import role_create_attributes
from datadog_api_client.v2.model import roles_type
from datadog_api_client.v2.api.roles_api import RolesApi
api = RolesApi(client)
body = role_create_request.RoleCreateRequest(
data=role_create_data.RoleCreateData(
type=roles_type.RolesType(value="roles"),
attributes=role_create_attributes.RoleCreateAttributes(
name=str(unique), ),
), )
response = api.create_role(body=body)
yield response
if vcr_cassette.record_mode != "none":
number_of_interactions = len(vcr_cassette.data)
try:
api.delete_role(response.data.id)
except ApiException as e:
warnings.warn(str(e))
vcr_cassette.data = vcr_cassette.data[:number_of_interactions]
def granted_permission(client, role, permission):
"""The permission is granted to the role."""
from datadog_api_client.v2.model import relationship_to_permission
from datadog_api_client.v2.model import relationship_to_permission_data
from datadog_api_client.v2.api.roles_api import RolesApi
api = RolesApi(client)
body = relationship_to_permission.RelationshipToPermission(
data=relationship_to_permission_data.RelationshipToPermissionData(
id=permission.id,
type=permission.type,
))
return api.add_permission_to_role(role.data.id, body=body)
def user_has_role(vcr_cassette, client, user, role):
"""The user has the role."""
from datadog_api_client.v2.exceptions import ApiException
from datadog_api_client.v2.model import relationship_to_user
from datadog_api_client.v2.model import relationship_to_user_data
from datadog_api_client.v2.api.roles_api import RolesApi
api = RolesApi(client)
body = relationship_to_user.RelationshipToUser(
data=relationship_to_user_data.RelationshipToUserData(
id=user.data.id,
type=user.data.type,
))
yield api.add_user_to_role(role.data.id, body=body)
if vcr_cassette.record_mode != "none":
number_of_interactions = len(vcr_cassette.data)
try:
api.remove_user_from_role(role.data.id, body=body)
except ApiException as e:
warnings.warn(str(e))
vcr_cassette.data = vcr_cassette.data[:number_of_interactions]
def permission(client):
"""There is a valid permission in the system."""
from datadog_api_client.v2.api.roles_api import RolesApi
return RolesApi(client).list_permissions().data[0]
def setUp(self):
self.api = RolesApi() # noqa: E501