Merge pull request #36 from mocketize/asyncio

Asyncio and Aiohttp support

Author: mindflayer

Makefile

@@ -11,7 +11,7 @@ test-python:
 
 lint-python:
 	@echo "Linting Python files"
-	flake8 --exit-zero --ignore=E501 --exclude=.git,compat.py mocket
+	flake8 --exit-zero --ignore=E501,E731 --exclude=.git,compat.py mocket
 	@echo ""
 
 develop: install-dev-requirements install-test-requirements

mocket/mocket.py

@@ -1,12 +1,13 @@
 # coding=utf-8
 from __future__ import unicode_literals
 import socket
-from collections import defaultdict
-from io import BytesIO
-from .compat import encode_utf8, basestring, byte_type, text_type
 import collections
+from io import BytesIO
+
 import decorator
 
+from .compat import encode_utf8, basestring, byte_type, text_type
+
 __all__ = (
     'true_socket',
     'true_create_connection',
@@ -50,6 +51,7 @@ def __init__(self, family, type, proto=0):
         self._sock = self
         self._connected = False
         self._buflen = 1024
+        self._entry = None
 
     def setsockopt(self, level, optname, value):
         if self.true_socket:
@@ -72,8 +74,11 @@ def makefile(self, mode='r', bufsize=-1):
         self._bufsize = bufsize
         return self.fd
 
+    def get_entry(self, data):
+        return Mocket.get_entry(self._host, self._port, data)
+
     def sendall(self, data, *args, **kwargs):
-        entry = Mocket.get_entry(self._host, self._port, data)
+        entry = self.get_entry(data)
         if not entry:
             return self.true_sendall(data, *args, **kwargs)
         entry.collect(data)
@@ -103,14 +108,22 @@ def true_sendall(self, data, *args, **kwargs):
                 break
         self.fd.seek(- written, 1)
 
+    def send(self, data, *args, **kwargs):
+        entry = self.get_entry(data)
+        if entry:
+            if self._entry != entry:
+                self.sendall(data, *args, **kwargs)
+        self._entry = entry
+        return len(data)
+
     def __getattr__(self, name):
         # useful when clients call methods on real
         # socket we do not provide on the fake one
         return getattr(self.true_socket, name)
 
 
 class Mocket(object):
-    _entries = defaultdict(list)
+    _entries = collections.defaultdict(list)
     _requests = []
 
     @classmethod
@@ -131,7 +144,7 @@ def collect(cls, data):
 
     @classmethod
     def reset(cls):
-        cls._entries = defaultdict(list)
+        cls._entries = collections.defaultdict(list)
         cls._requests = []
 
     @classmethod

runtests.py

@@ -8,10 +8,23 @@ def runtests(args=None):
     if not args:
         args = []
 
+    major, minor = sys.version_info[:2]
+
+    python35 = False
+
+    if major == 3 and minor >= 5:
+        python35 = True
+
+        import pip
+        pip.main(['install', 'aiohttp'])
+
     if not any(a for a in args[1:] if not a.startswith('-')):
         args.append('tests')
         args.append('mocket')
 
+        if python35:
+            args.append('tests35')
+
     sys.exit(pytest.main(args))
 
 

setup.cfg

@@ -1,3 +1,3 @@
-[pytest]
+[tool:pytest]
 python_files=test*.py
 addopts=--doctest-modules --cov=mocket --cov-report=term-missing -v -x

tests/test_http.py

@@ -4,9 +4,11 @@
 import json
 import mock
 from . import urlopen, urlencode, HTTPError
+
 import pytest
 import requests
 from unittest import TestCase
+
 from mocket.mockhttp import Entry, Response
 from mocket.mocket import Mocket, mocketize
 

tests35/__init__.py

@@ -0,0 +1,27 @@
+import aiohttp
+import asyncio
+from unittest import TestCase
+
+from mocket.mocket import mocketize
+from mocket.mockhttp import Entry
+
+loop = asyncio.get_event_loop()
+
+
+class AioHttpEntryTestCase(TestCase):
+    @mocketize
+    def test_session(self):
+        url = 'http://httpbin.org/ip'
+        body = "asd" * 100
+        Entry.single_register(Entry.GET, url, body=body, status=404)
+        Entry.single_register(Entry.POST, url, body=body*2, status=201)
+
+        async def main(l):
+            async with aiohttp.ClientSession(loop=l) as session:
+                get_response = await session.get(url)
+                post_response = await session.post(url, data=body*6)
+                assert get_response.status == 404
+                assert post_response.status == 201
+                assert await get_response.text() == body
+                assert await post_response.text() == body*2
+        loop.run_until_complete(main(loop))