fix web UI dashboard (#33)

mars-project · Dec 15, 2018 · efd0863 · efd0863
1 parent e186b3c
commit efd0863
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/mars/web/dashboard.py b/mars/web/dashboard.py
@@ -12,12 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .server import register_ui_handler, get_jinja_env
+from .server import register_ui_handler, get_jinja_env, MarsWebAPI
 
 
-def dashboard(web_api, doc):
+def dashboard(scheduler_ip, doc):
     doc.title = 'Mars UI'
 
+    web_api = MarsWebAPI(scheduler_ip)
     infos = web_api.get_schedulers_info()
     doc.template_variables['infos'] = infos
     jinja_env = get_jinja_env()

diff --git a/mars/web/tests/test_api.py b/mars/web/tests/test_api.py
@@ -166,3 +166,14 @@ def testApi(self):
             c = a.dot(b)
             value = sess.run(c, timeout=120)
             assert_array_equal(value, va.dot(vb))
+
+            # check web UI requests
+            res = requests.get(service_ep)
+            self.assertEqual(res.status_code, 200)
+
+            res = requests.get(service_ep + '/task')
+            self.assertEqual(res.status_code, 200)
+
+            res = requests.get(service_ep + '/worker')
+            self.assertEqual(res.status_code, 200)
+