import random
from .ut_utils import ForgeTestCase
from forge.exceptions import UnexpectedCall

class Obj(object):
    def f(self, a, b, c):
        raise NotImplementedError() # pragma: no cover
    def g(self, value):
        raise NotImplementedError() # pragma: no cover

class WheneverTest(ForgeTestCase):
    def setUp(self):
        super(WheneverTest, self).setUp()
        self.obj = self.forge.create_mock(Obj)
    def test__whenever_verifies_arguments(self):
        self.obj.f(1, 2, 3).whenever().and_return(2)
        self.forge.replay()
        self.assertEqual(self.obj.f(1, 2, 3), 2)
        with self.assertRaises(UnexpectedCall):
            self.obj.f(1, 2, 4)
        self.assertEqual(self.obj.f(1, 2, 3), 2)
    def test__whenever_accepts_zero_times(self):
        self.obj.f(1, 2, 3).whenever().and_return(2)
        self.forge.replay()
    def test__whenever_and_raise(self):
        e = Exception()
        self.obj.f(1, 2, 3).whenever().and_raise(e)
        self.forge.replay()
        with self.assertRaises(Exception) as caught:
            self.obj.f(1, 2, 3)
        self.assertIs(caught.exception, e)
    def test__whenever_with_multiple_groups(self):
        call = self.obj.f(1, 2, 3)
        self.obj.f(1, 1, 1).and_return(1)
        with self.forge.any_order():
            self.obj.f(2, 2, 2).and_return(2)
        self.obj.f(3, 3, 3).and_return(3)
        call.whenever().and_return("hey")
        self.forge.replay()
        for i in range(3):
            self.assertEqual("hey", self.obj.f(1, 2, 3))
        self.assertEqual(1, self.obj.f(1, 1, 1))
        self.assertEqual("hey", self.obj.f(1, 2, 3))
        self.assertEqual(2, self.obj.f(2, 2, 2))
        self.assertEqual(3, self.obj.f(3, 3, 3))
        self.assertEqual("hey", self.obj.f(1, 2, 3))
    def test__whenever_patterns(self):
        values = list(range(10))
        for value in values:
            self.obj.g(value).whenever().and_return(value*2)
        self.forge.replay()
        for i in range(3):
            random.shuffle(values)
            for value in values:
                self.assertEqual(self.obj.g(value), value * 2)
        with self.assertRaises(UnexpectedCall):
            self.obj.g(max(values)+1)
    def test__whenever_when_syntax(self):
        self.obj.g.when(1).then_return(2)
        self.forge.replay()
        self.assertEqual(self.obj.g(1), 2)
        self.assertEqual(self.obj.g(1), 2)
        self.assertEqual(self.obj.g(1), 2)
        with self.assertRaises(UnexpectedCall):
            self.obj.g(2)
    def test__whenever_when_syntax_disabled_in_replay(self):
        self.forge.replay()
        self.obj.g # make sure the stub exists
        with self.assertRaises(AttributeError):
            self.obj.g.when
    def test__whenever_when_syntax_in_group(self):
        with self.forge.ordered():
            self.obj.f(1, 2, 3).and_return(42)
            self.obj.g.when(1).then_return(2)
        with self.forge.ordered():
            self.obj.f(3, 2, 1).and_return(24)
            self.obj.g.when(1).then_return(3)
        self.forge.replay()
        self.assertEqual(self.obj.g(1), 2)
        self.assertEqual(self.obj.g(1), 2)
        self.assertEqual(self.obj.f(1, 2, 3), 42)

        self.assertEqual(self.obj.g(1), 3)
        self.assertEqual(self.obj.g(1), 3)
        self.assertEqual(self.obj.f(3, 2, 1), 24)
    def test__whenever_applies_only_in_group(self):
        result = object()
        with self.forge.ordered():
            self.obj.f(1, 1, 1)
            with self.forge.any_order():
                self.obj.g(1).whenever().and_return(result)
                self.obj.f(2, 2, 2)
            self.obj.f(3, 3, 3)
        self.forge.replay()
        with self.assertRaises(UnexpectedCall):
            self.obj.g(1)
        self.obj.f(1, 1, 1)
        self.assertIs(result, self.obj.g(1))
        self.obj.f(2, 2, 2)
        with self.assertRaises(UnexpectedCall):
            # once f(2, 2, 2) is called, the group is cleared and the whenever() no longer applies
            self.obj.g(1)
        self.obj.f(3, 3, 3)