import unittest

from pmock import *


class SystemError(Exception):

    def __init__(self, msg):
        self.msg = msg


class Greeting:

    def __init__(self, system):
        self._system = system

    def message(self):
        try:
            (hours, seconds) = self._system.time()
            if hours < 12:
                return "Good morning"
            elif hours >= 12 and hours < 19:
                return "Good afternoon"
            else:
                return "Good evening"
        except SystemError, err:
            self._system.log("time problem: %s" % err.msg)
            return "Good day"


class GreetingTest(MockTestCase):

    def setUp(self):
        self.system = self.mock()
        self.greeting = Greeting(self.system)

    def test_afternoon(self):
        self.system.expects(once()).time().will(return_value((12,10)))
        self.assertEqual(self.greeting.message(), "Good afternoon")

    def test_morning(self):
        self.system.expects(once()).time().will(return_value((6,50)))
        self.assertEqual(self.greeting.message(), "Good morning")

    def test_evening(self):
        self.system.expects(once()).time().will(return_value((19,50)))
        self.assertEqual(self.greeting.message(), "Good evening")

    def test_clock_failure(self):
        err_msg = "bzzz..malfunction"
        err = SystemError(err_msg)
        self.system.expects(once()).time().will(raise_exception(err))
        self.system.expects(once()).log(string_contains(err_msg))
        self.assertEqual(self.greeting.message(), "Good day")


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