Issue #528: Record cluster extensions

mozilla · Jul 25, 2017 · a877715 · a877715
1 parent f68aaa6
commit a877715
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/atmo/clusters/models.py b/atmo/clusters/models.py
@@ -318,6 +318,8 @@ def extend(self, hours):
         self.lifetime_extension_count = models.F('lifetime_extension_count') + 1
         self.save()
 
+        Metric.record('cluster-extension')
+
     def deactivate(self):
         """Shutdown the cluster and update its status accordingly"""
         self.provisioner.stop(self.jobflow_id)

diff --git a/tests/clusters/test_models.py b/tests/clusters/test_models.py
@@ -72,6 +72,8 @@ def test_extend(client, user, cluster_factory):
     assert cluster.expires_at > original_expires_at
     assert cluster.expires_at == original_expires_at + timedelta(hours=3)
 
+    assert Metric.objects.get(key='cluster-extension').value == 1
+
 
 def test_metric_records(cluster_provisioner_mocks, cluster_factory):
     cluster = cluster_factory()