From 24d682860fc1bcb5b1dc9d9472c3b9c64ab7f515 Mon Sep 17 00:00:00 2001
From: Kevin Froman <beardog@firemail.cc>
Date: Fri, 27 Sep 2019 10:42:35 -0500
Subject: [PATCH] fix ui test

---
 onionr/runtests/__init__.py | 2 +-
 onionr/runtests/uicheck.py  | 8 +++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/onionr/runtests/__init__.py b/onionr/runtests/__init__.py
index abc42f78..f98cfcb7 100644
--- a/onionr/runtests/__init__.py
+++ b/onionr/runtests/__init__.py
@@ -37,6 +37,6 @@ class OnionrRunTestManager:
                 last = i
                 i(self)
                 logger.info(last.__name__ + " passed")
-        except AssertionError:
+        except ValueError:
             logger.error(last.__name__ + ' failed')
     
\ No newline at end of file
diff --git a/onionr/runtests/uicheck.py b/onionr/runtests/uicheck.py
index 883f847d..530d1519 100644
--- a/onionr/runtests/uicheck.py
+++ b/onionr/runtests/uicheck.py
@@ -1,3 +1,9 @@
 from onionrutils import localcommand
 def check_ui(test_manager):
-    if not 'onionr' in localcommand.local_command('/mail/').lower(): raise AssertionError
\ No newline at end of file
+    endpoints = ['/', '/mail/', '/friends/', '/board/']
+    for point in endpoints:
+        result = localcommand.local_command(point)
+        if not result: raise ValueError
+        result = result.lower()
+        if not 'onionr' in result and not 'Onionr' in result:
+            raise ValueError
\ No newline at end of file