def setUp(self): """Set up communicator tests.""" super(ClientCommsTest, self).setUp() # These tests change the config so we preserve state. self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start() self.client_private_key = config_lib.CONFIG["Client.private_key"] self.server_serial_number = 0 self.server_certificate = config_lib.CONFIG["Frontend.certificate"] self.server_private_key = config_lib.CONFIG["PrivateKeys.server_key"] self.client_communicator = comms.ClientCommunicator( private_key=self.client_private_key) self.client_communicator.LoadServerCertificate( server_certificate=self.server_certificate, ca_certificate=config_lib.CONFIG["CA.certificate"]) self.server_communicator = front_end.ServerCommunicator( certificate=self.server_certificate, private_key=self.server_private_key, token=self.token) self.last_urlmock_error = None
def setUp(self): super(HTTPDataStoreMixin, self).setUp() # These tests change the config so we preserve state. self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start() if not PORT: self.SetupDataStore() else: self._SetConfig(self.temp_dir)
def setUp(self): """Set up communicator tests.""" super(HTTPClientTests, self).setUp() # These tests change the config so we preserve state. self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start() certificate = self.ClientCertFromPrivateKey( config_lib.CONFIG["Client.private_key"]) self.server_serial_number = 0 self.server_private_key = config_lib.CONFIG["PrivateKeys.server_key"] self.server_certificate = config_lib.CONFIG["Frontend.certificate"] self.client_cn = certificate.GetCN() # Make a new client self.CreateNewClientObject() # The housekeeper threads of the time based caches also call time.time and # interfere with some tests so we disable them here. utils.InterruptableThread.exit = True # The same also applies to the StatsCollector thread. stats.StatsCollector.exit = True # Make a client mock self.client = aff4.FACTORY.Create(self.client_cn, aff4_grr.VFSGRRClient, mode="rw", token=self.token) self.client.Set(self.client.Schema.CERT(certificate.AsPEM())) self.client.Flush() # Stop the client from actually processing anything self.out_queue_overrider = test_lib.ConfigOverrider( {"Client.max_out_queue": 0}) self.out_queue_overrider.Start() # And cache it in the server self.CreateNewServerCommunicator() self.requests_stubber = utils.Stubber(requests, "request", self.UrlMock) self.requests_stubber.Start() self.sleep_stubber = utils.Stubber(time, "sleep", lambda x: None) self.sleep_stubber.Start() self.messages = [] ca_enroller.enrolment_cache.Flush() # Response to send back to clients. self.server_response = dict(session_id="aff4:/W:session", name="Echo", response_id=2)
def setUp(self): super(TestCryptoTypeInfos, self).setUp() self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start()
def setUp(self): super(HTTPDataStoreMixin, self).setUp() # These tests change the config so we preserve state. self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start()
def setUp(self): super(ConfigActionTest, self).setUp() # These tests change the config so we preserve state. self.config_stubber = test_lib.PreserveConfig() self.config_stubber.Start()