def test_get_validated_value_string_rewrap_unsafe(self):
attribute = FieldAttribute(isa='string')
value = AssibleUnsafeText(u'bar')
templar = Templar(None)
bsc = self.ClassUnderTest()
result = bsc.get_validated_value('foo', attribute, value, templar)
self.assertIsInstance(result, AssibleUnsafeText)
self.assertEqual(result, AssibleUnsafeText(u'bar'))
def test_encode_decode_unsafe():
data = {
'key_value': AssibleUnsafeText(u'{#NOTACOMMENT#}'),
'list': [AssibleUnsafeText(u'{#NOTACOMMENT#}')],
'list_dict': [{
'key_value': AssibleUnsafeText(u'{#NOTACOMMENT#}')
}]
}
json_expected = (
'{"key_value": {"__assible_unsafe": "{#NOTACOMMENT#}"}, '
'"list": [{"__assible_unsafe": "{#NOTACOMMENT#}"}], '
'"list_dict": [{"key_value": {"__assible_unsafe": "{#NOTACOMMENT#}"}}]}'
)
assert json.dumps(data,
cls=AssibleJSONEncoder,
preprocess_unsafe=True,
sort_keys=True) == json_expected
assert json.loads(json_expected, cls=AssibleJSONDecoder) == data
def test_unicode(self):
u_text = u'nöel'
unsafe_object = AssibleUnsafeText(u_text)
yaml_out = self._dump_string(unsafe_object, dumper=self.dumper)
stream = self._build_stream(yaml_out)
loader = self._loader(stream)
data_from_yaml = loader.get_single_data()
self.assertEqual(u_text, data_from_yaml)
def test_task_executor_run_clean_res(self):
te = TaskExecutor(None, MagicMock(), None, None, None, None, None,
None)
te._get_loop_items = MagicMock(return_value=[1])
te._run_loop = MagicMock(
return_value=[{
'unsafe_bytes': AssibleUnsafeBytes(b'{{ $bar }}'),
'unsafe_text': AssibleUnsafeText(u'{{ $bar }}'),
'bytes': b'bytes',
'text': u'text',
'int': 1,
}])
res = te.run()
data = res['results'][0]
self.assertIsInstance(data['unsafe_bytes'], AssibleUnsafeText)
self.assertIsInstance(data['unsafe_text'], AssibleUnsafeText)
self.assertIsInstance(data['bytes'], text_type)
self.assertIsInstance(data['text'], text_type)
self.assertIsInstance(data['int'], int)
def test_to_bytes_unsafe():
assert isinstance(to_bytes(AssibleUnsafeText(u'foo')), AssibleUnsafeBytes)
assert to_bytes(AssibleUnsafeText(u'foo')) == AssibleUnsafeBytes(b'foo')
def test_wrap_var_unsafe_text():
assert isinstance(wrap_var(AssibleUnsafeText(u'foo')), AssibleUnsafeText)
def test_AssibleUnsafeText():
assert isinstance(AssibleUnsafeText(u'foo'), AssibleUnsafe)