round durations to 2 digits

[GitLab/stricted-build/lineage_builder.git] / test.py

diff --git a/test.py b/test.py
index 8f9ddabfc82ec14864b3b778f2205ed34d68b54d..40121fa2a3560f76c6e943eb297159b92243b276 100644 (file)
--- a/test.py
+++ b/test.py
@@ -1,6 +1,9 @@
+import contextlib
 import datetime
 import os
+import sys
 import unittest
+
 import flask_migrate
 import flask_testing
 
@@ -14,10 +17,12 @@ class UsesApp(flask_testing.TestCase):
 
 class UsesModels(UsesApp):
     def setUp(self):
-        flask_migrate.upgrade(revision="head")
+        with open(os.devnull, "w") as f, contextlib.redirect_stderr(f):
+            flask_migrate.upgrade(revision="head")
 
     def tearDown(self):
-        flask_migrate.downgrade(revision="base")
+        with open(os.devnull, "w") as f, contextlib.redirect_stderr(f):
+            flask_migrate.downgrade(revision="base")
 
 class TestModels(UsesModels):
 
@@ -85,7 +90,6 @@ class TestWebhooks(UsesModels):
         assert build.build_runner.runner_name == "phenom-test"
         assert build.build_version == "lineage-15.1"
         assert build.build_device == "FAKE_TEST"
-        print(build.build_status)
         assert build.build_status == "success"
         assert build.build_duration == 81
 
@@ -109,5 +113,14 @@ class TestWeb(UsesModels):
         response = self.client.get("/")
         assert response.status_code == 200
 
+        response = self.client.get("/runners/")
+        assert response.status_code == 200
+
+        response = self.client.get("/runners/foobar")
+        assert response.status_code == 200
+
+        response = self.client.get("/?status=success&device=mako")
+        assert response.status_code == 200
+
 if __name__ == "__main__":
     unittest.main()