Test ib3.Bot and ib3.connection.SSL (#8)

Add unit tests for basic behaviors of ib3.Bot and ib3.connection.SSL

Author: bd808

ib3/tests/test_bot.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of IRC Bot Behavior Bundle (IB3)
+# Copyright (C) 2017 Bryan Davis and contributors
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option)
+# any later version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+# more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program.  If not, see <http://www.gnu.org/licenses/>.
+
+# Dummy test file that really tests nothing other than that imports succeed
+
+import irc.client
+import jaraco.stream
+
+import ib3
+
+
+def test_construct_sets_lenient_decoding():
+    bot = ib3.Bot(
+        server_list=[('localhost', '9999')],
+        realname='ib3test',
+        nickname='ib3test',
+    )
+    assert len(bot.server_list) == 1
+    assert jaraco.stream.buffer.LenientDecodingLineBuffer.errors == 'replace'
+    assert irc.client.ServerConnection.buffer_class == \
+        jaraco.stream.buffer.LenientDecodingLineBuffer

ib3/tests/test_connection.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of IRC Bot Behavior Bundle (IB3)
+# Copyright (C) 2017 Bryan Davis and contributors
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option)
+# any later version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+# more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program.  If not, see <http://www.gnu.org/licenses/>.
+
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
+import ssl
+
+import ib3
+import ib3.connection
+
+
+class SSLBot(ib3.connection.SSL, ib3.Bot):
+    pass
+
+
+@mock.patch('irc.bot.SingleServerIRCBot.__init__')
+@mock.patch('irc.connection.Factory')
+def test_ssl(conn_factory, mock_init):
+    bot = SSLBot(
+        server_list=[('localhost', '9999')],
+        realname='ib3test',
+        nickname='ib3test',
+    )
+    assert isinstance(bot, ib3.connection.SSL)
+    assert isinstance(bot, ib3.Bot)
+    conn_factory.assert_called_once_with(wrapper=ssl.wrap_socket)
+    args, kwargs = mock_init.call_args
+    assert kwargs['connect_factory'] is conn_factory.return_value

setup.py

@@ -12,6 +12,7 @@
 
 tests_require = [
     'coverage',
+    "mock; python_version<'3.3'",
     'nose>=1.0',
 ]
 

tox.ini

@@ -8,6 +8,7 @@ deps=
     -r{toxinidir}/requirements.txt
     coverage
     nose
+    py27: mock
 setenv=
     PYTHONDONTWRITEBYTECODE=true
 commands=