# pylint: disable=C0111

import unittest

from buildscripts.resmokelib.core.programs import _format_shell_vars


class ResmokeProgramsTestCase(unittest.TestCase):
    def test_format_shell_vars_with_dot(self):
        string_builder = []
        with_dot = {'a.b': 'c'}
        _format_shell_vars(string_builder, ['dummy_key'], with_dot)
        expected = ['dummy_key = new Object()', 'dummy_key["a.b"] = "c"']
        self.assertEqual(string_builder, expected)

        string_builder = []
        without_dot = {'a': {'b': 'c'}}
        _format_shell_vars(string_builder, ['dummy_key'], without_dot)
        expected = [
            'dummy_key = new Object()', 'dummy_key["a"] = new Object()', 'dummy_key["a"]["b"] = "c"'
        ]
        self.assertEqual(string_builder, expected)