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)
