docker-compose/tests/unit/cli/verbose_proxy_test.py

from __future__ import unicode_literals
from __future__ import absolute_import
from tests import unittest

from compose.cli import verbose_proxy


class VerboseProxyTestCase(unittest.TestCase):

    def test_format_call(self):
        expected = "(u'arg1', True, key=u'value')"
        actual = verbose_proxy.format_call(
            ("arg1", True),
            {'key': 'value'})

        self.assertEqual(expected, actual)

    def test_format_return_sequence(self):
        expected = "(list with 10 items)"
        actual = verbose_proxy.format_return(list(range(10)), 2)
        self.assertEqual(expected, actual)

    def test_format_return(self):
        expected = "{u'Id': u'ok'}"
        actual = verbose_proxy.format_return({'Id': 'ok'}, 2)
        self.assertEqual(expected, actual)

    def test_format_return_no_result(self):
        actual = verbose_proxy.format_return(None, 2)
        self.assertEqual(None, actual)