def setUp(self): super(ApiLabelsRestrictedSearchClientsHandlerTest, self).setUp() self.client_ids = self.SetupClients(4) index = aff4.FACTORY.Create(client_index.MAIN_INDEX, aff4_type="ClientIndex", mode="rw", 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.AddLabels(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 setUp(self): super(ApiLabelsRestrictedSearchClientsHandlerTestRelational, self).setUp() self.enable_relational_db = test_lib.ConfigOverrider( {"Database.useForReads": True}) self.enable_relational_db.Start() 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(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(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( labels_whitelist=self.labels_whitelist, labels_owners_whitelist=self.labels_owners_whitelist)