def test_handler_b64(self, content_mock):
plain = u'Hello, world'
encoded = base64.b64encode(plain.encode('utf8'))
content_mock.return_value = plain
out = handler(u'base64:file://tmp/test')
self.assertEqual(encoded, out)
def test_handler_parameterized_b64(self, content_mock, codec_mock):
result = mock.Mock()
codec_mock.return_value = result
out = handler(u'parameterized-b64:file://tmp/test')
codec_mock.assert_called_once_with(content_mock.return_value, True)
self.assertEqual(result, out)
def test_handler_json(self, content_mock, codec_mock):
result = mock.Mock()
codec_mock.return_value = result
out = handler(u'json:file://tmp/test')
codec_mock.assert_called_once_with(content_mock.return_value,
parameterized=False)
self.assertEqual(result, out)
def test_handler_parameterized_b64(self):
expected = 'Regular text {{Psuedo::Var}} {{RegVar}} {{Non-Var}}'
processed = {
'Fn::Base64': {
'Fn::Join': [
'',
[
'Regular text ', {
'Ref': 'Psuedo::Var'
}, ' ', {
'Ref': 'RegVar'
}, ' {{Non-Var}}'
]
]
}
}
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value=expected):
out = handler('parameterized-b64:file://tmp/test')
self.assertEqual(troposphere.Base64, type(out))
self.assertEqual(processed, out.data)
def test_unknown_codec(self, _):
with self.assertRaises(KeyError):
handler(u'bad:file://tmp/test')
def test_handler_plain(self, _):
out = handler(u'plain:file://tmp/test')
self.assertEqual(u'Hello, world', out)
def test_file_loaded(self, content_mock):
handler(u'plain:file://tmp/test')
content_mock.assert_called_with(u'file://tmp/test')
def test_unknown_codec(self):
expected = 'Hello, world'
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value=expected):
with self.assertRaises(KeyError):
handler('bad:file://tmp/test')
def test_handler_parameterized_b64(self):
expected = 'Hello, world'
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value=expected):
out = handler('parameterized-b64:file://tmp/test')
self.assertEqual(troposphere.Base64, type(out))
def test_handler_b64(self):
expected = 'Hello, world'
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value=expected):
out = handler('base64:file://tmp/test')
self.assertEqual(expected, base64.b64decode(out))
def test_handler_plain(self):
expected = 'Hello, world'
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value=expected):
out = handler('plain:file://tmp/test')
self.assertEqual(expected, out)
def test_file_loaded(self):
with mock.patch('stacker.lookups.handlers.file.read_value_from_path',
return_value='') as amock:
handler('plain:file://tmp/test')
amock.assert_called_with('file://tmp/test')
