From f4cbeaff9bd9163210a81a7158f4202e77bcd9dd Mon Sep 17 00:00:00 2001 From: XZS Date: Wed, 27 Apr 2016 12:30:36 +0200 Subject: [PATCH] test Vim variant of stash segment --- tests/test_segments.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/tests/test_segments.py b/tests/test_segments.py index 207c2843..f15fffde 100644 --- a/tests/test_segments.py +++ b/tests/test_segments.py @@ -1413,6 +1413,30 @@ class TestVim(TestCase): {'divider_highlight_group': 'branch:divider', 'highlight_groups': ['branch_clean', 'branch'], 'contents': 'foo'} ]) + def test_stash(self): + pl = Pl() + create_watcher = get_fallback_create_watcher() + with vim_module._with('buffer', '/foo') as segment_info: + stash = partial(self.vim.stash, pl=pl, create_watcher=create_watcher, segment_info=segment_info) + + def forge_stash(n): + return replace_attr(self.vcs, 'guess', get_dummy_guess(stash=lambda: n)) + + with forge_stash(0): + self.assertEqual(stash(), None) + with forge_stash(1): + self.assertEqual(stash(), [{ + 'divider_highlight_group': 'stash:divider', + 'highlight_groups': ['stash'], + 'contents': '1' + }]) + with forge_stash(2): + self.assertEqual(stash(), [{ + 'divider_highlight_group': 'stash:divider', + 'highlight_groups': ['stash'], + 'contents': '2' + }]) + def test_file_vcs_status(self): pl = Pl() create_watcher = get_fallback_create_watcher()