From 13d271d3410a6b125eb1ac57d480c0173aa26407 Mon Sep 17 00:00:00 2001
From: "W. Trevor King" <wking@drexel.edu>
Date: Sat, 24 Mar 2012 16:07:11 -0400
Subject: [PATCH] Add return-value checking to AssuanClient response handling.

This makes calling code more Pythonic (catch is better than check ;).
---
 pyassuan/client.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/pyassuan/client.py b/pyassuan/client.py
index 1338288..7abe617 100644
--- a/pyassuan/client.py
+++ b/pyassuan/client.py
@@ -60,7 +60,7 @@ class AssuanClient (object):
         self.logger.info('S: {}'.format(response))
         return response
 
-    def make_request(self, request, response=True):
+    def make_request(self, request, response=True, expect=['OK']):
         rstring = str(request)
         self.logger.info('C: {}'.format(rstring))
         self.output.write(rstring)
@@ -70,9 +70,9 @@ class AssuanClient (object):
         except IOError:
             raise
         if response:
-            return self.get_responses(request=request)
+            return self.get_responses(request=request, expect=expect)
 
-    def get_responses(self, request=None):
+    def get_responses(self, request=None, expect=['OK']):
         responses = list(self.responses())
         if responses[-1].type == 'ERR':
             eresponse = responses[-1]
@@ -87,6 +87,8 @@ class AssuanClient (object):
                 error.request = request
             error.responses = responses
             raise error
+        if expect:
+            assert responses[-1].type in expect, [str(r) for r in responses]
         data = []
         for response in responses:
             if response.type == 'D':
-- 
2.26.2