def testClientInfo(self, db: abstract_db.Database): client_id = db_test_utils.InitializeClient(db) startup_info = rdf_client.StartupInfo() startup_info.client_info.client_name = "rrg" startup_info.client_info.client_version = 1337 db.WriteClientStartupInfo(client_id, startup_info) flow = rdf_flow_objects.Flow() flow.client_id = client_id flow.flow_id = self._FLOW_ID flow = FlowBaseTest.Flow(flow) self.assertIsInstance(flow.client_info, rdf_client.ClientInformation) self.assertEqual(flow.client_info.client_name, "rrg") self.assertEqual(flow.client_info.client_version, 1337)
def testClientInfo(self, db: abstract_db.Database): client_id = "C.0123456789ABCDEF" db.WriteClientMetadata(client_id, fleetspeak_enabled=False) startup_info = rdf_client.StartupInfo() startup_info.client_info.client_name = "rrg" startup_info.client_info.client_version = 1337 db.WriteClientStartupInfo(client_id, startup_info) flow = rdf_flow_objects.Flow() flow.client_id = client_id flow.flow_id = "FEDCBA9876543210" flow = FlowBaseTest.Flow(flow) self.assertIsInstance(flow.client_info, rdf_client.ClientInformation) self.assertEqual(flow.client_info.client_name, "rrg") self.assertEqual(flow.client_info.client_version, 1337)