def testGenClientConfig_ignoreBuilderContext(self):
with test_lib.PreserveConfig():
# Define a secondary config with special values for the ClientBuilder
# context.
str_override = """
Test Context:
Client.labels: [label0, label1]
ClientBuilder Context:
Client.labels: [build-label0, build-label1]
"""
parser = config_parser.YamlConfigFileParser("")
override = parser.RawDataFromBytes(str_override.encode("utf-8"))
config.CONFIG.MergeData(override)
# Sanity-check that the secondary config was merged into the global
# config.
self.assertEqual(config.CONFIG["Client.labels"],
["label0", "label1"])
context = [
"Test Context", "ClientBuilder Context", "Client Context"
]
str_client_config = build_helpers.GetClientConfig(context)
client_config = parser.RawDataFromBytes(
str_client_config.encode("utf-8"))
# Settings particular to the ClientBuilder context should not carry over
# into the generated client config.
self.assertEqual(client_config["Client.labels"],
["label0", "label1"])
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.CONFIG["Client.private_key"]
self.server_serial_number = 0
self.server_certificate = config.CONFIG["Frontend.certificate"]
self.server_private_key = config.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.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 testGenClientConfig_ignoreBuilderContext(self):
with test_lib.PreserveConfig():
# Define a secondary config with special values for the ClientBuilder
# context.
str_override = """
Test Context:
Client.labels: [label0, label1]
ClientBuilder Context:
Client.labels: [build-label0, build-label1]
"""
override = config_lib.YamlParser(data=str_override).RawData()
config.CONFIG.MergeData(override)
# Sanity-check that the secondary config was merged into the global
# config.
self.assertEqual(config.CONFIG["Client.labels"],
["label0", "label1"])
repacker = build.ClientRepacker()
context = [
"Test Context", "ClientBuilder Context", "Client Context"
]
str_client_config = repacker.GetClientConfig(context)
client_config = config_lib.YamlParser(
data=str_client_config).RawData()
# Settings particular to the ClientBuilder context should not carry over
# into the generated client config.
self.assertEqual(client_config["Client.labels"],
["label0", "label1"])
def setUp(self):
"""Set up communicator tests."""
super(HTTPClientTests, self).setUp()
# These tests change the config so we preserve state.
config_stubber = test_lib.PreserveConfig()
config_stubber.Start()
self.addCleanup(config_stubber.Stop)
self.server_private_key = config.CONFIG["PrivateKeys.server_key"]
self.server_certificate = config.CONFIG["Frontend.certificate"]
# Make a new client
self.CreateNewClientObject()
# And cache it in the server
self.CreateNewServerCommunicator()
requests_stubber = utils.Stubber(requests, "request", self.UrlMock)
requests_stubber.Start()
self.addCleanup(requests_stubber.Stop)
sleep_stubber = utils.Stubber(time, "sleep", lambda x: None)
sleep_stubber.Start()
self.addCleanup(sleep_stubber.Stop)
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):
"""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.CONFIG["Client.private_key"])
self.server_serial_number = 0
self.server_private_key = config.CONFIG["PrivateKeys.server_key"]
self.server_certificate = config.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):
"""Set up communicator tests."""
super().setUp()
# These tests change the config so we preserve state.
config_stubber = test_lib.PreserveConfig()
config_stubber.Start()
self.addCleanup(config_stubber.Stop)
self.client_private_key = config.CONFIG["Client.private_key"]
self.server_certificate = config.CONFIG["Frontend.certificate"]
self.server_private_key = config.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.CONFIG["CA.certificate"])
self.last_urlmock_error = None
self._SetupCommunicator()
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()
self.server_serial_number = 0
self.server_private_key = config.CONFIG["PrivateKeys.server_key"]
self.server_certificate = config.CONFIG["Frontend.certificate"]
# 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
# 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()
config_stubber = test_lib.PreserveConfig()
config_stubber.Start()
self.addCleanup(config_stubber.Stop)
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()
def setUp(self): super().setUp() # These tests change the config so we preserve state. config_stubber = test_lib.PreserveConfig() config_stubber.Start() self.addCleanup(config_stubber.Stop)
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().setUp() config_stubber = test_lib.PreserveConfig() config_stubber.Start() self.addCleanup(config_stubber.Stop)
