Python UserIO._get_option_from_argsの例

コード例 #1

ファイル: userio_test.py プロジェクト: MordodeMaru/client

class TestUserIO(TestCase):
    def setUp(self):
        out = BiiOutputStream()
        ins = StringIO()
        self.paths = UserIO(ins, out)
        self.paths._request_while = Mock(side_effect=Exception('Boom!'))

    def test_request_boolean(self):
        self.paths.ins.write('yes')
        self.paths.ins.seek(0)
        self.assertTrue(self.paths.request_boolean('msg'))

        self.paths.ins.truncate(0)
        self.paths.ins.write('no')
        self.paths.ins.seek(0)
        self.assertFalse(self.paths.request_boolean('msg'))

        self.paths.ins.truncate(0)
        self.paths.ins.write('yds\nyes\n')
        self.paths.ins.seek(0)
        self.assertTrue(self.paths.request_boolean('msg'))

    def test_request_option(self):
        Namespace = namedtuple('Namespace', ['block', 'description', 'name', 'version'])
        args = Namespace(block='dummy/dummy/geom/master', description='description example',
                         name='geom2', version='0')
        version = self.paths._get_option_from_args('version', args, [], type(0), None)
        self.assertEquals(version, 0)

コード例 #2

ファイル: userio_test.py プロジェクト: toeb/client

class TestUserIO(TestCase):
    def setUp(self):
        out = BiiOutputStream()
        ins = StringIO()
        self.paths = UserIO(ins, out)
        self.paths._request_while = Mock(side_effect=Exception('Boom!'))

    def test_request_boolean(self):
        self.paths.ins.write('yes')
        self.paths.ins.seek(0)
        self.assertTrue(self.paths.request_boolean('msg'))

        self.paths.ins.truncate(0)
        self.paths.ins.write('no')
        self.paths.ins.seek(0)
        self.assertFalse(self.paths.request_boolean('msg'))

        self.paths.ins.truncate(0)
        self.paths.ins.write('yds\nyes\n')
        self.paths.ins.seek(0)
        self.assertTrue(self.paths.request_boolean('msg'))

    def test_request_option(self):
        Namespace = namedtuple('Namespace',
                               ['block', 'description', 'name', 'version'])
        args = Namespace(block='dummy/dummy/geom/master',
                         description='description example',
                         name='geom2',
                         version='0')
        version = self.paths._get_option_from_args('version', args, [],
                                                   type(0), None)
        self.assertEquals(version, 0)