Generate graphs when data is updated.

1 files changed, 39 insertions, 0 deletions

diff --git a/rrd/tests/test_graphs.py b/rrd/tests/test_graphs.py
new file mode 100644
index 0000000..f2a4948
--- /dev/null
+++ b/rrd/tests/test_graphs.py
@@ -0,0 +1,39 @@
+import datetime
+import os
+
+from django.conf import settings
+from django.test import TestCase
+
+from .. import models
+
+
+class TestGraphs(TestCase):
+    def setUp(self):
+        self.rrd_file = "test.rrd"
+        try:
+            os.remove(os.path.join(settings.RRD_DB_PATH, self.rrd_file))
+        except FileNotFoundError:
+            pass
+
+    def tearDown(self):
+        try:
+            os.remove(os.path.join(settings.RRD_DB_PATH, self.rrd_file))
+        except FileNotFoundError:
+            pass
+
+    def test_generate_at_ds_update(self):
+        ds = models.DataSource.objects.create(
+            topic="test",
+            path=self.rrd_file,
+        )
+        graph = models.Graph.objects.create(
+            title="Test Graph",
+            path="test/test.png",
+        )
+        graph.data_sources.add(ds)
+        graph.save()
+        now = datetime.datetime.now()
+        ts = int(now.timestamp())
+        ds.update(ts, 10)
+        stat = os.stat(os.path.join(settings.RRD_GRAPH_PATH, "test/test.png"))
+        self.assertGreaterEqual(stat.st_mtime, now.timestamp())