def Run(self): # Fix the time to avoid regressions. with test_lib.FakeTime(42): client_id = self.SetupClient(0) ip = rdf_client_network.NetworkAddress( human_readable_address="192.168.100.42", address_type=rdf_client_network.NetworkAddress.Family.INET) data_store.REL_DB.WriteClientMetadata(client_id, last_ip=ip) self.Check( "GetLastClientIPAddress", args=client_plugin.ApiGetLastClientIPAddressArgs(client_id=client_id))
def Run(self): # Fix the time to avoid regressions. with test_lib.FakeTime(42): if data_store.RelationalDBReadEnabled(): client_obj = self.SetupTestClientObject(0) client_id = client_obj.client_id ip = rdf_client.NetworkAddress( human_readable_address="192.168.100.42", address_type=rdf_client.NetworkAddress.Family.INET) data_store.REL_DB.WriteClientMetadata(client_id, last_ip=ip) else: client_urn = self.SetupClient(0) client_id = client_urn.Basename() with aff4.FACTORY.Open( client_id, mode="rw", token=self.token) as grr_client: grr_client.Set(grr_client.Schema.CLIENT_IP("192.168.100.42")) self.Check( "GetLastClientIPAddress", args=client_plugin.ApiGetLastClientIPAddressArgs(client_id=client_id))