From 16d78db72e499067feb99b62d4aa8b785693875c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C4=93teris=20Caune?= <cuu508@gmail.com>
Date: Thu, 8 Nov 2018 11:59:04 +0200
Subject: [PATCH] Refactoring and a testcase for channels=None

---
 hc/api/tests/test_update_check.py |  8 ++++++++
 hc/api/views.py                   | 21 ++++++++++-----------
 2 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/hc/api/tests/test_update_check.py b/hc/api/tests/test_update_check.py
index 07fc339f..d21ed8e0 100644
--- a/hc/api/tests/test_update_check.py
+++ b/hc/api/tests/test_update_check.py
@@ -151,3 +151,11 @@ class UpdateCheckTestCase(BaseTestCase):
 
         self.check.refresh_from_db()
         self.assertEqual(self.check.channel_set.count(), 0)
+
+    def test_it_rejects_non_string_channels_key(self):
+        r = self.post(self.check.code, {
+            "api_key": "X" * 32,
+            "channels": None
+        })
+
+        self.assertEqual(r.status_code, 400)
diff --git a/hc/api/views.py b/hc/api/views.py
index 28395111..7d1024fc 100644
--- a/hc/api/views.py
+++ b/hc/api/views.py
@@ -1,5 +1,4 @@
 from datetime import timedelta as td
-from uuid import UUID
 
 from django.conf import settings
 from django.core.exceptions import SuspiciousOperation
@@ -83,17 +82,17 @@ def _update(check, spec):
     if "channels" in spec:
         if spec["channels"] == "*":
             check.assign_all_channels()
-        else:
+        elif spec["channels"] == "":
             check.channel_set.clear()
-            if spec["channels"] is not None and spec["channels"] != "":
-                channels = []
-                for raw_channel in spec["channels"].split(","):
-                    try:
-                        channel = Channel.objects.get(code=UUID(raw_channel))
-                        channels.append(channel)
-                    except Channel.objects.model.DoesNotExist:
-                        raise SuspiciousOperation("One of the specified channels is missing")
-                check.channel_set.add(*channels)
+        else:
+            channels = []
+            for chunk in spec["channels"].split(","):
+                try:
+                    channel = Channel.objects.get(code=chunk)
+                    channels.append(channel)
+                except Channel.DoesNotExist:
+                    raise SuspiciousOperation("Invalid channel identifier")
+            check.channel_set.set(channels)
 
     return check