From b82e2ad1c507c1b0e3dbf3fdfd53d2ee5e2ccddc Mon Sep 17 00:00:00 2001
From: Itamar Turner-Trauring <itamar@itamarst.org>
Date: Fri, 7 Aug 2020 13:26:44 -0400
Subject: [PATCH] Tests for allmydata.monitor.
---
src/allmydata/test/test_monitor.py | 52 ++++++++++++++++++++++++++++++
src/allmydata/util/_python3.py | 1 +
2 files changed, 53 insertions(+)
create mode 100644 src/allmydata/test/test_monitor.py
diff --git a/src/allmydata/test/test_monitor.py b/src/allmydata/test/test_monitor.py
new file mode 100644
index 000000000..7010da73a
--- /dev/null
+++ b/src/allmydata/test/test_monitor.py
@@ -0,0 +1,52 @@
+"""
+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
diff --git a/src/allmydata/util/_python3.py b/src/allmydata/util/_python3.py
index 2b18eaa1d..f8260078f 100644
--- a/src/allmydata/util/_python3.py
+++ b/src/allmydata/util/_python3.py
@@ -61,6 +61,7 @@ PORTED_TEST_MODULES = [
"allmydata.test.test_humanreadable",
"allmydata.test.test_iputil",
"allmydata.test.test_log",
+ "allmydata.test.test_monitor",
"allmydata.test.test_netstring",
"allmydata.test.test_observer",
"allmydata.test.test_pipeline",