generated from daniil-berg/boilerplate-py
removed a few functions from the public API; fixed some docstrings/comments
This commit is contained in:
parent
3d104c979e
commit
3503c0bf44
@ -1,6 +1,6 @@
|
|||||||
[metadata]
|
[metadata]
|
||||||
name = asyncio-taskpool
|
name = asyncio-taskpool
|
||||||
version = 0.6.3
|
version = 0.6.4
|
||||||
author = Daniil Fajnberg
|
author = Daniil Fajnberg
|
||||||
author_email = mail@daniil.fajnberg.de
|
author_email = mail@daniil.fajnberg.de
|
||||||
description = Dynamically manage pools of asyncio tasks
|
description = Dynamically manage pools of asyncio tasks
|
||||||
|
@ -41,7 +41,7 @@ class ControlClient(ABC):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def client_info() -> dict:
|
def _client_info() -> dict:
|
||||||
"""Returns a dictionary of client information relevant for the handshake with the server."""
|
"""Returns a dictionary of client information relevant for the handshake with the server."""
|
||||||
return {CLIENT_INFO.TERMINAL_WIDTH: shutil.get_terminal_size().columns}
|
return {CLIENT_INFO.TERMINAL_WIDTH: shutil.get_terminal_size().columns}
|
||||||
|
|
||||||
@ -73,7 +73,7 @@ class ControlClient(ABC):
|
|||||||
writer: The `asyncio.StreamWriter` returned by the `_open_connection()` method
|
writer: The `asyncio.StreamWriter` returned by the `_open_connection()` method
|
||||||
"""
|
"""
|
||||||
self._connected = True
|
self._connected = True
|
||||||
writer.write(json.dumps(self.client_info()).encode())
|
writer.write(json.dumps(self._client_info()).encode())
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
print("Connected to", (await reader.read(SESSION_MSG_BYTES)).decode())
|
print("Connected to", (await reader.read(SESSION_MSG_BYTES)).decode())
|
||||||
print("Type '-h' to get help and usage instructions for all available commands.\n")
|
print("Type '-h' to get help and usage instructions for all available commands.\n")
|
||||||
|
@ -23,10 +23,10 @@ from argparse import Action, ArgumentParser, ArgumentDefaultsHelpFormatter, Help
|
|||||||
from asyncio.streams import StreamWriter
|
from asyncio.streams import StreamWriter
|
||||||
from inspect import Parameter, getmembers, isfunction, signature
|
from inspect import Parameter, getmembers, isfunction, signature
|
||||||
from shutil import get_terminal_size
|
from shutil import get_terminal_size
|
||||||
from typing import Callable, Container, Dict, Set, Type, TypeVar
|
from typing import Any, Callable, Container, Dict, Set, Type, TypeVar
|
||||||
|
|
||||||
from ..constants import CLIENT_INFO, CMD, STREAM_WRITER
|
from ..constants import CLIENT_INFO, CMD, STREAM_WRITER
|
||||||
from ..exceptions import HelpRequested
|
from ..exceptions import HelpRequested, ParserError
|
||||||
from ..helpers import get_first_doc_line
|
from ..helpers import get_first_doc_line
|
||||||
|
|
||||||
|
|
||||||
@ -35,7 +35,6 @@ ParsersDict = Dict[str, 'ControlParser']
|
|||||||
|
|
||||||
OMIT_PARAMS_DEFAULT = ('self', )
|
OMIT_PARAMS_DEFAULT = ('self', )
|
||||||
|
|
||||||
FORMATTER_CLASS = 'formatter_class'
|
|
||||||
NAME, PROG, HELP, DESCRIPTION = 'name', 'prog', 'help', 'description'
|
NAME, PROG, HELP, DESCRIPTION = 'name', 'prog', 'help', 'description'
|
||||||
|
|
||||||
|
|
||||||
@ -79,24 +78,23 @@ class ControlParser(ArgumentParser):
|
|||||||
def __init__(self, stream_writer: StreamWriter, terminal_width: int = None,
|
def __init__(self, stream_writer: StreamWriter, terminal_width: int = None,
|
||||||
**kwargs) -> None:
|
**kwargs) -> None:
|
||||||
"""
|
"""
|
||||||
Sets additional internal attributes depending on whether a parent-parser was defined.
|
Subclass of the `ArgumentParser` geared towards asynchronous interaction with an object "from the outside".
|
||||||
|
|
||||||
The `help_formatter_factory` is called and the returned class is mapped to the `FORMATTER_CLASS` keyword.
|
Allows directing output to a specified writer rather than stdout/stderr and setting terminal width explicitly.
|
||||||
By default, `exit_on_error` is set to `False` (as opposed to how the parent class handles it).
|
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
stream_writer:
|
stream_writer:
|
||||||
The instance of the `asyncio.StreamWriter` to use for message output.
|
The instance of the `asyncio.StreamWriter` to use for message output.
|
||||||
terminal_width (optional):
|
terminal_width (optional):
|
||||||
The terminal width to assume for all message formatting. Defaults to `shutil.get_terminal_size`.
|
The terminal width to use for all message formatting. Defaults to `shutil.get_terminal_size().columns`.
|
||||||
**kwargs(optional):
|
**kwargs(optional):
|
||||||
In addition to the regular `ArgumentParser` constructor parameters, this method expects the instance of
|
Passed to the parent class constructor. The exception is the `formatter_class` parameter: Even if a
|
||||||
the `StreamWriter` as well as the terminal width both to be passed explicitly, if the `parent` argument
|
class is specified, it will always be subclassed in the `help_formatter_factory`.
|
||||||
is empty.
|
Also, by default, `exit_on_error` is set to `False` (as opposed to how the parent class handles it).
|
||||||
"""
|
"""
|
||||||
self._stream_writer: StreamWriter = stream_writer
|
self._stream_writer: StreamWriter = stream_writer
|
||||||
self._terminal_width: int = terminal_width if terminal_width is not None else get_terminal_size().columns
|
self._terminal_width: int = terminal_width if terminal_width is not None else get_terminal_size().columns
|
||||||
kwargs[FORMATTER_CLASS] = self.help_formatter_factory(self._terminal_width, kwargs.get(FORMATTER_CLASS))
|
kwargs['formatter_class'] = self.help_formatter_factory(self._terminal_width, kwargs.get('formatter_class'))
|
||||||
kwargs.setdefault('exit_on_error', False)
|
kwargs.setdefault('exit_on_error', False)
|
||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self._flags: Set[str] = set()
|
self._flags: Set[str] = set()
|
||||||
@ -219,7 +217,7 @@ class ControlParser(ArgumentParser):
|
|||||||
def error(self, message: str) -> None:
|
def error(self, message: str) -> None:
|
||||||
"""This just adds the custom `HelpRequested` exception after the parent class' method."""
|
"""This just adds the custom `HelpRequested` exception after the parent class' method."""
|
||||||
super().error(message=message)
|
super().error(message=message)
|
||||||
raise HelpRequested
|
raise ParserError
|
||||||
|
|
||||||
def print_help(self, file=None) -> None:
|
def print_help(self, file=None) -> None:
|
||||||
"""This just adds the custom `HelpRequested` exception after the parent class' method."""
|
"""This just adds the custom `HelpRequested` exception after the parent class' method."""
|
||||||
@ -267,9 +265,8 @@ class ControlParser(ArgumentParser):
|
|||||||
# This is to be able to later unpack an arbitrary number of positional arguments.
|
# This is to be able to later unpack an arbitrary number of positional arguments.
|
||||||
kwargs.setdefault('nargs', '*')
|
kwargs.setdefault('nargs', '*')
|
||||||
if not kwargs.get('action') == 'store_true':
|
if not kwargs.get('action') == 'store_true':
|
||||||
# The lambda wrapper around the type annotation is to avoid ValueError being raised on suppressed arguments.
|
# Set the type from the parameter annotation.
|
||||||
# See: https://bugs.python.org/issue36078
|
kwargs.setdefault('type', _get_arg_type_wrapper(parameter.annotation))
|
||||||
kwargs.setdefault('type', get_arg_type_wrapper(parameter.annotation))
|
|
||||||
return self.add_argument(*name_or_flags, **kwargs)
|
return self.add_argument(*name_or_flags, **kwargs)
|
||||||
|
|
||||||
def add_function_args(self, function: Callable, omit: Container[str] = OMIT_PARAMS_DEFAULT) -> None:
|
def add_function_args(self, function: Callable, omit: Container[str] = OMIT_PARAMS_DEFAULT) -> None:
|
||||||
@ -293,7 +290,13 @@ class ControlParser(ArgumentParser):
|
|||||||
self.add_function_arg(param, help=repr(param.annotation))
|
self.add_function_arg(param, help=repr(param.annotation))
|
||||||
|
|
||||||
|
|
||||||
def get_arg_type_wrapper(cls: Type) -> Callable:
|
def _get_arg_type_wrapper(cls: Type) -> Callable[[Any], Any]:
|
||||||
def wrapper(arg):
|
"""
|
||||||
return arg if arg is SUPPRESS else cls(arg)
|
Returns a wrapper for the constructor of `cls` to avoid a ValueError being raised on suppressed arguments.
|
||||||
|
|
||||||
|
See: https://bugs.python.org/issue36078
|
||||||
|
"""
|
||||||
|
def wrapper(arg: Any) -> Any: return arg if arg is SUPPRESS else cls(arg)
|
||||||
|
# Copy the name of the class to maintain useful help messages when incorrect arguments are passed.
|
||||||
|
wrapper.__name__ = cls.__name__
|
||||||
return wrapper
|
return wrapper
|
||||||
|
@ -125,6 +125,7 @@ class ControlServer(ABC): # TODO: Implement interface for normal TaskPool insta
|
|||||||
async def serve_forever(self) -> Task:
|
async def serve_forever(self) -> Task:
|
||||||
"""
|
"""
|
||||||
This method actually starts the server and begins listening to client connections on the specified interface.
|
This method actually starts the server and begins listening to client connections on the specified interface.
|
||||||
|
|
||||||
It should never block because the serving will be performed in a separate task.
|
It should never block because the serving will be performed in a separate task.
|
||||||
"""
|
"""
|
||||||
log.debug("Starting %s...", self.__class__.__name__)
|
log.debug("Starting %s...", self.__class__.__name__)
|
||||||
|
@ -27,7 +27,7 @@ from inspect import isfunction, signature
|
|||||||
from typing import Callable, Optional, Union, TYPE_CHECKING
|
from typing import Callable, Optional, Union, TYPE_CHECKING
|
||||||
|
|
||||||
from ..constants import CLIENT_INFO, CMD, CMD_OK, SESSION_MSG_BYTES, STREAM_WRITER
|
from ..constants import CLIENT_INFO, CMD, CMD_OK, SESSION_MSG_BYTES, STREAM_WRITER
|
||||||
from ..exceptions import CommandError, HelpRequested
|
from ..exceptions import CommandError, HelpRequested, ParserError
|
||||||
from ..helpers import return_or_exception
|
from ..helpers import return_or_exception
|
||||||
from ..pool import TaskPool, SimpleTaskPool
|
from ..pool import TaskPool, SimpleTaskPool
|
||||||
from .parser import ControlParser
|
from .parser import ControlParser
|
||||||
@ -157,7 +157,7 @@ class ControlSession:
|
|||||||
log.debug("%s got an ArgumentError", self._client_class_name)
|
log.debug("%s got an ArgumentError", self._client_class_name)
|
||||||
self._writer.write(str(e).encode())
|
self._writer.write(str(e).encode())
|
||||||
return
|
return
|
||||||
except HelpRequested:
|
except (HelpRequested, ParserError):
|
||||||
log.debug("%s received usage help", self._client_class_name)
|
log.debug("%s received usage help", self._client_class_name)
|
||||||
return
|
return
|
||||||
command = kwargs.pop(CMD)
|
command = kwargs.pop(CMD)
|
||||||
|
@ -67,5 +67,9 @@ class HelpRequested(ServerException):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class ParserError(ServerException):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
class CommandError(ServerException):
|
class CommandError(ServerException):
|
||||||
pass
|
pass
|
||||||
|
@ -15,7 +15,7 @@ You should have received a copy of the GNU Lesser General Public License along w
|
|||||||
If not, see <https://www.gnu.org/licenses/>."""
|
If not, see <https://www.gnu.org/licenses/>."""
|
||||||
|
|
||||||
__doc__ = """
|
__doc__ = """
|
||||||
Miscellaneous helper functions.
|
Miscellaneous helper functions. None of these should be considered part of the public API.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
@ -53,6 +53,6 @@ class Queue(_Queue):
|
|||||||
Implements an asynchronous context manager for the queue.
|
Implements an asynchronous context manager for the queue.
|
||||||
|
|
||||||
Upon exiting `item_processed()` is called. This is why this context manager may not always be what you want,
|
Upon exiting `item_processed()` is called. This is why this context manager may not always be what you want,
|
||||||
but in some situations it makes the codes much cleaner.
|
but in some situations it makes the code much cleaner.
|
||||||
"""
|
"""
|
||||||
self.item_processed()
|
self.item_processed()
|
||||||
|
@ -55,7 +55,7 @@ class ControlClientTestCase(IsolatedAsyncioTestCase):
|
|||||||
|
|
||||||
def test_client_info(self):
|
def test_client_info(self):
|
||||||
self.assertEqual({CLIENT_INFO.TERMINAL_WIDTH: shutil.get_terminal_size().columns},
|
self.assertEqual({CLIENT_INFO.TERMINAL_WIDTH: shutil.get_terminal_size().columns},
|
||||||
client.ControlClient.client_info())
|
client.ControlClient._client_info())
|
||||||
|
|
||||||
async def test_abstract(self):
|
async def test_abstract(self):
|
||||||
with self.assertRaises(NotImplementedError):
|
with self.assertRaises(NotImplementedError):
|
||||||
@ -65,12 +65,12 @@ class ControlClientTestCase(IsolatedAsyncioTestCase):
|
|||||||
self.assertEqual(self.kwargs, self.client._conn_kwargs)
|
self.assertEqual(self.kwargs, self.client._conn_kwargs)
|
||||||
self.assertFalse(self.client._connected)
|
self.assertFalse(self.client._connected)
|
||||||
|
|
||||||
@patch.object(client.ControlClient, 'client_info')
|
@patch.object(client.ControlClient, '_client_info')
|
||||||
async def test__server_handshake(self, mock_client_info: MagicMock):
|
async def test__server_handshake(self, mock__client_info: MagicMock):
|
||||||
mock_client_info.return_value = mock_info = {FOO: 1, BAR: 9999}
|
mock__client_info.return_value = mock_info = {FOO: 1, BAR: 9999}
|
||||||
self.assertIsNone(await self.client._server_handshake(self.mock_reader, self.mock_writer))
|
self.assertIsNone(await self.client._server_handshake(self.mock_reader, self.mock_writer))
|
||||||
self.assertTrue(self.client._connected)
|
self.assertTrue(self.client._connected)
|
||||||
mock_client_info.assert_called_once_with()
|
mock__client_info.assert_called_once_with()
|
||||||
self.mock_write.assert_called_once_with(json.dumps(mock_info).encode())
|
self.mock_write.assert_called_once_with(json.dumps(mock_info).encode())
|
||||||
self.mock_drain.assert_awaited_once_with()
|
self.mock_drain.assert_awaited_once_with()
|
||||||
self.mock_read.assert_awaited_once_with(SESSION_MSG_BYTES)
|
self.mock_read.assert_awaited_once_with(SESSION_MSG_BYTES)
|
||||||
|
@ -25,7 +25,7 @@ from unittest import TestCase
|
|||||||
from unittest.mock import MagicMock, call, patch
|
from unittest.mock import MagicMock, call, patch
|
||||||
|
|
||||||
from asyncio_taskpool.control import parser
|
from asyncio_taskpool.control import parser
|
||||||
from asyncio_taskpool.exceptions import HelpRequested
|
from asyncio_taskpool.exceptions import HelpRequested, ParserError
|
||||||
|
|
||||||
|
|
||||||
FOO, BAR = 'foo', 'bar'
|
FOO, BAR = 'foo', 'bar'
|
||||||
@ -41,7 +41,7 @@ class ControlServerTestCase(TestCase):
|
|||||||
self.kwargs = {
|
self.kwargs = {
|
||||||
'stream_writer': self.stream_writer,
|
'stream_writer': self.stream_writer,
|
||||||
'terminal_width': self.terminal_width,
|
'terminal_width': self.terminal_width,
|
||||||
parser.FORMATTER_CLASS: FOO
|
'formatter_class': FOO
|
||||||
}
|
}
|
||||||
self.parser = parser.ControlParser(**self.kwargs)
|
self.parser = parser.ControlParser(**self.kwargs)
|
||||||
|
|
||||||
@ -183,7 +183,7 @@ class ControlServerTestCase(TestCase):
|
|||||||
|
|
||||||
@patch.object(parser.ArgumentParser, 'error')
|
@patch.object(parser.ArgumentParser, 'error')
|
||||||
def test_error(self, mock_supercls_error: MagicMock):
|
def test_error(self, mock_supercls_error: MagicMock):
|
||||||
with self.assertRaises(HelpRequested):
|
with self.assertRaises(ParserError):
|
||||||
self.parser.error(FOO + BAR)
|
self.parser.error(FOO + BAR)
|
||||||
mock_supercls_error.assert_called_once_with(message=FOO + BAR)
|
mock_supercls_error.assert_called_once_with(message=FOO + BAR)
|
||||||
|
|
||||||
@ -194,11 +194,11 @@ class ControlServerTestCase(TestCase):
|
|||||||
self.parser.print_help(arg)
|
self.parser.print_help(arg)
|
||||||
mock_print_help.assert_called_once_with(arg)
|
mock_print_help.assert_called_once_with(arg)
|
||||||
|
|
||||||
@patch.object(parser, 'get_arg_type_wrapper')
|
@patch.object(parser, '_get_arg_type_wrapper')
|
||||||
@patch.object(parser.ArgumentParser, 'add_argument')
|
@patch.object(parser.ArgumentParser, 'add_argument')
|
||||||
def test_add_function_arg(self, mock_add_argument: MagicMock, mock_get_arg_type_wrapper: MagicMock):
|
def test_add_function_arg(self, mock_add_argument: MagicMock, mock__get_arg_type_wrapper: MagicMock):
|
||||||
mock_add_argument.return_value = expected_output = 'action'
|
mock_add_argument.return_value = expected_output = 'action'
|
||||||
mock_get_arg_type_wrapper.return_value = mock_type = 'fake'
|
mock__get_arg_type_wrapper.return_value = mock_type = 'fake'
|
||||||
|
|
||||||
foo_type, args_type, bar_type, baz_type, boo_type = tuple, str, int, float, complex
|
foo_type, args_type, bar_type, baz_type, boo_type = tuple, str, int, float, complex
|
||||||
bar_default, baz_default, boo_default = 1, 0.1, 1j
|
bar_default, baz_default, boo_default = 1, 0.1, 1j
|
||||||
@ -211,42 +211,42 @@ class ControlServerTestCase(TestCase):
|
|||||||
kwargs = {FOO + BAR: 'xyz'}
|
kwargs = {FOO + BAR: 'xyz'}
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_foo, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_foo, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('foo', type=mock_type, **kwargs)
|
mock_add_argument.assert_called_once_with('foo', type=mock_type, **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_called_once_with(foo_type)
|
mock__get_arg_type_wrapper.assert_called_once_with(foo_type)
|
||||||
|
|
||||||
mock_add_argument.reset_mock()
|
mock_add_argument.reset_mock()
|
||||||
mock_get_arg_type_wrapper.reset_mock()
|
mock__get_arg_type_wrapper.reset_mock()
|
||||||
|
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_args, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_args, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('args', nargs='*', type=mock_type, **kwargs)
|
mock_add_argument.assert_called_once_with('args', nargs='*', type=mock_type, **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_called_once_with(args_type)
|
mock__get_arg_type_wrapper.assert_called_once_with(args_type)
|
||||||
|
|
||||||
mock_add_argument.reset_mock()
|
mock_add_argument.reset_mock()
|
||||||
mock_get_arg_type_wrapper.reset_mock()
|
mock__get_arg_type_wrapper.reset_mock()
|
||||||
|
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_bar, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_bar, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('-b', '--bar', default=bar_default, type=mock_type, **kwargs)
|
mock_add_argument.assert_called_once_with('-b', '--bar', default=bar_default, type=mock_type, **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_called_once_with(bar_type)
|
mock__get_arg_type_wrapper.assert_called_once_with(bar_type)
|
||||||
|
|
||||||
mock_add_argument.reset_mock()
|
mock_add_argument.reset_mock()
|
||||||
mock_get_arg_type_wrapper.reset_mock()
|
mock__get_arg_type_wrapper.reset_mock()
|
||||||
|
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_baz, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_baz, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('-B', '--baz', default=baz_default, type=mock_type, **kwargs)
|
mock_add_argument.assert_called_once_with('-B', '--baz', default=baz_default, type=mock_type, **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_called_once_with(baz_type)
|
mock__get_arg_type_wrapper.assert_called_once_with(baz_type)
|
||||||
|
|
||||||
mock_add_argument.reset_mock()
|
mock_add_argument.reset_mock()
|
||||||
mock_get_arg_type_wrapper.reset_mock()
|
mock__get_arg_type_wrapper.reset_mock()
|
||||||
|
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_boo, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_boo, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('--boo', default=boo_default, type=mock_type, **kwargs)
|
mock_add_argument.assert_called_once_with('--boo', default=boo_default, type=mock_type, **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_called_once_with(boo_type)
|
mock__get_arg_type_wrapper.assert_called_once_with(boo_type)
|
||||||
|
|
||||||
mock_add_argument.reset_mock()
|
mock_add_argument.reset_mock()
|
||||||
mock_get_arg_type_wrapper.reset_mock()
|
mock__get_arg_type_wrapper.reset_mock()
|
||||||
|
|
||||||
self.assertEqual(expected_output, self.parser.add_function_arg(param_flag, **kwargs))
|
self.assertEqual(expected_output, self.parser.add_function_arg(param_flag, **kwargs))
|
||||||
mock_add_argument.assert_called_once_with('-f', '--flag', action='store_true', **kwargs)
|
mock_add_argument.assert_called_once_with('-f', '--flag', action='store_true', **kwargs)
|
||||||
mock_get_arg_type_wrapper.assert_not_called()
|
mock__get_arg_type_wrapper.assert_not_called()
|
||||||
|
|
||||||
@patch.object(parser.ControlParser, 'add_function_arg')
|
@patch.object(parser.ControlParser, 'add_function_arg')
|
||||||
def test_add_function_args(self, mock_add_function_arg: MagicMock):
|
def test_add_function_args(self, mock_add_function_arg: MagicMock):
|
||||||
@ -261,7 +261,8 @@ class ControlServerTestCase(TestCase):
|
|||||||
|
|
||||||
|
|
||||||
class RestTestCase(TestCase):
|
class RestTestCase(TestCase):
|
||||||
def test_get_arg_type_wrapper(self):
|
def test__get_arg_type_wrapper(self):
|
||||||
type_wrap = parser.get_arg_type_wrapper(int)
|
type_wrap = parser._get_arg_type_wrapper(int)
|
||||||
|
self.assertEqual('int', type_wrap.__name__)
|
||||||
self.assertEqual(SUPPRESS, type_wrap(SUPPRESS))
|
self.assertEqual(SUPPRESS, type_wrap(SUPPRESS))
|
||||||
self.assertEqual(13, type_wrap('13'))
|
self.assertEqual(13, type_wrap('13'))
|
||||||
|
Loading…
Reference in New Issue
Block a user