salt/tests/unit/pillar/test_csvpillar.py

# -*- coding: utf-8 -*-
'''test for pillar csvpillar.py'''

# Import python libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Testing libs
from tests.support.unit import TestCase
from tests.support.mock import patch, mock_open


# Import Salt Libs
import salt.pillar.csvpillar as csvpillar


class CSVPillarTestCase(TestCase):
    def test_001_load_utf8_csv(self):
        fake_csv = "id,foo,bar\r\nminion1,foo1,bar1"
        fake_dict = {'id': 'minion1', 'foo': 'foo1', 'bar': 'bar1'}
        fopen_mock = mock_open(fake_csv)
        with patch('salt.utils.files.fopen', fopen_mock):
            result = csvpillar.ext_pillar(mid='minion1', pillar=None,
                                          path="/fake/path/file.csv", idkey="id", namespace=None)
            self.assertDictEqual(fake_dict, result)

    def test_002_load_utf8_csv_namespc(self):
        fake_csv = "id,foo,bar\r\nminion1,foo1,bar1"
        fake_dict = {'baz': {'id': 'minion1', 'foo': 'foo1', 'bar': 'bar1'}}
        fopen_mock = mock_open(fake_csv)
        with patch('salt.utils.files.fopen', fopen_mock):
            result = csvpillar.ext_pillar(mid='minion1', pillar=None,
                                          path="/fake/path/file.csv", idkey="id", namespace='baz')
            self.assertDictEqual(fake_dict, result)