From 44e715e6c9bed3243ac92e730924a19c571762ee Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Sat, 18 Aug 2018 10:34:57 +0900
Subject: [PATCH] Fix test_daemon

---
 tests/server/test_daemon.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/tests/server/test_daemon.py b/tests/server/test_daemon.py
index 2a905a9..712e02a 100644
--- a/tests/server/test_daemon.py
+++ b/tests/server/test_daemon.py
@@ -104,11 +104,14 @@ class ClientSessionGood(ClientSessionBase):
             assert request.args == args
             return JSONResponse(result, request_id)
         else:
-            assert isinstance(request, Batch)
-            for request, args in zip(request, args):
-                assert request.method == method
-                assert request.args == args
-            return JSONResponse(result, request_id)
+            batch = request
+            assert isinstance(batch, list)
+            request_ids = []
+            for payload, args in zip(batch, args):
+                assert payload['method'] == method
+                assert payload['params'] == args
+                request_ids.append(payload['id'])
+            return JSONResponse(result, request_ids)
 
 
 class ClientSessionBadAuth(ClientSessionBase):