paasio: fix formatting
This commit is contained in:
committed by
BethanyG
parent
b3d9e40f14
commit
5efb80bf27
@@ -7,6 +7,7 @@ from test_utils import MockSock, MockFile, MockException, ZEN, SuperMock
|
|||||||
|
|
||||||
from paasio import MeteredFile, MeteredSocket
|
from paasio import MeteredFile, MeteredSocket
|
||||||
|
|
||||||
|
|
||||||
class PaasioTest(unittest.TestCase):
|
class PaasioTest(unittest.TestCase):
|
||||||
def test_meteredsocket_context_manager(self):
|
def test_meteredsocket_context_manager(self):
|
||||||
wrapped = MockSock()
|
wrapped = MockSock()
|
||||||
@@ -34,7 +35,9 @@ class PaasioTest(unittest.TestCase):
|
|||||||
socket.recv(4096)
|
socket.recv(4096)
|
||||||
self.assertFalse(mock.__enter__.called)
|
self.assertFalse(mock.__enter__.called)
|
||||||
mock.__exit__.assert_called_once_with(
|
mock.__exit__.assert_called_once_with(
|
||||||
MockException, err.exception, ANY,
|
MockException,
|
||||||
|
err.exception,
|
||||||
|
ANY,
|
||||||
)
|
)
|
||||||
self.assertEqual(exception, err.exception)
|
self.assertEqual(exception, err.exception)
|
||||||
|
|
||||||
@@ -48,7 +51,9 @@ class PaasioTest(unittest.TestCase):
|
|||||||
socket.recv(4096)
|
socket.recv(4096)
|
||||||
self.assertFalse(mock.__enter__.called)
|
self.assertFalse(mock.__enter__.called)
|
||||||
mock.__exit__.assert_called_once_with(
|
mock.__exit__.assert_called_once_with(
|
||||||
MockException, exception, ANY,
|
MockException,
|
||||||
|
exception,
|
||||||
|
ANY,
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_meteredsocket_recv_once(self):
|
def test_meteredsocket_recv_once(self):
|
||||||
@@ -220,14 +225,11 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertFalse(mock.__enter__.called)
|
self.assertFalse(mock.__enter__.called)
|
||||||
mock.__exit__.assert_called_once_with(None, None, None)
|
mock.__exit__.assert_called_once_with(None, None, None)
|
||||||
self.assertEqual(2, len(mock.mock_calls))
|
self.assertEqual(2, len(mock.mock_calls))
|
||||||
with self.assertRaisesRegex(
|
with self.assertRaisesRegex(ValueError, "I/O operation on closed file."):
|
||||||
ValueError, "I/O operation on closed file."
|
|
||||||
):
|
|
||||||
file.read()
|
file.read()
|
||||||
with self.assertRaisesRegex(
|
with self.assertRaisesRegex(ValueError, "I/O operation on closed file."):
|
||||||
ValueError, "I/O operation on closed file."
|
|
||||||
):
|
|
||||||
file.write(b"data")
|
file.write(b"data")
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_context_manager_exception_raise(self, super_mock):
|
def test_meteredfile_context_manager_exception_raise(self, super_mock):
|
||||||
exception = MockException("Should raise")
|
exception = MockException("Should raise")
|
||||||
@@ -241,9 +243,12 @@ class PaasioTest(unittest.TestCase):
|
|||||||
file.read()
|
file.read()
|
||||||
self.assertFalse(mock.__enter__.called)
|
self.assertFalse(mock.__enter__.called)
|
||||||
mock.__exit__.assert_called_once_with(
|
mock.__exit__.assert_called_once_with(
|
||||||
MockException, err.exception, ANY,
|
MockException,
|
||||||
|
err.exception,
|
||||||
|
ANY,
|
||||||
)
|
)
|
||||||
self.assertEqual(exception, err.exception)
|
self.assertEqual(exception, err.exception)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_context_manager_exception_suppress(self, super_mock):
|
def test_meteredfile_context_manager_exception_suppress(self, super_mock):
|
||||||
exception = MockException("Should suppress")
|
exception = MockException("Should suppress")
|
||||||
@@ -256,8 +261,11 @@ class PaasioTest(unittest.TestCase):
|
|||||||
file.read()
|
file.read()
|
||||||
self.assertFalse(mock.__enter__.called)
|
self.assertFalse(mock.__enter__.called)
|
||||||
mock.__exit__.assert_called_once_with(
|
mock.__exit__.assert_called_once_with(
|
||||||
MockException, exception, ANY,
|
MockException,
|
||||||
|
exception,
|
||||||
|
ANY,
|
||||||
)
|
)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_iteration(self, super_mock):
|
def test_meteredfile_iteration(self, super_mock):
|
||||||
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
||||||
@@ -266,9 +274,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
file = MeteredFile()
|
file = MeteredFile()
|
||||||
for line in file:
|
for line in file:
|
||||||
actual_reads += line
|
actual_reads += line
|
||||||
self.assertLess(
|
self.assertLess(0, mock.readline.call_count, "File's readline not called")
|
||||||
0, mock.readline.call_count, "File's readline not called"
|
|
||||||
)
|
|
||||||
self.assertGreater(
|
self.assertGreater(
|
||||||
50, mock.readline.call_count, "Possible infinte loop detected"
|
50, mock.readline.call_count, "Possible infinte loop detected"
|
||||||
)
|
)
|
||||||
@@ -276,6 +282,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertFalse(mock.__iter__.called)
|
self.assertFalse(mock.__iter__.called)
|
||||||
self.assertEqual(len(ZEN), file.read_bytes)
|
self.assertEqual(len(ZEN), file.read_bytes)
|
||||||
self.assertEqual(ZEN, actual_reads)
|
self.assertEqual(ZEN, actual_reads)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_read_once(self, super_mock):
|
def test_meteredfile_read_once(self, super_mock):
|
||||||
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
||||||
@@ -302,6 +309,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual((len(ZEN)), file.read_bytes)
|
self.assertEqual((len(ZEN)), file.read_bytes)
|
||||||
self.assertEqual(1, file.read_ops)
|
self.assertEqual(1, file.read_ops)
|
||||||
self.assertEqual(mock.read.call_count, file.read_ops)
|
self.assertEqual(mock.read.call_count, file.read_ops)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_read_multiple(self, super_mock):
|
def test_meteredfile_read_multiple(self, super_mock):
|
||||||
wrapped = MockFile(ZEN)
|
wrapped = MockFile(ZEN)
|
||||||
@@ -315,6 +323,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(5, file.read_ops)
|
self.assertEqual(5, file.read_ops)
|
||||||
self.assertEqual(150, file.read_bytes)
|
self.assertEqual(150, file.read_bytes)
|
||||||
self.assertEqual(5, mock.read.call_count)
|
self.assertEqual(5, mock.read.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_read_multiple_chunk(self, super_mock):
|
def test_meteredfile_read_multiple_chunk(self, super_mock):
|
||||||
wrapped = MockFile(ZEN, chunk=20)
|
wrapped = MockFile(ZEN, chunk=20)
|
||||||
@@ -341,6 +350,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(7, file.read_ops)
|
self.assertEqual(7, file.read_ops)
|
||||||
self.assertEqual(73, file.read_bytes)
|
self.assertEqual(73, file.read_bytes)
|
||||||
self.assertEqual(7, mock.read.call_count)
|
self.assertEqual(7, mock.read.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_read_under_size(self, super_mock):
|
def test_meteredfile_read_under_size(self, super_mock):
|
||||||
wrapped = MockFile(ZEN, chunk=257) # largish odd number
|
wrapped = MockFile(ZEN, chunk=257) # largish odd number
|
||||||
@@ -352,6 +362,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(1, file.read_ops)
|
self.assertEqual(1, file.read_ops)
|
||||||
self.assertEqual(257, file.read_bytes)
|
self.assertEqual(257, file.read_bytes)
|
||||||
self.assertEqual(1, mock.read.call_count)
|
self.assertEqual(1, mock.read.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_write_once(self, super_mock):
|
def test_meteredfile_write_once(self, super_mock):
|
||||||
wrapped = MockFile(chunk=257) # largish odd number
|
wrapped = MockFile(chunk=257) # largish odd number
|
||||||
@@ -364,6 +375,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(1, file.write_ops)
|
self.assertEqual(1, file.write_ops)
|
||||||
self.assertEqual(257, file.write_bytes)
|
self.assertEqual(257, file.write_bytes)
|
||||||
self.assertEqual(1, mock.write.call_count)
|
self.assertEqual(1, mock.write.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_write_multiple(self, super_mock):
|
def test_meteredfile_write_multiple(self, super_mock):
|
||||||
wrapped = MockFile()
|
wrapped = MockFile()
|
||||||
@@ -381,6 +393,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(4, file.write_ops)
|
self.assertEqual(4, file.write_ops)
|
||||||
self.assertEqual(39, file.write_bytes)
|
self.assertEqual(39, file.write_bytes)
|
||||||
self.assertEqual(4, mock.write.call_count)
|
self.assertEqual(4, mock.write.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_write_under_size(self, super_mock):
|
def test_meteredfile_write_under_size(self, super_mock):
|
||||||
wrapped = MockFile(chunk=257) # largish odd number
|
wrapped = MockFile(chunk=257) # largish odd number
|
||||||
@@ -393,6 +406,7 @@ class PaasioTest(unittest.TestCase):
|
|||||||
self.assertEqual(1, file.write_ops)
|
self.assertEqual(1, file.write_ops)
|
||||||
self.assertEqual(123, file.write_bytes)
|
self.assertEqual(123, file.write_bytes)
|
||||||
self.assertEqual(1, mock.write.call_count)
|
self.assertEqual(1, mock.write.call_count)
|
||||||
|
|
||||||
@patch("paasio.super", create=True, new_callable=SuperMock)
|
@patch("paasio.super", create=True, new_callable=SuperMock)
|
||||||
def test_meteredfile_stats_read_only(self, super_mock):
|
def test_meteredfile_stats_read_only(self, super_mock):
|
||||||
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
mock = NonCallableMagicMock(wraps=MockFile(ZEN), autospec=True)
|
||||||
|
|||||||
@@ -4,7 +4,6 @@ import io
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
ZEN = b"""Beautiful is better than ugly.
|
ZEN = b"""Beautiful is better than ugly.
|
||||||
Explicit is better than implicit.
|
Explicit is better than implicit.
|
||||||
Simple is better than complex.
|
Simple is better than complex.
|
||||||
@@ -84,14 +83,10 @@ class MockSock:
|
|||||||
if self.__closed:
|
if self.__closed:
|
||||||
raise OSError(errno.EBADF, os.strerror(errno.EBADF))
|
raise OSError(errno.EBADF, os.strerror(errno.EBADF))
|
||||||
if bufsize is None:
|
if bufsize is None:
|
||||||
raise TypeError(
|
raise TypeError("'NoneType' object cannot be interpreted as an integer")
|
||||||
"'NoneType' object cannot be interpreted as an integer"
|
|
||||||
)
|
|
||||||
if not isinstance(flags, int):
|
if not isinstance(flags, int):
|
||||||
raise TypeError(
|
raise TypeError(
|
||||||
"an integer is required (got type {})".format(
|
"an integer is required (got type {})".format(type(flags).__name__)
|
||||||
type(flags).__name__
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
self.flags = flags
|
self.flags = flags
|
||||||
if self.__exception is not None:
|
if self.__exception is not None:
|
||||||
@@ -106,9 +101,7 @@ class MockSock:
|
|||||||
raise OSError(errno.EBADF, os.strerror(errno.EBADF))
|
raise OSError(errno.EBADF, os.strerror(errno.EBADF))
|
||||||
if not isinstance(flags, int):
|
if not isinstance(flags, int):
|
||||||
raise TypeError(
|
raise TypeError(
|
||||||
"an integer is required (got type {})".format(
|
"an integer is required (got type {})".format(type(flags).__name__)
|
||||||
type(flags).__name__
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
self.flags = flags
|
self.flags = flags
|
||||||
if self.__chunk is None:
|
if self.__chunk is None:
|
||||||
@@ -130,9 +123,7 @@ class SuperMock:
|
|||||||
if frame is None:
|
if frame is None:
|
||||||
raise RuntimeError("Could not get current frame object")
|
raise RuntimeError("Could not get current frame object")
|
||||||
stack = inspect.getouterframes(frame)
|
stack = inspect.getouterframes(frame)
|
||||||
if any(
|
if any(frame[3] == "__init__" and "paasio" in frame[1] for frame in stack):
|
||||||
frame[3] == "__init__" and "paasio" in frame[1] for frame in stack
|
|
||||||
):
|
|
||||||
return self
|
return self
|
||||||
else:
|
else:
|
||||||
return self.mock_object
|
return self.mock_object
|
||||||
|
|||||||
Reference in New Issue
Block a user