Add the manual test file... in case I hadn't

1 files changed, 216 insertions, 0 deletions

diff --git a/t/test_manual.py b/t/test_manual.py
new file mode 100644
index 0000000..ee3c95f
--- /dev/null
+++ b/t/test_manual.py
@@ -0,0 +1,216 @@
+import unittest
+import yappi
+import utils
+import asyncio
+import tempfile
+try:
+    import aiofiles
+    HAS_AIOFILES = True
+except ImportError:
+    HAS_AIOFILES = False
+try:
+    from aiohttp import web
+    import aiohttp
+    HAS_AIOHTTP = True
+except ImportError:
+    HAS_AIOHTTP = False
+
+# A test file containing advanced asyncio tests which have to be verified
+# manually.
+
+
+class Manual(utils.AergiaUnitTestCase):
+
+    @unittest.skipIf(not HAS_AIOHTTP,
+                     "Skipping 'test_fetch' because aiohttp "
+                     "could not be imported.")
+    def test_fetch(self):
+
+        async def handle(request):
+            return web.Response(text="Hello, world")
+
+        def get_runner():
+            app = web.Application()
+            app.router.add_get('/', handle)
+            return web.AppRunner(app)
+
+        async def start_server():
+            await runner.setup()
+            site = web.TCPSite(runner, 'localhost', 8080)
+            await site.start()
+
+        async def stop_server():
+            await runner.cleanup()
+
+        async def fetch(url):
+            async with aiohttp.ClientSession() as session:
+                async with session.get(url) as response:
+                    return await response.text()
+
+        async def query(urls):
+            tasks = [fetch(url) for url in urls]
+            return await asyncio.gather(*tasks)
+
+        async def main():
+            await start_server()
+            # wait for server to start
+            await asyncio.sleep(1)
+
+            urls = ['http://localhost:8080'] * 150
+
+            await query(urls)
+            await stop_server()
+
+        print('')
+        runner = get_runner()
+        yappi.start()
+        asyncio.run(main())
+        yappi.stop()
+        self.yappi_print_traceable_results(yappi.get_func_stats())
+
+        runner = get_runner()
+        self.Aergia.start()
+        asyncio.run(main())
+        self.Aergia.print_samples()
+        self.Aergia.stop()
+        print('')
+
+    @unittest.skipIf(not HAS_AIOHTTP,
+                     "Skipping 'test_fetch_with_burn' because aiohttp "
+                     "could not be imported.")
+    def test_fetch_with_burn(self):
+
+        async def handle(request):
+            utils.burn_cpu(0.05)
+            return web.Response(text="Hello, world")
+
+        def get_runner():
+            app = web.Application()
+            app.router.add_get('/', handle)
+            return web.AppRunner(app)
+
+        async def start_server():
+            await runner.setup()
+            site = web.TCPSite(runner, 'localhost', 8080)
+            await site.start()
+
+        async def stop_server():
+            await runner.cleanup()
+
+        async def fetch(url):
+            async with aiohttp.ClientSession() as session:
+                async with session.get(url) as response:
+                    return await response.text()
+
+        async def query(urls):
+            tasks = [fetch(url) for url in urls]
+            return await asyncio.gather(*tasks)
+
+        async def main():
+            await start_server()
+            # wait for server to start
+            await asyncio.sleep(1)
+
+            urls = ['http://localhost:8080'] * 150
+
+            await query(urls)
+            await stop_server()
+
+        print('')
+        runner = get_runner()
+        yappi.start()
+        asyncio.run(main())
+        yappi.stop()
+        self.yappi_print_traceable_results(yappi.get_func_stats())
+
+        runner = get_runner()
+        self.Aergia.start()
+        asyncio.run(main())
+        self.Aergia.print_samples()
+        self.Aergia.stop()
+        print('')
+
+    @unittest.skipIf(not HAS_AIOFILES,
+                     "Skipping 'test_file' because aiofiles "
+                     "could not be imported.")
+    def test_file(self):
+
+        async def write_file(filename, content):
+            async with aiofiles.open(filename, mode='w') as f:
+                await f.write(content)
+
+        async def read_file(filename):
+            async with aiofiles.open(filename, mode='r') as f:
+                content = await f.read()
+                return content
+
+        async def read_lots(filename, iterations):
+            tasks = [read_file(filename) for _ in range(iterations)]
+            await asyncio.gather(*tasks)
+
+        async def main():
+            # Create a temp file (/tmp)
+            with tempfile.NamedTemporaryFile(delete=True, suffix='.txt') as temp_file:
+                filename = temp_file.name
+                content = 'Hello, World!'
+
+                # write the file
+                await write_file(filename, content)
+
+                # read the file 200 times
+                await read_lots(filename, 200)
+
+        print('')
+        yappi.start()
+        asyncio.run(main())
+        yappi.stop()
+        self.yappi_print_traceable_results(yappi.get_func_stats())
+
+        self.Aergia.start()
+        asyncio.run(main())
+        self.Aergia.print_samples()
+        self.Aergia.stop()
+        print('')
+
+    @unittest.skipIf(not HAS_AIOFILES,
+                     "Skipping 'test_file_with_burn' because aiofiles "
+                     "could not be imported.")
+    def test_file_with_burn(self):
+
+        async def write_file(filename, content):
+            async with aiofiles.open(filename, mode='w') as f:
+                await f.write(content)
+
+        async def read_file(filename):
+            utils.burn_cpu(0.05)
+            async with aiofiles.open(filename, mode='r') as f:
+                content = await f.read()
+                return content
+
+        async def read_lots(filename, iterations):
+            tasks = [read_file(filename) for _ in range(iterations)]
+            await asyncio.gather(*tasks)
+
+        async def main():
+            # Create a temp file (/tmp)
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.txt') as temp_file:
+                filename = temp_file.name
+                content = 'Hello, World!'
+
+                # write the file
+                await write_file(filename, content)
+
+                # read the file 200 times
+                await read_lots(filename, 200)
+
+        print('')
+        yappi.start()
+        asyncio.run(main())
+        yappi.stop()
+        self.yappi_print_traceable_results(yappi.get_func_stats())
+
+        self.Aergia.start()
+        asyncio.run(main())
+        self.Aergia.print_samples()
+        self.Aergia.stop()
+        print('')