def setUp(self):
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir, 'agentConfig.json')
self.__config_fragments_dir = os.path.join(self.__config_dir, 'configs.d')
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, 'w')
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
def setUp(self):
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir, "agentConfig.json")
self.__config_fragments_dir = os.path.join(self.__config_dir, "configs.d")
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, "w")
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
class CopyingParamsTest(unittest.TestCase):
def setUp(self):
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir,
'agentConfig.json')
self.__config_fragments_dir = os.path.join(self.__config_dir,
'configs.d')
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, 'w')
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
def test_initial_settings(self):
self.assertEquals(self.test_params.current_bytes_allowed_to_send,
ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_no_events_being_sent(self):
for i in range(0, 5):
self.test_params.update_params('success', 0)
self.assertEquals(self.test_params.current_bytes_allowed_to_send,
ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_small_events_being_sent(self):
self.test_params.current_sleep_interval = 1
self.run_test_case('success', 10 * 1024, [1.5, ONE_MB], [2.25, ONE_MB],
[3.375, ONE_MB], [5, ONE_MB])
def test_too_many_events_being_sent(self):
self.test_params.current_sleep_interval = 5
self.run_test_case('success', 200 * 1024, [3.0, ONE_MB], [1.8, ONE_MB],
[1.08, ONE_MB], [1, ONE_MB])
def test_request_too_big(self):
self.test_params.current_sleep_interval = 1
self.test_params.update_params('requestTooLarge', 300 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send,
150 * 1024)
self.test_params.update_params('requestTooLarge', 150 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send,
100 * 1024)
def test_error_back_off(self):
self.test_params.current_sleep_interval = 3
self.run_test_case('error', 200 * 1024, [4.5, ONE_MB], [6.75, ONE_MB],
[10.125, ONE_MB], [15.1875, ONE_MB],
[22.78125, ONE_MB], [30, ONE_MB])
def run_test_case(self, status, bytes_sent,
*expected_sleep_interval_allowed_bytes):
"""Verifies that when test_params is updated with the specified status and bytes sent the current sleep
interval and allowed bytes is updated to the given values.
This will call test_params.update_params N times where N is the number of additional arguments supplied.
After the ith invocation of test_params.update_params, the values for the current_sleep_interval and
current_bytes_allowed_to_send will be checked against the ith additional parameter.
@param status: The status to use when invoking test_params.update_params.
@param bytes_sent: The number of bytes sent to use when invoking test_params.update_params.
@param expected_sleep_interval_allowed_bytes: A variable number of two element arrays where the first element
is the expected value for current_sleep_interval and the second is the expected value of
current_bytes_allowed_to_send. Each subsequent array represents what those values should be after invoking
test_params.update_param again.
"""
for expected_result in expected_sleep_interval_allowed_bytes:
self.test_params.update_params(status, bytes_sent)
self.assertAlmostEquals(self.test_params.current_sleep_interval,
expected_result[0])
self.assertAlmostEquals(
self.test_params.current_bytes_allowed_to_send,
expected_result[1])
class LogObject(object):
def __init__(self, config):
self.config = config
self.log_path = config['path']
class MonitorObject(object):
def __init__(self, config):
self.module_name = config['module']
self.config = config
self.log_config = {
'path': self.module_name.split('.')[-1] + '.log'
}
def __create_test_configuration_instance(self):
default_paths = DefaultPaths('/var/log/scalyr-agent-2',
'/etc/scalyr-agent-2/agent.json',
'/var/lib/scalyr-agent-2')
def log_factory(config):
return CopyingParamsTest.LogObject(config)
def monitor_factory(config, _):
return CopyingParamsTest.MonitorObject(config)
monitors = [
JsonObject(
module='scalyr_agent.builtin_monitors.linux_system_metrics'),
JsonObject(
module='scalyr_agent.builtin_monitors.linux_process_metrics',
pid='$$',
id='agent')
]
return Configuration(self.__config_file, default_paths, monitors,
log_factory, monitor_factory)
class CopyingParamsTest(ScalyrTestCase):
def setUp(self):
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir, "agentConfig.json")
self.__config_fragments_dir = os.path.join(self.__config_dir, "configs.d")
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, "w")
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
def test_initial_settings(self):
self.assertEquals(self.test_params.current_bytes_allowed_to_send, ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_no_events_being_sent(self):
for i in range(0, 5):
self.test_params.update_params("success", 0)
self.assertEquals(self.test_params.current_bytes_allowed_to_send, ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_small_events_being_sent(self):
self.test_params.current_sleep_interval = 1
self.run_test_case("success", 10 * 1024, [1.5, ONE_MB], [2.25, ONE_MB], [3.375, ONE_MB], [5, ONE_MB])
def test_too_many_events_being_sent(self):
self.test_params.current_sleep_interval = 5
self.run_test_case("success", 200 * 1024, [3.0, ONE_MB], [1.8, ONE_MB], [1.08, ONE_MB], [1, ONE_MB])
def test_request_too_big(self):
self.test_params.current_sleep_interval = 1
self.test_params.update_params("requestTooLarge", 300 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, 150 * 1024)
self.test_params.update_params("requestTooLarge", 150 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, 100 * 1024)
def test_error_back_off(self):
self.test_params.current_sleep_interval = 3
self.run_test_case(
"error",
200 * 1024,
[4.5, ONE_MB],
[6.75, ONE_MB],
[10.125, ONE_MB],
[15.1875, ONE_MB],
[22.78125, ONE_MB],
[30, ONE_MB],
)
def run_test_case(self, status, bytes_sent, *expected_sleep_interval_allowed_bytes):
"""Verifies that when test_params is updated with the specified status and bytes sent the current sleep
interval and allowed bytes is updated to the given values.
This will call test_params.update_params N times where N is the number of additional arguments supplied.
After the ith invocation of test_params.update_params, the values for the current_sleep_interval and
current_bytes_allowed_to_send will be checked against the ith additional parameter.
@param status: The status to use when invoking test_params.update_params.
@param bytes_sent: The number of bytes sent to use when invoking test_params.update_params.
@param expected_sleep_interval_allowed_bytes: A variable number of two element arrays where the first element
is the expected value for current_sleep_interval and the second is the expected value of
current_bytes_allowed_to_send. Each subsequent array represents what those values should be after invoking
test_params.update_param again.
"""
for expected_result in expected_sleep_interval_allowed_bytes:
self.test_params.update_params(status, bytes_sent)
self.assertAlmostEquals(self.test_params.current_sleep_interval, expected_result[0])
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, expected_result[1])
class LogObject(object):
def __init__(self, config):
self.config = config
self.log_path = config["path"]
class MonitorObject(object):
def __init__(self, config):
self.module_name = config["module"]
self.config = config
self.log_config = {"path": self.module_name.split(".")[-1] + ".log"}
def __create_test_configuration_instance(self):
default_paths = DefaultPaths(
"/var/log/scalyr-agent-2", "/etc/scalyr-agent-2/agent.json", "/var/lib/scalyr-agent-2"
)
return Configuration(self.__config_file, default_paths)
class CopyingParamsTest(unittest.TestCase):
def setUp(self):
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir, 'agentConfig.json')
self.__config_fragments_dir = os.path.join(self.__config_dir, 'configs.d')
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, 'w')
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
def test_initial_settings(self):
self.assertEquals(self.test_params.current_bytes_allowed_to_send, ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_no_events_being_sent(self):
for i in range(0, 5):
self.test_params.update_params('success', 0)
self.assertEquals(self.test_params.current_bytes_allowed_to_send, ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_small_events_being_sent(self):
self.test_params.current_sleep_interval = 1
self.run_test_case('success', 10 * 1024, [1.5, ONE_MB], [2.25, ONE_MB], [3.375, ONE_MB], [5, ONE_MB])
def test_too_many_events_being_sent(self):
self.test_params.current_sleep_interval = 5
self.run_test_case('success', 200 * 1024, [3.0, ONE_MB], [1.8, ONE_MB], [1.08, ONE_MB], [1, ONE_MB])
def test_request_too_big(self):
self.test_params.current_sleep_interval = 1
self.test_params.update_params('requestTooLarge', 300 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, 150 * 1024)
self.test_params.update_params('requestTooLarge', 150 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, 100 * 1024)
def test_error_back_off(self):
self.test_params.current_sleep_interval = 3
self.run_test_case('error', 200 * 1024, [4.5, ONE_MB], [6.75, ONE_MB], [10.125, ONE_MB], [15.1875, ONE_MB],
[22.78125, ONE_MB], [30, ONE_MB])
def run_test_case(self, status, bytes_sent, *expected_sleep_interval_allowed_bytes):
"""Verifies that when test_params is updated with the specified status and bytes sent the current sleep
interval and allowed bytes is updated to the given values.
This will call test_params.update_params N times where N is the number of additional arguments supplied.
After the ith invocation of test_params.update_params, the values for the current_sleep_interval and
current_bytes_allowed_to_send will be checked against the ith additional parameter.
@param status: The status to use when invoking test_params.update_params.
@param bytes_sent: The number of bytes sent to use when invoking test_params.update_params.
@param expected_sleep_interval_allowed_bytes: A variable number of two element arrays where the first element
is the expected value for current_sleep_interval and the second is the expected value of
current_bytes_allowed_to_send. Each subsequent array represents what those values should be after invoking
test_params.update_param again.
"""
for expected_result in expected_sleep_interval_allowed_bytes:
self.test_params.update_params(status, bytes_sent)
self.assertAlmostEquals(self.test_params.current_sleep_interval, expected_result[0])
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send, expected_result[1])
class LogObject(object):
def __init__(self, config):
self.config = config
self.log_path = config['path']
class MonitorObject(object):
def __init__(self, config):
self.module_name = config['module']
self.config = config
self.log_config = {'path': self.module_name.split('.')[-1] + '.log'}
def __create_test_configuration_instance(self):
default_paths = DefaultPaths('/var/log/scalyr-agent-2', '/etc/scalyr-agent-2/agent.json',
'/var/lib/scalyr-agent-2')
def log_factory(config):
return CopyingParamsTest.LogObject(config)
def monitor_factory(config, _):
return CopyingParamsTest.MonitorObject(config)
monitors = [JsonObject(module='scalyr_agent.builtin_monitors.linux_system_metrics'),
JsonObject(module='scalyr_agent.builtin_monitors.linux_process_metrics',
pid='$$', id='agent')]
return Configuration(self.__config_file, default_paths, monitors, log_factory, monitor_factory)
class CopyingParamsTest(ScalyrTestCase):
def setUp(self):
super(CopyingParamsTest, self).setUp()
self.__config_dir = tempfile.mkdtemp()
self.__config_file = os.path.join(self.__config_dir,
"agentConfig.json")
self.__config_fragments_dir = os.path.join(self.__config_dir,
"configs.d")
os.makedirs(self.__config_fragments_dir)
fp = open(self.__config_file, "w")
fp.write('{api_key: "fake"}')
fp.close()
config = self.__create_test_configuration_instance()
config.parse()
self.test_params = CopyingParameters(config)
def test_initial_settings(self):
self.assertEquals(self.test_params.current_bytes_allowed_to_send,
ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_no_events_being_sent(self):
for i in range(0, 5):
self.test_params.update_params("success", 0)
self.assertEquals(self.test_params.current_bytes_allowed_to_send,
ONE_MB)
self.assertEquals(self.test_params.current_sleep_interval, 5.0)
def test_small_events_being_sent(self):
self.test_params.current_sleep_interval = 1
self._run(
"success",
10 * 1024,
[1.5, ONE_MB],
[2.25, ONE_MB],
[3.375, ONE_MB],
[5, ONE_MB],
)
def test_too_many_events_being_sent(self):
self.test_params.current_sleep_interval = 5
self._run(
"success",
200 * 1024,
[3.0, ONE_MB],
[1.8, ONE_MB],
[1.08, ONE_MB],
[1, ONE_MB],
)
def test_request_too_big(self):
self.test_params.current_sleep_interval = 1
self.test_params.update_params("requestTooLarge", 300 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send,
150 * 1024)
self.test_params.update_params("requestTooLarge", 150 * 1024)
self.assertAlmostEquals(self.test_params.current_bytes_allowed_to_send,
100 * 1024)
def test_error_back_off(self):
self.test_params.current_sleep_interval = 3
self._run(
"error",
200 * 1024,
[4.5, ONE_MB],
[6.75, ONE_MB],
[10.125, ONE_MB],
[15.1875, ONE_MB],
[22.78125, ONE_MB],
[30, ONE_MB],
)
def _run(self, status, bytes_sent, *expected_sleep_interval_allowed_bytes):
"""Verifies that when test_params is updated with the specified status and bytes sent the current sleep
interval and allowed bytes is updated to the given values.
This will call test_params.update_params N times where N is the number of additional arguments supplied.
After the ith invocation of test_params.update_params, the values for the current_sleep_interval and
current_bytes_allowed_to_send will be checked against the ith additional parameter.
@param status: The status to use when invoking test_params.update_params.
@param bytes_sent: The number of bytes sent to use when invoking test_params.update_params.
@param expected_sleep_interval_allowed_bytes: A variable number of two element arrays where the first element
is the expected value for current_sleep_interval and the second is the expected value of
current_bytes_allowed_to_send. Each subsequent array represents what those values should be after invoking
test_params.update_param again.
"""
for expected_result in expected_sleep_interval_allowed_bytes:
self.test_params.update_params(status, bytes_sent)
self.assertAlmostEquals(self.test_params.current_sleep_interval,
expected_result[0])
self.assertAlmostEquals(
self.test_params.current_bytes_allowed_to_send,
expected_result[1])
class LogObject(object):
def __init__(self, config):
self.config = config
self.log_path = config["path"]
class MonitorObject(object):
def __init__(self, config):
self.module_name = config["module"]
self.config = config
self.log_config = {
"path": self.module_name.split(".")[-1] + ".log"
}
def __create_test_configuration_instance(self):
default_paths = DefaultPaths(
"/var/log/scalyr-agent-2",
"/etc/scalyr-agent-2/agent.json",
"/var/lib/scalyr-agent-2",
)
return Configuration(self.__config_file, default_paths, None)
