"""
Tests for allmydata.monitor.
"""

from __future__ import unicode_literals
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from future.utils import PY2
if PY2:
    from builtins import filter, map, zip, ascii, chr, hex, input, next, oct, open, pow, round, super, bytes, dict, list, object, range, str, max, min  # noqa: F401

from twisted.trial import unittest

from allmydata.monitor import Monitor, OperationCancelledError


class MonitorTests(unittest.TestCase):
    """Tests for the Monitor class."""

    def test_cancellation(self):
        """The monitor can be cancelled."""
        m = Monitor()
        self.assertFalse(m.is_cancelled())
        m.raise_if_cancelled()
        m.cancel()
        self.assertTrue(m.is_cancelled())
        with self.assertRaises(OperationCancelledError):
            m.raise_if_cancelled()

    def test_status(self):
        """The monitor can have its status set."""
        m = Monitor()
        self.assertEqual(m.get_status(), None)
        m.set_status("discombobulated")
        self.assertEqual(m.get_status(), "discombobulated")

    def test_finish(self):
        """The monitor can finish."""
        m = Monitor()
        self.assertFalse(m.is_finished())
        d = m.when_done()
        self.assertNoResult(d)

        result = m.finish(300)
        self.assertEqual(result, 300)
        self.assertEqual(m.get_status(), 300)
        self.assertTrue(m.is_finished())

        d.addBoth(self.assertEqual, 300)
        return d