Merge pull request #1 from rakiru/code-review

Code review

1 files changed, 4 insertions, 1 deletions

diff --git a/asciifarm/client/connection.py b/asciifarm/client/connection.py
index 4eb0ffe..20f4964 100644
--- a/asciifarm/client/connection.py
+++ b/asciifarm/client/connection.py
@@ -10,6 +10,8 @@ class Connection:
             sockType = socket.AF_UNIX
         elif socketType == "inet":
             sockType = socket.AF_INET
+        else:
+            raise ValueError("Invalid socket type: %r" % (socketType,))
         self.sock = socket.socket(sockType, socket.SOCK_STREAM)
     
     def connect(self, address):
@@ -21,7 +23,8 @@ class Connection:
                 data = receive(self.sock)
             except Exception as err:
                 onError(err)
-            callback(data)
+            else:
+                callback(data)
     
     def send(self, message):
         send(self.sock, bytes(message, 'utf-8'))