def testAddLabels(self):
client_ref = self.api.Client(client_id=self.client_urn.Basename())
self.assertEqual(list(client_ref.Get().data.labels), [])
with test_lib.FakeTime(42):
client_ref.AddLabels(["foo", "bar"])
self.assertEqual(
sorted(client_ref.Get().data.labels, key=lambda l: l.name), [
jobs_pb2.AFF4ObjectLabel(
name="bar", owner=self.token.username, timestamp=42000000),
jobs_pb2.AFF4ObjectLabel(
name="foo", owner=self.token.username, timestamp=42000000)
])
def testRemoveLabels(self):
with test_lib.FakeTime(42):
with aff4.FACTORY.Open(self.client_urn,
aff4_type=aff4_grr.VFSGRRClient,
mode="rw",
token=self.token) as client_obj:
client_obj.AddLabels(["bar", "foo"])
client_ref = self.api.Client(client_id=self.client_urn.Basename())
self.assertEqual(
sorted(client_ref.Get().data.labels, key=lambda l: l.name), [
jobs_pb2.AFF4ObjectLabel(
name="bar", owner=self.token.username, timestamp=42000000),
jobs_pb2.AFF4ObjectLabel(
name="foo", owner=self.token.username, timestamp=42000000)
])
client_ref.RemoveLabel("foo")
self.assertEqual(
sorted(client_ref.Get().data.labels, key=lambda l: l.name), [
jobs_pb2.AFF4ObjectLabel(
name="bar", owner=self.token.username, timestamp=42000000)
])
