test_commands.py 文件源码

def test_register(self, mock_get_pass, mock_create, mock_save):
        output = '''Please input username and password of the registry:
repository_name'''
        __builtin__.raw_input = lambda _: 'username'  # set username
        repository = dockercloudcli.commands.dockercloud.Repository()
        repository.name = 'repository_name'
        mock_create.return_value = repository
        repository_register('repository', None, None)

        self.assertEqual(output, self.buf.getvalue().strip())
        self.buf.truncate(0)