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)
