def create_mocked_vim_vars(blog_url, username, password, blog_name): def vim_variable_mock(self, var_name, prefix=True): if prefix: var_name = "_".join((self.blog_name, var_name)) val = getattr(self, var_name) return val if not isinstance(val, Mock) else None vim_vars = Mock("VimVars") vim_vars.vim_variable.mock_returns_func = partial(vim_variable_mock, vim_vars) vim_vars.blog_url = blog_url vim_vars.blog_username = username vim_vars.blog_password = password vim_vars.blog_name = blog_name vim_vars.blog_postsource = False return vim_vars
def create_mocked_vim_vars(blog_url, username, password, blog_name): def vim_variable_mock(self, var_name, prefix=True): if prefix: var_name = '_'.join((self.blog_name, var_name)) val = getattr(self, var_name) return (val if not isinstance(val, Mock) else None) vim_vars = Mock('VimVars') vim_vars.vim_variable.mock_returns_func = partial(vim_variable_mock, vim_vars) vim_vars.blog_url = blog_url vim_vars.blog_username = username vim_vars.blog_password = password vim_vars.blog_name = blog_name vim_vars.blog_postsource = False return vim_vars