#!/usr/bin/python2
# Copyright 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittests for the lsbrelease_utils module."""

import unittest

import common
from autotest_lib.client.common_lib import lsbrelease_utils


# pylint: disable=line-too-long
_GUADO_MOBLAB_LSB_RELEASE_REDACTED = """
DEVICETYPE=CHROMEBOX
CHROMEOS_RELEASE_BUILDER_PATH=guado_moblab-release/R61-9641.0.0
GOOGLE_RELEASE=9641.0.0
CHROMEOS_DEVSERVER=
CHROMEOS_RELEASE_BOARD=guado_moblab
CHROMEOS_RELEASE_BUILD_NUMBER=9641
CHROMEOS_RELEASE_BRANCH_NUMBER=0
CHROMEOS_RELEASE_CHROME_MILESTONE=61
CHROMEOS_RELEASE_PATCH_NUMBER=0
CHROMEOS_RELEASE_TRACK=testimage-channel
CHROMEOS_RELEASE_DESCRIPTION=9641.0.0 (Official Build) dev-channel guado_moblab test
CHROMEOS_RELEASE_BUILD_TYPE=Official Build
CHROMEOS_RELEASE_NAME=Chrome OS
CHROMEOS_RELEASE_VERSION=9641.0.0
CHROMEOS_AUSERVER=https://tools.google.com/service/update2
"""

_LINK_LSB_RELEASE_REDACTED = """
DEVICETYPE=CHROMEBOOK
CHROMEOS_RELEASE_BUILDER_PATH=link-release/R61-9641.0.0
GOOGLE_RELEASE=9641.0.0
CHROMEOS_DEVSERVER=
CHROMEOS_RELEASE_BOARD=link
CHROMEOS_RELEASE_BUILD_NUMBER=9641
CHROMEOS_RELEASE_BRANCH_NUMBER=0
CHROMEOS_RELEASE_CHROME_MILESTONE=61
CHROMEOS_RELEASE_PATCH_NUMBER=0
CHROMEOS_RELEASE_TRACK=testimage-channel
CHROMEOS_RELEASE_DESCRIPTION=9641.0.0 (Official Build) dev-channel link test
CHROMEOS_RELEASE_BUILD_TYPE=Official Build
CHROMEOS_RELEASE_NAME=Chrome OS
CHROMEOS_RELEASE_VERSION=9641.0.0
CHROMEOS_AUSERVER=https://tools.google.com/service/update2
"""

_GALE_LSB_RELEASE_REDACTED = """
DEVICETYPE=OTHER
HWID_OVERRIDE=GALE DOGFOOD
CHROMEOS_RELEASE_BUILDER_PATH=gale-release/R61-9641.0.0
GOOGLE_RELEASE=9641.0.0
CHROMEOS_DEVSERVER=
CHROMEOS_RELEASE_BOARD=gale
CHROMEOS_RELEASE_BUILD_NUMBER=9641
CHROMEOS_RELEASE_BRANCH_NUMBER=0
CHROMEOS_RELEASE_CHROME_MILESTONE=61
CHROMEOS_RELEASE_PATCH_NUMBER=0
CHROMEOS_RELEASE_TRACK=testimage-channel
CHROMEOS_RELEASE_DESCRIPTION=9641.0.0 (Official Build) dev-channel gale test
CHROMEOS_RELEASE_BUILD_TYPE=Official Build
CHROMEOS_RELEASE_NAME=Chrome OS
CHROMEOS_RELEASE_VERSION=9641.0.0
CHROMEOS_AUSERVER=https://tools.google.com/service/update2
"""

# pylint: disable=line-too-long
_WHIRLWIND_LSB_RELEASE_REDACTED = """
DEVICETYPE=OTHER
HWID_OVERRIDE=WHIRLWIND DOGFOOD
CHROMEOS_RELEASE_BUILDER_PATH=whirlwind-release/R61-9641.0.0
GOOGLE_RELEASE=9641.0.0
CHROMEOS_DEVSERVER=
CHROMEOS_RELEASE_BOARD=whirlwind
CHROMEOS_RELEASE_BUILD_NUMBER=9641
CHROMEOS_RELEASE_BRANCH_NUMBER=0
CHROMEOS_RELEASE_CHROME_MILESTONE=61
CHROMEOS_RELEASE_PATCH_NUMBER=0
CHROMEOS_RELEASE_TRACK=testimage-channel
CHROMEOS_RELEASE_DESCRIPTION=9641.0.0 (Official Build) dev-channel whirlwind test
CHROMEOS_RELEASE_BUILD_TYPE=Official Build
CHROMEOS_RELEASE_NAME=Chrome OS
CHROMEOS_RELEASE_VERSION=9641.0.0
CHROMEOS_AUSERVER=https://tools.google.com/service/update2
"""


class LsbreleaseUtilsTestCase(unittest.TestCase):
    """Validates the helper free functions in lsbrelease_utils."""

    def test_is_jetstream_with_link_lsbrelease(self):
        """Test helper function."""
        self.assertFalse(lsbrelease_utils.is_jetstream(
            _LINK_LSB_RELEASE_REDACTED))

    def test_is_jetstream_with_moblab_lsbrelease(self):
        """Test helper function."""
        self.assertFalse(lsbrelease_utils.is_jetstream(
            _GUADO_MOBLAB_LSB_RELEASE_REDACTED))

    def test_is_jestream_with_gale_lsbrelease(self):
        """Test helper function."""
        self.assertTrue(lsbrelease_utils.is_jetstream(
            _GALE_LSB_RELEASE_REDACTED))

    def test_is_jestream_with_whirlwind_lsbrelease(self):
        """Test helper function."""
        self.assertTrue(lsbrelease_utils.is_jetstream(
            _WHIRLWIND_LSB_RELEASE_REDACTED))

    def test_is_moblab_with_empty_lsbrelease(self):
        """is_moblab correctly validates trivial lsb-release information."""
        self.assertFalse(lsbrelease_utils.is_moblab(''))

    def test_is_moblab_with_link_lsbrelease(self):
        """is_moblab correctly validates the contents from some other board."""
        self.assertFalse(lsbrelease_utils.is_moblab(
                _LINK_LSB_RELEASE_REDACTED))

    def test_is_moblab_with_moblab_lsbrelease(self):
        """is_moblab correctly validates the contents from a moblab device."""
        self.assertTrue(lsbrelease_utils.is_moblab(
                _GUADO_MOBLAB_LSB_RELEASE_REDACTED))

    def test_get_chromeos_release_version(self):
        """Test helper function."""
        result = lsbrelease_utils.get_chromeos_release_builder_path(
                _GUADO_MOBLAB_LSB_RELEASE_REDACTED)

        self.assertEqual(result, 'guado_moblab-release/R61-9641.0.0')


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