test_edit.py 文件源码

def setup_edit_test(tweet_factory, repository, mock, *, nvim_returncode):
    tweet_1 = tweet_factory.make_tweet(42, "First tweet!", date="2017-07-07")
    tweet_2 = tweet_factory.make_tweet(57, "Second tweet", date="2017-08-02")
    repository.add_tweets([tweet_1, tweet_2])

    spy = types.SimpleNamespace()
    spy.cmd = None

    def fake_run(cmd):
        stub_process = mock.Mock()
        spy.cmd = cmd
        stub_process.returncode = nvim_returncode
        path = cmd[1]
        with open(path, "w") as stream:
            stream.write("changed")
        return stub_process

    mock.patch("subprocess.run", fake_run)
    return spy