def setUp(self): super(TestHostInformation, self).setUp() self.client_id = "C.0000000000000001" with self.ACLChecksDisabled(): self.RequestAndGrantClientApproval(self.client_id) with test_lib.FakeTime(TIME_0): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("6.1.7000")) fd.Set(fd.Schema.HOSTNAME("Hostname T0")) with test_lib.FakeTime(TIME_1): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("6.1.8000")) fd.Set(fd.Schema.HOSTNAME("Hostname T1")) with test_lib.FakeTime(TIME_2): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("7.0.0000")) fd.Set(fd.Schema.HOSTNAME("Hostname T2"))
def setUp(self): super(TestHostInformation, self).setUp() self.client_id = "C.0000000000000001" fixture_test_lib.ClientFixture(self.client_id, token=self.token) self.RequestAndGrantClientApproval(self.client_id) with test_lib.FakeTime(gui_test_lib.TIME_0): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("6.1.7000")) fd.Set(fd.Schema.HOSTNAME("Hostname T0")) fd.Set(fd.Schema.MEMORY_SIZE(4294967296)) with test_lib.FakeTime(gui_test_lib.TIME_1): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("6.1.8000")) fd.Set(fd.Schema.HOSTNAME("Hostname T1")) fd.Set(fd.Schema.MEMORY_SIZE(8589934592)) with test_lib.FakeTime(gui_test_lib.TIME_2): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("7.0.0000")) fd.Set(fd.Schema.HOSTNAME("Hostname T2")) fd.Set(fd.Schema.MEMORY_SIZE(12884901888))
def setUp(self): super(TestHostInformation, self).setUp() self.client_id = "C.0000000000000001" with self.ACLChecksDisabled(): self.RequestAndGrantClientApproval(self.client_id) with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString("6.1.7601"))
def _WriteClientSnapshot(self, timestamp, version, hostname, memory): with test_lib.FakeTime(timestamp): with aff4.FACTORY.Open(self.client_id, mode="rw", token=self.token) as fd: fd.Set(fd.Schema.OS_VERSION, rdf_client.VersionString(version)) fd.Set(fd.Schema.HOSTNAME(hostname)) fd.Set(fd.Schema.MEMORY_SIZE(memory)) if data_store.RelationalDBReadEnabled(): snapshot = data_store.REL_DB.ReadClientSnapshot(self.client_id) with test_lib.FakeTime(timestamp): snapshot.os_version = version snapshot.knowledge_base.fqdn = hostname snapshot.memory_size = memory data_store.REL_DB.WriteClientSnapshot(snapshot)