# Copyright 2014 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 LOAS helper functions."""

import datetime

from chromite.lib import alerts
from chromite.lib import cros_test_lib
from chromite.lib import loas
from chromite.lib import partial_mock


class TestLoas(cros_test_lib.MockTestCase):
    """General tests for the LOAS module"""

    def setUp(self) -> None:
        self.rc_mock = self.StartPatcher(cros_test_lib.RunCommandMock())
        self.email_mock = self.PatchObject(alerts, "SendEmail")

        self.user = "foo"
        self.email = "some@email.com"
        self.loas = loas.Loas(self.user, self.email)

    def testCheckSuccess(self) -> None:
        """Verify Check() behavior when loas_check passes."""
        self.rc_mock.AddCmdResult(partial_mock.In("runloas"), returncode=0)
        self.loas.Check()

    def testCheckError(self) -> None:
        """Verify Check() behavior when loas_check fails."""
        self.rc_mock.AddCmdResult(partial_mock.In("runloas"), returncode=1)
        self.assertRaises(loas.LoasError, self.loas.Check)

    def testStatusError(self) -> None:
        """Verify that errors from gcertstatus result in an e-mail."""
        self.rc_mock.AddCmdResult(
            partial_mock.In("gcertstatus"),
            returncode=1,
            stderr="WARNING no LOAS2 certificate found",
        )
        self.loas.Status()
        self.assertEqual(self.email_mock.call_count, 1)

    def testStatusUpToDate(self) -> None:
        """Verify that up-to-date certs delay further checks for a while."""
        self.rc_mock.AddCmdResult(
            partial_mock.In("gcertstatus"),
            returncode=0,
            stderr="LOAS2 expires in 1234h",
        )

        # This should invoke gcertstatus.
        self.loas.Status()
        self.assertEqual(self.rc_mock.call_count, 1)

        # While this should return quickly.
        self.loas.Status()
        self.assertEqual(self.rc_mock.call_count, 1)

    def testStatusExpiresSoon(self) -> None:
        """Verify that expiring certs generate e-mails once a day."""
        self.rc_mock.AddCmdResult(
            partial_mock.In("gcertstatus"),
            returncode=1,
            stderr="  WARNING LOAS2 expires in 76h",
        )

        # This should invoke gcertstatus & send an e-mail.
        self.loas.Status()
        self.assertEqual(self.email_mock.call_count, 1)
        self.assertIn(
            "WARNING LOAS2", self.email_mock.call_args.kwargs["message"]
        )

        # While this should do nothing but return (only one e-mail a day).
        self.loas.Status()
        self.loas.Status()
        self.loas.Status()
        self.assertEqual(self.email_mock.call_count, 1)

        # Grub around in internal state to fast forward the clock by a day :/.
        self.loas.last_notification += datetime.timedelta(days=-1)

        # This should send out one e-mail.
        self.loas.Status()
        self.assertEqual(self.email_mock.call_count, 2)
        self.loas.Status()
        self.loas.Status()
        self.loas.Status()
        self.assertEqual(self.email_mock.call_count, 2)