def setUp(self): super(ApprovalByLabelE2ETest, self).setUp() self.SetUpLegacy() if data_store.RelationalDBReadEnabled(): self.SetUpRelationalDB() cls = (api_call_router_with_approval_checks. ApiCallRouterWithApprovalChecks) cls.ClearCache() self.approver = test_lib.ConfigOverrider({ "API.DefaultRouter": cls.__name__, "ACL.approvers_config_file": os.path.join(self.base_path, "approvers.yaml") }) self.approver.Start() # Get a fresh approval manager object and reload with test approvers. self.approval_manager_stubber = utils.Stubber( client_approval_auth, "CLIENT_APPROVAL_AUTH_MGR", client_approval_auth.ClientApprovalAuthorizationManager()) self.approval_manager_stubber.Start() # Force creation of new APIAuthorizationManager, so that configuration # changes are picked up. api_auth_manager.APIACLInit.InitApiAuthManager()
def setUp(self): super(ApprovalByLabelE2ETest, self).setUp() # Set up clients and labels before we turn on the FullACM. We need to create # the client because to check labels the client needs to exist. client_ids = self.SetupClients(3) self.client_nolabel = rdf_client.ClientURN(client_ids[0]) self.client_nolabel_id = self.client_nolabel.Basename() self.client_legal = rdf_client.ClientURN(client_ids[1]) self.client_legal_id = self.client_legal.Basename() self.client_prod = rdf_client.ClientURN(client_ids[2]) self.client_prod_id = self.client_prod.Basename() with aff4.FACTORY.Open( self.client_legal, aff4_type=aff4_grr.VFSGRRClient, mode="rw", token=self.token) as client_obj: client_obj.AddLabel("legal_approval") with aff4.FACTORY.Open( self.client_prod, aff4_type=aff4_grr.VFSGRRClient, mode="rw", token=self.token) as client_obj: client_obj.AddLabels(["legal_approval", "prod_admin_approval"]) cls = (api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks) cls.ClearCache() self.approver = test_lib.ConfigOverrider({ "API.DefaultRouter": cls.__name__, "ACL.approvers_config_file": os.path.join(self.base_path, "approvers.yaml") }) self.approver.Start() # Get a fresh approval manager object and reload with test approvers. self.approval_manager_stubber = utils.Stubber( client_approval_auth, "CLIENT_APPROVAL_AUTH_MGR", client_approval_auth.ClientApprovalAuthorizationManager()) self.approval_manager_stubber.Start() # Force creation of new APIAuthorizationManager, so that configuration # changes are picked up. api_auth_manager.APIACLInit.InitApiAuthManager()
def setUp(self): super(ClientApprovalAuthorizationManager, self).setUp() self.mgr = client_approval_auth.ClientApprovalAuthorizationManager() self.urn = rdf_client.ClientURN("C.0000000000000000")