def test_log_param_sets_input_parameter_data_with_multiple_params(self):
from foundations_internal.foundations_serializer import loads
self.foundations_job.project_name = self.project_name
self._set_job_running()
log_param(self.fake_key, self.fake_value)
log_param(self.fake_key_two, self.fake_value_two)
expected_params = [{
'argument': {
'name': self.fake_key,
'value': {
'type': 'dynamic',
'name': self.fake_key
}
},
'stage_uuid': 'stageless'
}, {
'argument': {
'name': self.fake_key_two,
'value': {
'type': 'dynamic',
'name': self.fake_key_two
}
},
'stage_uuid': 'stageless'
}]
actual_params = loads(
self.redis_connection.get('jobs:{}:{}'.format(
self.job_id, 'input_parameters')))
self.assertEqual(expected_params, actual_params)
def test_log_param_inserts_parameter_key_into_input_params_keys_set(self):
self.foundations_job.project_name = self.project_name
self._set_job_running()
log_param(self.fake_key, self.fake_value)
self.assertEqual(
set([bytes(self.fake_key, 'ascii')]),
self.redis_connection.smembers('projects:{}:{}'.format(
self.project_name, 'input_parameter_names')))
def test_log_param_sets_job_run_data(self):
import json
self.foundations_job.project_name = self.project_name
self._set_job_running()
log_param(self.fake_key, self.fake_value)
expected_params = {self.fake_key: self.fake_value}
actual_params = json.loads(
self.redis_connection.get('jobs:{}:{}'.format(
self.job_id, 'parameters')))
self.assertEqual(expected_params, actual_params)
def test_log_param_sets_job_run_data_with_multiple_params(self):
import json
self.foundations_context.project_name.return_value = self.project_name
self._set_job_running()
log_param(self.fake_key, self.fake_value)
log_param(self.fake_key_two, self.fake_value_two)
expected_params = {
self.fake_key: self.fake_value,
self.fake_key_two: self.fake_value_two
}
actual_params = json.loads(
self.redis_connection.get('jobs:{}:{}'.format(
self.job_id, 'parameters')))
self.assertEqual(expected_params, actual_params)
