salt/tests/unit/utils/test_iam.py

# -*- coding: utf-8 -*-
'''
Tests for salt.utils.iam
'''
# Import Python libs
from __future__ import absolute_import, print_function, unicode_literals

import salt.utils.iam as iam
from salt.ext import six
from tests.support.unit import TestCase, skipIf


class IamTestCase(TestCase):

    @skipIf(six.PY3, 'Only needs to run on Python 2')
    def test__convert_key_to_str(self):
        '''
        Makes sure that unicode string is converted to a str type on PY2
        '''
        key = 'foo'
        expected = key.encode('utf-8')
        self.assertEqual(iam._convert_key_to_str(key), expected)