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) ])