From 98354ad1d1e3044b928f73b51b9408501adc1880 Mon Sep 17 00:00:00 2001 From: LV-426 Date: Sun, 28 Sep 2014 23:55:52 +0300 Subject: cleanup test_add_node_set --- src-py/test_rhizi_api.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'src-py') diff --git a/src-py/test_rhizi_api.py b/src-py/test_rhizi_api.py index bbea53bf..102d0cc3 100644 --- a/src-py/test_rhizi_api.py +++ b/src-py/test_rhizi_api.py @@ -4,6 +4,8 @@ import rhizi_api import json from rhizi_server import Config +from werkzeug.test import EnvironBuilder +from werkzeug.test import Client class TestRhiziAPI(unittest.TestCase): @@ -16,11 +18,13 @@ class TestRhiziAPI(unittest.TestCase): """ add node set test """ - node_map = { 'Skill': { 'name': 'kung-fu' } } - + node_map = { 'Skill': [{ 'name': 'kung-fu' }, { 'name': 'judo' }] } with rhizi_api.webapp.test_client() as c: - req = c.post('/add/node-set', data={'node_map':node_map}) - rz_data = json.loads(req.data)['data'] + req = c.post('/add/node-set', content_type='application/json', + data=json.dumps(dict(node_map=node_map))) + id_set = json.loads(req.data)['data'] + self.assertEqual(2, len(id_set)) + self.assertTrue(isinstance(id_set[0], int)) def test_load_node_non_existing(self): """ -- cgit v1.3.1