import os
import subprocess
import unittest
import yaml
from commontest import RBBin
from rdiff_backup import Globals


class ApiVersionTest(unittest.TestCase):
    """Test api versioning functionality"""

    def test_runtime_info_calling(self):
        """make sure that the info output can be read back as YAML when API is 201"""
        output = subprocess.check_output([RBBin, b'--api-version', b'201', b'info'])
        out_info = yaml.safe_load(output)

        Globals.api_version['actual'] = 201
        info = Globals.get_runtime_info()

        # because the current test will have a different call than rdiff-backup itself
        # we can't compare certain keys
        self.assertIn('exec', out_info)
        self.assertIn('argv', out_info['exec'])
        out_info['exec'].pop('argv')
        info['exec'].pop('argv')
        # info['python']['executable'] could also be different but I think that
        # our test environments make sure that it doesn't happen, unless Windows
        if os.name == "nt":
            info['python']['executable'] = info['python']['executable'].lower()
            out_info['python']['executable'] = \
                out_info['python']['executable'].lower()
        self.assertEqual(info, out_info)

    def test_default_actual_api(self):
        """validate that the default version is the actual one or the one explicitly set"""
        output = subprocess.check_output([RBBin, b"info"])
        api_version = yaml.safe_load(output)['exec']['api_version']
        self.assertEqual(Globals.get_api_version(), api_version['default'])
        api_param = os.fsencode(str(api_version['max']))
        output = subprocess.check_output([RBBin, b'--api-version', api_param, b'info'])
        out_info = yaml.safe_load(output)
        self.assertEqual(out_info['exec']['api_version']['actual'], api_version['max'])

    def test_debug_output(self):
        """we use verbosity 9 only to cover debug functions in logging"""
        output = subprocess.check_output([RBBin, b"-v", b"9",
                                          b"--terminal-verbosity", b"9",
                                          b"info"])
        self.assertIn(b"DEBUG: Runtime information =>", output)

    def test_env_var(self):
        """Test usage of the environment variable"""
        # Python under Windows needs the variable SYSTEMROOT
        # or it fails with a randomization init error, hence we copy the
        # environment and extend it instead of creating a new one.
        local_env = os.environ.copy()
        local_env["RDIFF_BACKUP_API_VERSION"] = "{min: 111, max: 999}"
        output = subprocess.check_output([RBBin, b"info"], env=local_env)
        api_version = yaml.safe_load(output)['exec']['api_version']
        self.assertEqual(api_version['min'], 111)
        self.assertEqual(api_version['max'], 999)
        # make sure the untouched variables are really untouched
        self.assertEqual(Globals.get_api_version(), api_version['default'])


if __name__ == "__main__":
    unittest.main()