BlitzGateway.connect: test raiseOnError

snoopycrimecop · Apr 8, 2020 · d68e639 · d68e639
1 parent 7080f9f
commit d68e639
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/test/unit/test_gateway.py b/test/unit/test_gateway.py
@@ -360,3 +360,19 @@ def test_simple_marshal_not_tiled(self, wrapped_image):
         data = wrapped_image.simpleMarshal(xtra={'tiled': True})
         self.assert_data(data)
         assert data['tiled'] is False
+
+
+class TestBlitzGatewayConnect(object):
+
+    def test_connect_default(self):
+        conn = BlitzGateway(
+            host='this.host.does.not.exist.example.org',
+            username='username', passwd='passwd')
+        assert conn.connect() == False
+
+    def test_connect_raise(self):
+        conn = BlitzGateway(
+            host='this.host.does.not.exist.example.org',
+            username='username', passwd='passwd')
+        with pytest.raises(Ice.DNSException):
+            conn.connect(raiseOnError=True)