Merge branch 'master' into feature/capture-requests

.gitignore

@@ -5,6 +5,7 @@ storage.db
 storage.json
 *.dca
 .eggs/
+.cache/
 
 # Documentation stuff
 docs/api/

.travis.yml

@@ -8,6 +8,8 @@ python:
   - '3.4'
   - '3.5'
   - '3.6'
-  - 'nightly'
+
+install:
+  - pip install -U pip setuptools pytest
 
 script: 'python setup.py test'

setup.cfg

@@ -0,0 +1,5 @@
+[aliases]
+test=pytest
+
+[tool:pytest]
+python_files = tests/*.py

setup.py

@@ -32,6 +32,8 @@ setup(
     install_requires=requirements,
     extras_require=extras_require,
     test_suite='tests',
+    setup_requires=['pytest-runner==2.11.1'],
+    tests_require=['pytest==3.1.3', 'pytest-benchmark==3.1.0a2'],
     classifiers=[
         'Development Status :: 4 - Beta',
         'License :: OSI Approved :: MIT License',

tests/test_reason.py → tests/api/reason.py

@@ -1,18 +1,18 @@
 from unittest import TestCase
 
+from tests.utils import APIClient
+
 
 class TestReason(TestCase):
     def test_set_unicode_reason(self):
-        from tests.utils import TestAPIClient
-        api = TestAPIClient()
+        api = APIClient()
         api.guilds_channels_modify(1, 2, 3, reason=u'yo \U0001F4BF test')
 
         _, kwargs = api.http.calls[0]
         self.assertEqual(kwargs['headers']['X-Audit-Log-Reason'], 'yo%20%F0%9F%92%BF%20test')
 
     def test_null_reason(self):
-        from tests.utils import TestAPIClient
-        api = TestAPIClient()
+        api = APIClient()
         api.guilds_channels_modify(1, 2, 3, reason=None)
 
         _, kwargs = api.http.calls[0]

tests/gateway/events.py

@@ -0,0 +1,20 @@
+from disco.gateway.events import GatewayEvent, Resumed
+
+
+def create_resumed_payload():
+    return GatewayEvent.from_dispatch(None, {
+        't': 'RESUMED',
+        'd': {
+            '_trace': ['test', '1', '2', '3'],
+        }
+    })
+
+
+def test_from_dispatch():
+    event = create_resumed_payload()
+    assert isinstance(event, Resumed)
+    assert event.trace == ['test', '1', '2', '3']
+
+
+def test_event_creation(benchmark):
+    benchmark(create_resumed_payload)

tests/test_channel.py → tests/types/channel.py

@@ -4,7 +4,7 @@ from disco.types.channel import Channel, ChannelType
 
 
 class TestChannel(TestCase):
-    def test_nsfw_channel(self):
+    def test_deprecated_nsfw_channel(self):
         channel = Channel(
             name='nsfw-testing',
             type=ChannelType.GUILD_TEXT)
@@ -19,3 +19,7 @@ class TestChannel(TestCase):
             name='nsfw_testing',
             type=ChannelType.GUILD_TEXT)
         self.assertFalse(channel.is_nsfw)
+
+    def test_nsfw_channel(self):
+        channel = Channel(name='test', nsfw=True, type=ChannelType.GUILD_TEXT)
+        assert channel.is_nsfw

tests/utils.py

@@ -1,6 +1,4 @@
-import time
-import contextlib
-from disco.api.client import APIClient
+from disco.api.client import APIClient as _APIClient
 
 
 class CallContainer(object):
@@ -11,28 +9,7 @@ class CallContainer(object):
         self.calls.append((args, kwargs))
 
 
-class TestAPIClient(APIClient):
+class APIClient(_APIClient):
     def __init__(self):
         self.client = None
         self.http = CallContainer()
-
-
-def bench(times, func):
-    main_start = time.time()
-
-    worst = None
-    best = None
-
-    for _ in range(times):
-        start = time.time()
-        func()
-        dur = time.time() - start
-
-        if not worst or dur > worst:
-            worst = dur
-
-        if not best or dur < best:
-            best = dur
-
-    main_dur = time.time() - main_start
-    return main_dur, worst, best