# Copyright 2022 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for server/cros/chrome_sideloader.py."""

import base64
import unittest
from unittest import mock

from autotest_lib.server.cros import chrome_sideloader


class TestGetTastExpr(unittest.TestCase):
    """Test GetTastExpr in ChromeSideloader library."""

    class MockHost:
        """
        Mock class for host
        """

        def path_exists(self, path):
            """
            Determine if path exists on the remote machine.

            @param path: path to check

            @return: bool(path exists)

            """
            return True

        def get_file(self, src, dst, delete_dest):
            """
            Retrieve a file from the host.

            @param source: Remote file path (directory, file or list).
            @param dest: Local file path (directory, file or list).
            @param delete_dest: Delete files in remote path that are not in local
            path.

            """
            pass

    def testTastExpr(self):
        tast_expr = "lacros.Basic"
        args_dict = {'tast_expr': tast_expr}
        self.assertEqual(chrome_sideloader.get_tast_expr(args_dict), tast_expr)

    def testEmptyArgsDict(self):
        args_dict = {}
        with self.assertRaises(Exception):
            chrome_sideloader.get_tast_expr(args_dict)

    def testTastExprB64(self):
        tast_expr = '''("group:mainline" && !informational)'''
        tast_expr_b64 = _base64_encode_str(tast_expr)
        args_dict = {'tast_expr_b64': tast_expr_b64}
        self.assertEqual(chrome_sideloader.get_tast_expr(args_dict), tast_expr)

    def testTastExprB64Corrupted(self):
        tast_expr = '''("group:mainline" && !informational)'''
        tast_expr_b64 = _base64_encode_str(tast_expr)
        # remove last character to corrupt the encoding
        tast_expr_b64 = tast_expr_b64[:-1]
        args_dict = {'tast_expr_b64': tast_expr_b64}
        with self.assertRaises(Exception):
            chrome_sideloader.get_tast_expr(args_dict)

    def testTastFileWithKey(self):
        read_data = '{"default": "(\\"group:mainline\\" && !informational)"}'
        file_mock = mock.mock_open(read_data=read_data)
        args_dict = {
            'tast_expr_file': 'mocked_file',
            'tast_expr_key': 'default'
        }
        with mock.patch('builtins.open', file_mock),\
                mock.patch('os.stat'),\
                mock.patch('os.chmod'):
            expr = chrome_sideloader.get_tast_expr_from_file(
                TestGetTastExpr.MockHost(), args_dict, 'mock/path/')
            self.assertEqual('("group:mainline" && !informational)', expr)

    def testVersionSkew(self):
        self.assertTrue(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "100.0.1000.0", "100.0.1000.0"))
        self.assertTrue(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "100.0.1000.0", "100.0.1000.9"))
        self.assertTrue(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "100.0.9999.0", "100.0.1000.0"))
        self.assertTrue(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "102.0.1000.0", "100.0.1000.0"))
        self.assertFalse(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "103.0.1000.0", "100.0.1000.0"))
        self.assertFalse(
                chrome_sideloader.is_lacros_version_skew_valid(
                        "99.0.1000.0", "100.0.9999.0"))


def _base64_encode_str(s):
    return base64.b64encode(s.encode('utf-8')).decode('ascii')


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