diff options
author | Antonio Ojea <6450081+aojea@users.noreply.github.com> | 2019-01-27 10:23:26 +0100 |
---|---|---|
committer | Asif Saif Uddin <auvipy@gmail.com> | 2019-01-27 15:23:26 +0600 |
commit | 40e0ef54a0befc1330ac772ef1d3151001014639 (patch) | |
tree | c8bc3b4c840b0965eb5ad53900c5002056504dbf | |
parent | e45ea3ece36ee5e58eaa3421d49b77ad86fdb5e5 (diff) | |
download | py-amqp-40e0ef54a0befc1330ac772ef1d3151001014639.tar.gz |
Add unit test for SSLTransport _write function (#251)
Reference:
https://github.com/celery/py-amqp/issues/249
-rw-r--r-- | t/unit/test_transport.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/t/unit/test_transport.py b/t/unit/test_transport.py index 85bd301..eeeedb0 100644 --- a/t/unit/test_transport.py +++ b/t/unit/test_transport.py @@ -622,6 +622,27 @@ class test_SSLTransport: match=r'.*Server unexpectedly closed connection.*'): self.t._read(64) + def test_write_success(self): + self.t.sock = Mock(name='SSLSocket') + self.t.sock.write.return_value = 2 + self.t._write('foo') + self.t.sock.write.assert_called() + + def test_write_socket_closed(self): + self.t.sock = Mock(name='SSLSocket') + self.t.sock.write.return_value = '' + with pytest.raises(IOError, + match=r'.*Socket closed.*'): + self.t._write('foo') + + def test_write_ValueError(self): + self.t.sock = Mock(name='SSLSocket') + self.t.sock.write.return_value = 2 + self.t.sock.write.side_effect = ValueError("Some error") + with pytest.raises(IOError, + match=r'.*Socket closed.*'): + self.t._write('foo') + class test_TCPTransport: |