From 08c076b7945dec8257efc10db4c9b141a8cf7f8c Mon Sep 17 00:00:00 2001
From: maheshgudi
Date: Mon, 6 Nov 2017 19:35:13 +0530
Subject: Modify testcases wrt changes in assertion error output

---
 yaksh/tests/test_code_server.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'yaksh/tests/test_code_server.py')

diff --git a/yaksh/tests/test_code_server.py b/yaksh/tests/test_code_server.py
index 5f80f2d..1309624 100644
--- a/yaksh/tests/test_code_server.py
+++ b/yaksh/tests/test_code_server.py
@@ -61,7 +61,7 @@ class TestCodeServer(unittest.TestCase):
         # Then
         data = json.loads(result.get('result'))
         self.assertFalse(data['success'])
-        self.assertTrue('infinite loop' in data['error'][0])
+        self.assertTrue('infinite loop' in data['error'][0]['message'])
 
     def test_correct_answer(self):
         # Given
@@ -104,7 +104,7 @@ class TestCodeServer(unittest.TestCase):
         # Then
         data = json.loads(result.get('result'))
         self.assertFalse(data['success'])
-        self.assertTrue('AssertionError' in data['error'][0])
+        self.assertTrue('AssertionError' in data['error'][0]['exception'])
 
     def test_multiple_simultaneous_hits(self):
         # Given
@@ -143,7 +143,7 @@ class TestCodeServer(unittest.TestCase):
         for i in range(N):
             data = results.get()
             self.assertFalse(data['success'])
-            self.assertTrue('infinite loop' in data['error'][0])
+            self.assertTrue('infinite loop' in data['error'][0]['message'])
 
     def test_server_pool_status(self):
         # Given
-- 
cgit