Python ApiGrrUserInterfaceTraits Examples

Example #1

File: api_labels_restricted_call_router.py Project: bharadwaj1997/grr

    def GetGrrUser(self, args, token=None):
        # Everybody can get their own user object.

        interface_traits = api_user.ApiGrrUserInterfaceTraits(
            search_clients_action_enabled=True)
        return api_user.ApiGetOwnGrrUserHandler(
            interface_traits=interface_traits)

Example #2

File: api_call_router_with_approval_checks_test.py Project: swipswaps/grr

    def testGetGrrUserReturnsRestrictedTraitsForNonAdminUser(self):
        error = access_control.UnauthorizedAccess("some error")
        self.access_checker_mock.CheckIfUserIsAdmin.side_effect = error
        handler = self.router.GetGrrUser(None, context=self.context)

        self.assertNotEqual(handler.interface_traits,
                            api_user.ApiGrrUserInterfaceTraits().EnableAll())

Example #3

File: api_call_router_with_approval_checks.py Project: youngjun-chang/grr

  def GetGrrUser(self, args, token=None):
    # Everybody can get their own user settings.

    interface_traits = api_user.ApiGrrUserInterfaceTraits().EnableAll()
    try:
      self.access_checker.CheckIfUserIsAdmin(token.username)
    except access_control.UnauthorizedAccess:
      interface_traits.manage_binaries_nav_item_enabled = False

    return api_user.ApiGetOwnGrrUserHandler(interface_traits=interface_traits)

Example #4

  def testRendersTraitsPassedInConstructor(self):
    result = self.handler.Handle(
        None, context=api_call_context.ApiCallContext(username=u"foo"))
    self.assertFalse(result.interface_traits.create_hunt_action_enabled)

    handler = user_plugin.ApiGetOwnGrrUserHandler(
        interface_traits=user_plugin.ApiGrrUserInterfaceTraits(
            create_hunt_action_enabled=True))
    result = handler.Handle(
        None, context=api_call_context.ApiCallContext(username=u"foo"))
    self.assertTrue(result.interface_traits.create_hunt_action_enabled)

Example #5

    def testRendersTraitsPassedInConstructor(self):
        result = self.handler.Handle(
            None, token=access_control.ACLToken(username="******"))
        self.assertFalse(result.interface_traits.create_hunt_action_enabled)

        handler = user_plugin.ApiGetOwnGrrUserHandler(
            interface_traits=user_plugin.ApiGrrUserInterfaceTraits(
                create_hunt_action_enabled=True))
        result = handler.Handle(None,
                                token=access_control.ACLToken(username="******"))
        self.assertTrue(result.interface_traits.create_hunt_action_enabled)

Example #6

    def GetGrrUser(self, args, context=None):
        # Everybody can get their own user settings.

        interface_traits = api_user.ApiGrrUserInterfaceTraits().EnableAll()
        try:
            # Without access to restricted flows, one can not launch Python hacks and
            # binaries. Hence, we don't display the "Manage binaries" page.
            self.access_checker.CheckIfHasAccessToRestrictedFlows(
                context.username)
        except access_control.UnauthorizedAccess:
            interface_traits.manage_binaries_nav_item_enabled = False

        return api_user.ApiGetOwnGrrUserHandler(
            interface_traits=interface_traits)

Example #7

File: api_call_router_without_checks.py Project: khanhgithead/grr

 def GetGrrUser(self, args, context=None):
     return api_user.ApiGetOwnGrrUserHandler(
         interface_traits=api_user.ApiGrrUserInterfaceTraits().EnableAll())

Example #8

 def testRaisesIfTraitsSetInRequest(self):
   user = user_plugin.ApiGrrUser(
       interface_traits=user_plugin.ApiGrrUserInterfaceTraits())
   with self.assertRaises(ValueError):
     self.handler.Handle(
         user, context=api_call_context.ApiCallContext(username=u"foo"))

Example #9

File: api_call_router_with_approval_checks_test.py Project: swipswaps/grr

    def testGetGrrUserReturnsFullTraitsForAdminUser(self):
        handler = self.router.GetGrrUser(None, context=self.context)

        self.assertEqual(handler.interface_traits,
                         api_user.ApiGrrUserInterfaceTraits().EnableAll())

Example #10

File: user_test.py Project: cdstelly/grr

 def testRaisesIfTraitsSetInRequest(self):
     user = user_plugin.ApiGrrUser(
         interface_traits=user_plugin.ApiGrrUserInterfaceTraits())
     with self.assertRaises(ValueError):
         self.handler.Handle(user,
                             token=access_control.ACLToken(username=u"foo"))

Example #11

File: api_call_router_without_checks.py Project: avmi/grr

 def GetGrrUser(self, args, context=None):
     interface_traits = api_user.ApiGrrUserInterfaceTraits().EnableAll()
     interface_traits.hunt_approval_required = False
     return api_user.ApiGetOwnGrrUserHandler(
         interface_traits=interface_traits)

Example #12

File: api_call_router_with_approval_checks_test.py Project: avmi/grr

    def testGetGrrUserReturnsFullTraitsForWhenWithRestrictedFlowsAccess(self):
        handler = self.router.GetGrrUser(None, context=self.context)

        self.assertEqual(handler.interface_traits,
                         api_user.ApiGrrUserInterfaceTraits().EnableAll())