tutor/tests/test_config.py

import unittest
import unittest.mock
import tempfile

from tutor.commands import config as tutor_config
from tutor import env


class ConfigTests(unittest.TestCase):
    def setUp(self):
        # This is necessary to avoid cached mocks
        env.Renderer.reset()

    def test_merge(self):
        config = {}
        defaults = tutor_config.load_defaults()
        with unittest.mock.patch.object(
            tutor_config.utils, "random_string", return_value="abcd"
        ):
            tutor_config.merge(config, defaults)

        self.assertEqual("abcd", config["MYSQL_ROOT_PASSWORD"])
    
    @unittest.mock.patch.object(tutor_config.fmt, "echo")
    def test_save_twice(self, mock_echo):
        with tempfile.TemporaryDirectory() as root:
            tutor_config.save(root, silent=True)
            config1 = tutor_config.load_user(root)

            tutor_config.save(root, silent=True)
            config2 = tutor_config.load_user(root)

        self.assertEqual(config1, config2)

    @unittest.mock.patch.object(tutor_config.fmt, "echo")
    def test_removed_entry_is_added_on_save(self, mock_echo):
        with tempfile.TemporaryDirectory() as root:
            with unittest.mock.patch.object(
                tutor_config.utils, "random_string"
            ) as mock_random_string:
                mock_random_string.return_value = "abcd"
                defaults = tutor_config.load_defaults()
                config1 = tutor_config.load_current(root, defaults)
                password1 = config1["MYSQL_ROOT_PASSWORD"]

                config1.pop("MYSQL_ROOT_PASSWORD")
                tutor_config.save_config(root, config1)

                mock_random_string.return_value = "efgh"
                defaults = tutor_config.load_defaults()
                config2 = tutor_config.load_current(root, defaults)
                password2 = config2["MYSQL_ROOT_PASSWORD"]

        self.assertEqual("abcd", password1)
        self.assertEqual("efgh", password2)