def setUp(self): super(ApiLabelsRestrictedSearchClientsHandlerTestRelational, self).setUp() self.client_ids = sorted(self.SetupTestClientObjects(4)) data_store.REL_DB.AddClientLabels(self.client_ids[0], "david", ["foo"]) data_store.REL_DB.AddClientLabels(self.client_ids[1], "david", ["not-foo"]) data_store.REL_DB.AddClientLabels(self.client_ids[2], "peter_oth", ["bar"]) data_store.REL_DB.AddClientLabels(self.client_ids[3], "peter", ["bar"]) index = client_index.ClientIndex() index.AddClientLabels(self.client_ids[0], ["foo"]) index.AddClientLabels(self.client_ids[1], ["not-foo"]) index.AddClientLabels(self.client_ids[2], ["bar"]) index.AddClientLabels(self.client_ids[3], ["bar"]) self.handler = client_plugin.ApiLabelsRestrictedSearchClientsHandler( labels_whitelist=["foo", "bar"], labels_owners_whitelist=["david", "peter"])
def setUp(self): super().setUp() self.client_ids = self.SetupClients(4) data_store.REL_DB.AddClientLabels(self.client_ids[0], u"david", [u"foo"]) data_store.REL_DB.AddClientLabels(self.client_ids[1], u"david", [u"not-foo"]) data_store.REL_DB.AddClientLabels(self.client_ids[2], u"peter_oth", [u"bar"]) data_store.REL_DB.AddClientLabels(self.client_ids[3], u"peter", [u"bar"]) index = client_index.ClientIndex() index.AddClientLabels(self.client_ids[0], [u"foo"]) index.AddClientLabels(self.client_ids[1], [u"not-foo"]) index.AddClientLabels(self.client_ids[2], [u"bar"]) index.AddClientLabels(self.client_ids[3], [u"bar"]) self.handler = client_plugin.ApiLabelsRestrictedSearchClientsHandler( allow_labels=[u"foo", u"bar"], allow_labels_owners=[u"david", u"peter"])
def setUp(self): super(ApiLabelsRestrictedSearchClientsHandlerTestAFF4, self).setUp() self.client_ids = [u.Basename() for u in self.SetupClients(4)] index = client_index.CreateClientIndex(token=self.token) def LabelClient(i, label, owner): with aff4.FACTORY.Open( self.client_ids[i], mode="rw", token=self.token) as grr_client: grr_client.AddLabel(label, owner=owner) index.AddClient(grr_client) LabelClient(0, "foo", "david") LabelClient(1, "not-foo", "david") LabelClient(2, "bar", "peter_another") LabelClient(3, "bar", "peter") self.handler = client_plugin.ApiLabelsRestrictedSearchClientsHandler( labels_whitelist=["foo", "bar"], labels_owners_whitelist=["david", "peter"])
def SearchClients(self, args, token=None): return api_client.ApiLabelsRestrictedSearchClientsHandler( allow_labels=self.allow_labels, allow_labels_owners=self.allow_labels_owners)
def SearchClients(self, args, token=None): return api_client.ApiLabelsRestrictedSearchClientsHandler( labels_whitelist=self.labels_whitelist, labels_owners_whitelist=self.labels_owners_whitelist)