def test_smoke(self):
config = PlainBoxConfig()
self.assertIs(config.secure_id, Unset)
secure_id = "0123456789ABCDE"
config.secure_id = secure_id
self.assertEqual(config.secure_id, secure_id)
with self.assertRaises(ValueError):
config.secure_id = "bork"
self.assertEqual(config.secure_id, secure_id)
del config.secure_id
self.assertIs(config.secure_id, Unset)
def setUp(self):
self.provider = mock.Mock(spec_set=IProvider1, name='provider')
self.provider.namespace = "com.canonical.plainbox"
self.provider.job_list = []
self.provider.units_dir = None
self.provider.jobs_dir = None
self.provider.data_dir = None
self.provider.executable_list = []
self.provider.id_map = defaultdict(list)
self.provider.extra_PYTHONPATH = None
self.provider.CHECKBOX_SHARE = None
self.provider_loader = lambda: [self.provider]
self.config_loader = lambda: PlainBoxConfig()
def test_with_config_and_variables_not_in_environ(self):
env = {'bar': 'old-bar-value'}
# Setup a configuration object with values for baz.
# Note that baz is *not* declared in the job's environ line.
from plainbox.impl.applogic import PlainBoxConfig
config = PlainBoxConfig()
config.environment = {'baz': 'baz-value'}
# Ask the job to modify the environment
self.job.modify_execution_environment(env, self.session_dir,
self.checkbox_data_dir, config)
# bar from the old environment
self.assertEqual(env['bar'], 'old-bar-value')
# baz from the config environment
self.assertEqual(env['baz'], 'baz-value')
def proxy_test(self, environment, proxies):
test_environment = environment
test_proxies = proxies
test_config = PlainBoxConfig()
test_config.environment = test_environment
transport = SubmissionServiceTransport(
self.valid_url, self.valid_option_string)
dummy_data = BytesIO(b"some data to send")
requests.post.return_value = MagicMock(name='response')
requests.post.return_value.status_code = 200
requests.post.return_value.text = '{"id": 768}'
result = transport.send(dummy_data, config=test_config)
self.assertTrue(result)
requests.post.assert_called_with(
self.valid_url, data=dummy_data,
proxies=test_proxies)
def proxy_test(self, environment, proxies):
test_environment = environment
test_proxies = proxies
test_config = PlainBoxConfig()
test_config.environment = test_environment
transport = CertificationTransport(
self.valid_url, self.valid_option_string)
dummy_data = BytesIO(b"some data to send")
requests.post.return_value = MagicMock(name='response')
requests.post.return_value.status_code = 200
requests.post.return_value.text = '{"id": 768}'
result = transport.send(dummy_data, config=test_config)
self.assertTrue(result)
requests.post.assert_called_with(
self.valid_url, files={'data': dummy_data},
headers={'X_HARDWARE_ID': self.valid_secure_id},
proxies=test_proxies)
def test_with_config_and_environ_variables(self):
env = {
"PATH": "",
# foo is not defined in the environment
'bar': 'old-bar-value'
# froz is not defined in the environment
}
# Setup a configuration object with values for foo and bar
from plainbox.impl.applogic import PlainBoxConfig
config = PlainBoxConfig()
config.environment = {'foo': 'foo-value', 'bar': 'bar-value'}
# Ask the job to modify the environment
self.job.modify_execution_environment(env, self.session_dir,
self.checkbox_data_dir, config)
# foo got copied from the config
self.assertEqual(env['foo'], 'foo-value')
# bar from the old environment
self.assertEqual(env['bar'], 'old-bar-value')
# froz is still empty
self.assertNotIn('froz', env)
def setUp(self):
self.provider = mock.Mock()
self.config = PlainBoxConfig()
self.job_name = mock.Mock()
def setUp(self):
self.provider_list = mock.Mock()
self.config = PlainBoxConfig()
self.job_id = mock.Mock()
