import Queue

from unittest import TestCase

from mock import Mock, patch, call

from fixture import TempIO

from rattail.filemon import actions

from rattail.filemon.config import Profile, ProfileAction

class TestAction(TestCase):

class TestInvokeAction(TestCase):

    def setUp(self):

        self.action = ProfileAction()

        self.action.action = Mock(return_value=None)

        self.action.retry_attempts = 6

        self.tmp = TempIO()

        self.file = self.tmp.putfile(u'file', u'')

    def test_action_which_succeeds_is_only_called_once(self):

        self.assertEqual(self.action.action.call_count, 3)

        self.assertTrue(time.time() - start >= 2.0)

    def test_action_which_fails_is_only_attempted_the_specified_number_of_times(self):

        self.action.action.side_effect = RuntimeError

        # Last attempt will not handle the exception; assert that as well.

        self.assertEqual(self.action.action.call_count, 6)

    def test_action_which_fails_then_succeeds_stops_retrying(self):

        # First 2 attempts fail, third succeeds.

        self.action.action.side_effect = [RuntimeError, RuntimeError, None]

        actions.invoke_action(self.action, self.file)