diff options
Diffstat (limited to 'test/units/parsing/test_dataloader.py')
-rw-r--r-- | test/units/parsing/test_dataloader.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/test/units/parsing/test_dataloader.py b/test/units/parsing/test_dataloader.py index a7f8b1d2..9ec49a8d 100644 --- a/test/units/parsing/test_dataloader.py +++ b/test/units/parsing/test_dataloader.py @@ -25,7 +25,8 @@ from units.compat import unittest from unittest.mock import patch, mock_open from ansible.errors import AnsibleParserError, yaml_strings, AnsibleFileNotFound from ansible.parsing.vault import AnsibleVaultError -from ansible.module_utils.common.text.converters import to_text +from ansible.module_utils._text import to_text +from ansible.module_utils.six import PY3 from units.mock.vault_helper import TextVaultSecret from ansible.parsing.dataloader import DataLoader @@ -91,11 +92,11 @@ class TestDataLoader(unittest.TestCase): - { role: 'testrole' } testrole/tasks/main.yml: - - include_tasks: "include1.yml" + - include: "include1.yml" static: no testrole/tasks/include1.yml: - - include_tasks: include2.yml + - include: include2.yml static: no testrole/tasks/include2.yml: @@ -228,7 +229,11 @@ class TestDataLoaderWithVault(unittest.TestCase): 3135306561356164310a343937653834643433343734653137383339323330626437313562306630 3035 """ + if PY3: + builtins_name = 'builtins' + else: + builtins_name = '__builtin__' - with patch('builtins.open', mock_open(read_data=vaulted_data.encode('utf-8'))): + with patch(builtins_name + '.open', mock_open(read_data=vaulted_data.encode('utf-8'))): output = self._loader.load_from_file('dummy_vault.txt') self.assertEqual(output, dict(foo='bar')) |