def testAuditEntryIsCreatedForEveryClient(self):
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=self.client_ids, labels=["drei", "ein", "zwei"]),
token=self.token)
# We need to run .Simulate() so that the appropriate event is fired,
# collected, and finally written to the logs that we inspect.
mock_worker = test_lib.MockWorker(token=self.token)
mock_worker.Simulate()
parentdir = aff4.FACTORY.Open("aff4:/audit/logs", token=self.token)
log = list(parentdir.ListChildren())[0]
fd = aff4.FACTORY.Open(log, token=self.token)
for client_id in self.client_ids:
found_event = None
for event in fd:
if (event.action == events.AuditEvent.Action.CLIENT_ADD_LABEL
and event.client == rdf_client.ClientURN(client_id)):
found_event = event
break
self.assertFalse(found_event is None)
self.assertEqual(found_event.user, self.token.username)
self.assertEqual(
found_event.description, "%s.drei,%s.ein,%s.zwei" %
(self.token.username, self.token.username,
self.token.username))
def testAddsSingleLabelToSingleClient(self):
for client_id in self.client_ids:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
data_store.REL_DB.WriteClientMetadata(client_id.Basename(),
fleetspeak_enabled=False)
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=[self.client_ids[0]], labels=["foo"]),
token=self.token)
# AFF4 labels.
labels = aff4.FACTORY.Open(self.client_ids[0],
token=self.token).GetLabels()
self.assertEqual(len(labels), 1)
self.assertEqual(labels[0].name, "foo")
self.assertEqual(labels[0].owner, self.token.username)
for client_id in self.client_ids[1:]:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
# Relational DB labels.
cid = self.client_ids[0].Basename()
labels = data_store.REL_DB.ReadClientLabels(cid)
self.assertEqual(len(labels), 1)
self.assertEqual(labels[0].name, "foo")
self.assertEqual(labels[0].owner, self.token.username)
for client_id in self.client_ids[1:]:
self.assertFalse(
data_store.REL_DB.ReadClientLabels(client_id.Basename()))
def testAddsTwoLabelsToTwoClients(self):
for client_id in self.client_ids:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=[self.client_ids[0], self.client_ids[1]],
labels=["foo", "bar"]),
token=self.token)
# AFF4 labels.
for client_id in self.client_ids[:2]:
labels = aff4.FACTORY.Open(client_id, token=self.token).GetLabels()
self.assertEqual(len(labels), 2)
self.assertEqual(labels[0].name, "foo")
self.assertEqual(labels[0].owner, self.token.username)
self.assertEqual(labels[1].name, "bar")
self.assertEqual(labels[1].owner, self.token.username)
self.assertFalse(
aff4.FACTORY.Open(self.client_ids[2],
token=self.token).GetLabels())
# Relational labels.
for client_id in self.client_ids[:2]:
labels = data_store.REL_DB.GetClientLabels(client_id.Basename())
self.assertEqual(len(labels), 2)
self.assertEqual(labels[0].owner, self.token.username)
self.assertEqual(labels[1].owner, self.token.username)
self.assertItemsEqual([labels[0].name, labels[1].name],
["bar", "foo"])
self.assertFalse(
data_store.REL_DB.GetClientLabels(self.client_ids[2].Basename()))
def testAuditEntryIsCreatedForEveryClient(self):
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=self.client_ids, labels=["drei", "ein", "zwei"]),
token=self.token)
# We need to run .Simulate() so that the appropriate event is fired,
# collected, and finally written to the logs that we inspect.
mock_worker = worker_test_lib.MockWorker(token=self.token)
mock_worker.Simulate()
event = self._FindAuditEvent()
self.assertIsNotNone(event)
self.assertEqual(event.user, self.token.username)
self.assertEqual(
event.description, "%s.drei,%s.ein,%s.zwei" %
(self.token.username, self.token.username, self.token.username))
def testAddsSingleLabelToSingleClient(self):
for client_id in self.client_ids:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=[self.client_ids[0]], labels=["foo"]),
token=self.token)
labels = aff4.FACTORY.Open(self.client_ids[0],
token=self.token).GetLabels()
self.assertEqual(len(labels), 1)
self.assertEqual(labels[0].name, "foo")
self.assertEqual(labels[0].owner, self.token.username)
for client_id in self.client_ids[1:]:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
def testAddsTwoLabelsToTwoClients(self):
for client_id in self.client_ids:
self.assertFalse(
aff4.FACTORY.Open(client_id, token=self.token).GetLabels())
self.handler.Handle(client_plugin.ApiAddClientsLabelsArgs(
client_ids=[self.client_ids[0], self.client_ids[1]],
labels=["foo", "bar"]),
token=self.token)
for client_id in self.client_ids[:2]:
labels = aff4.FACTORY.Open(client_id, token=self.token).GetLabels()
self.assertEqual(len(labels), 2)
self.assertEqual(labels[0].name, "foo")
self.assertEqual(labels[0].owner, self.token.username)
self.assertEqual(labels[1].name, "bar")
self.assertEqual(labels[1].owner, self.token.username)
self.assertFalse(
aff4.FACTORY.Open(self.client_ids[2],
token=self.token).GetLabels())
