test/stash_model_test.py

   1 from cola.models.stash import StashModel
   2
   3 from . import helper
   4 from .helper import app_context
   5
   6
   7 # Prevent unused imports lint errors.
   8 assert app_context is not None
   9
  10
  11 def test_stash_info_for_message_without_slash(app_context):
  12     helper.commit_files()
  13     helper.write_file('A', 'change')
  14     helper.run_git('stash', 'save', 'some message')
  15     assert StashModel(app_context).stash_info()[0] == [
  16         r'stash@{0}: On main: some message'
  17     ]
  18
  19
  20 def test_stash_info_for_message_with_slash(app_context):
  21     helper.commit_files()
  22     helper.write_file('A', 'change')
  23     helper.run_git('stash', 'save', 'some message/something')
  24     model = StashModel(app_context)
  25     stash_details = model.stash_info()[0]
  26     assert stash_details == [r'stash@{0}: On main: some message/something']
  27
  28
  29 def test_stash_info_on_branch_with_slash(app_context):
  30     helper.commit_files()
  31     helper.run_git('checkout', '-b', 'feature/a')
  32     helper.write_file('A', 'change')
  33     helper.run_git('stash', 'save', 'some message')
  34
  35     model = StashModel(app_context)
  36     stash_info = model.stash_info()
  37
  38     stash_details = stash_info[0][0]
  39     assert stash_details in (
  40         'stash@{0}: On feature/a: some message',
  41         # Some versions of Git do not report the full branch name
  42         'stash@{0}: On a: some message',
  43     )
  44
  45     stash_rev = stash_info[1][0]
  46     assert stash_rev == r'stash@{0}'
  47
  48     stash_message = stash_info[3][0]
  49     assert stash_message in (
  50         'On feature/a: some message',
  51         'On a: some message',
  52     )