Add tests to check that the UI is setup correctly

* Check that the links for OAuth2 logins via ORCID and GitHub are setup
  correctly.

2 files changed, 14 insertions, 0 deletions

diff --git a/test/requests/test_login_github.py b/test/requests/test_login_github.py
index 15bf18ae..1bf4f695 100644
--- a/test/requests/test_login_github.py
+++ b/test/requests/test_login_github.py
@@ -1,6 +1,7 @@
 import uuid
 import requests
 from time import sleep
+from wqflask import app
 from parameterized import parameterized
 from parametrized_test import ParametrizedTest
 
@@ -29,6 +30,12 @@ class TestLoginGithub(ParametrizedTest):
         super(TestLoginGithub, self).tearDown()
         self.es.delete(index="users", doc_type="local", id=uid)
 
+    def testLoginUrl(self):
+        login_button_text = '<a href="https://github.com/login/oauth/authorize?client_id=' + app.config.get("GITHUB_CLIENT_ID") + '&amp;client_secret=' + app.config.get("GITHUB_CLIENT_SECRET") + '" title="Login with GitHub" class="btn btn-info btn-group">Login with Github</a>'
+        result = requests.get(self.gn2_url+"/n/login")
+        index = result.content.find(login_button_text)
+        self.assertTrue(index >= 0, "Should have found `Login with Github` button")
+
     @parameterized.expand([
         ("1234", login_link_text, "Login should have failed with non-existing user")
         , (uid, logout_link_text, "Login should have been successful with existing user")
diff --git a/test/requests/test_login_orcid.py b/test/requests/test_login_orcid.py
index 6e40ece5..ea15642e 100644
--- a/test/requests/test_login_orcid.py
+++ b/test/requests/test_login_orcid.py
@@ -1,6 +1,7 @@
 import uuid
 import requests
 from time import sleep
+from wqflask import app
 from parameterized import parameterized
 from parametrized_test import ParametrizedTest
 
@@ -29,6 +30,12 @@ class TestLoginOrcid(ParametrizedTest):
         super(TestLoginOrcid, self).tearDown()
         self.es.delete(index="users", doc_type="local", id=uid)
 
+    def testLoginUrl(self):
+        login_button_text = 'a href="https://sandbox.orcid.org/oauth/authorize?response_type=code&amp;scope=/authenticate&amp;show_login=true&amp;client_id=' + app.config.get("ORCID_CLIENT_ID") + '&amp;client_secret=' + app.config.get("ORCID_CLIENT_SECRET") + '" title="Login with ORCID" class="btn btn-info btn-group">Login with ORCID</a>'
+        result = requests.get(self.gn2_url+"/n/login")
+        index = result.content.find(login_button_text)
+        self.assertTrue(index >= 0, "Should have found `Login with ORCID` button")
+
     @parameterized.expand([
         ("1234", login_link_text, "Login should have failed with non-existing user")
         , (uid, logout_link_text, "Login should have been successful with existing user")