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)
