init CTFd source

tests/utils/test_updates.py

@@ -0,0 +1,104 @@
+from unittest.mock import Mock, patch
+
+import requests
+
+from CTFd.utils import get_config, set_config
+from CTFd.utils.updates import update_check
+from tests.helpers import create_ctfd, destroy_ctfd, login_as_user
+
+
+def test_update_check_is_called():
+    """Update checks happen on start"""
+    app = create_ctfd()
+    with app.app_context():
+        assert get_config("version_latest") is None
+
+
+@patch.object(requests, "get")
+def test_update_check_identifies_update(fake_get_request):
+    """Update checks properly identify new versions"""
+    app = create_ctfd()
+    with app.app_context():
+        app.config["UPDATE_CHECK"] = True
+        fake_response = Mock()
+        fake_get_request.return_value = fake_response
+        fake_response.json = lambda: {
+            "resource": {
+                "download_url": "https://api.github.com/repos/CTFd/CTFd/zipball/9.9.9",
+                "html_url": "https://github.com/CTFd/CTFd/releases/tag/9.9.9",
+                "id": 12,
+                "latest": True,
+                "next": 1542212248,
+                "prerelease": False,
+                "published_at": "Wed, 25 Oct 2017 19:39:42 -0000",
+                "tag": "9.9.9",
+            }
+        }
+        update_check()
+        assert (
+            get_config("version_latest")
+            == "https://github.com/CTFd/CTFd/releases/tag/9.9.9"
+        )
+        assert get_config("next_update_check") == 1542212248
+    destroy_ctfd(app)
+
+
+def test_update_check_notifies_user():
+    """If an update is available, admin users are notified in the panel"""
+    app = create_ctfd()
+    with app.app_context():
+        app.config["UPDATE_CHECK"] = True
+        set_config("version_latest", "https://github.com/CTFd/CTFd/releases/tag/9.9.9")
+        client = login_as_user(app, name="admin", password="password")
+
+        r = client.get("/admin/config")
+        assert r.status_code == 200
+
+        response = r.get_data(as_text=True)
+        assert "https://github.com/CTFd/CTFd/releases/tag/9.9.9" in response
+
+    destroy_ctfd(app)
+
+
+@patch.object(requests, "post")
+def test_update_check_ignores_downgrades(fake_post_request):
+    """Update checks do nothing on old or same versions"""
+    app = create_ctfd()
+    with app.app_context():
+        app.config["UPDATE_CHECK"] = True
+        fake_response = Mock()
+        fake_post_request.return_value = fake_response
+        fake_response.json = lambda: {
+            "resource": {
+                "html_url": "https://github.com/CTFd/CTFd/releases/tag/0.0.1",
+                "download_url": "https://api.github.com/repos/CTFd/CTFd/zipball/0.0.1",
+                "published_at": "Wed, 25 Oct 2017 19:39:42 -0000",
+                "tag": "0.0.1",
+                "prerelease": False,
+                "id": 6,
+                "latest": True,
+            }
+        }
+        update_check()
+        assert get_config("version_latest") is None
+
+        fake_response = Mock()
+        fake_post_request.return_value = fake_response
+        fake_response.json = lambda: {
+            "resource": {
+                "html_url": "https://github.com/CTFd/CTFd/releases/tag/{}".format(
+                    app.VERSION
+                ),
+                "download_url": "https://api.github.com/repos/CTFd/CTFd/zipball/{}".format(
+                    app.VERSION
+                ),
+                "published_at": "Wed, 25 Oct 2017 19:39:42 -0000",
+                "tag": "{}".format(app.VERSION),
+                "prerelease": False,
+                "id": 6,
+                "latest": True,
+            }
+        }
+        update_check()
+        assert get_config("version_latest") is None
+    destroy_ctfd(app)