19
from bzrlib import shelf
20
from bzrlib.tests import TestCaseWithTransport
21
from bzrlib.tests.script import ScriptRunner
19
from breezy import shelf
20
from breezy.tests import TestCaseWithTransport
21
from breezy.tests.script import ScriptRunner
24
24
class TestShelveList(TestCaseWithTransport):
85
85
sr = ScriptRunner()
86
86
sr.run_script(self, '''
89
$ bzr shelve --all -m Foo
89
$ brz shelve --all -m Foo
90
90
2>Selected changes:
92
92
2>Changes shelved with id "1".
96
96
2>Using changes with id "1".
99
99
2>All changes applied successfully.
106
107
class TestUnshelvePreview(TestCaseWithTransport):
108
109
def test_non_ascii(self):
109
110
"""Test that we can show a non-ascii diff that would result from unshelving"""
111
init_content = u'Initial: \u0418\u0437\u043d\u0430\u0447\n'.encode('utf-8')
112
init_content = u'Initial: \u0418\u0437\u043d\u0430\u0447\n'.encode(
112
114
more_content = u'More: \u0415\u0449\u0451\n'.encode('utf-8')
113
115
next_content = init_content + more_content
114
diff_part = '@@ -1,1 +1,2 @@\n %s+%s' % (init_content, more_content)
116
diff_part = b'@@ -1,1 +1,2 @@\n %s+%s' % (init_content, more_content)
116
118
tree = self.make_branch_and_tree('.')
117
119
self.build_tree_contents([('a_file', init_content)])
118
120
tree.add('a_file')
119
121
tree.commit(message='committed')
120
122
self.build_tree_contents([('a_file', next_content)])
121
123
self.run_bzr(['shelve', '--all'])
122
out, err = self.run_bzr(['unshelve', '--preview'], encoding='latin-1')
124
out, err = self.run_bzr_raw(
125
['unshelve', '--preview'], encoding='latin-1')
124
127
self.assertContainsString(out, diff_part)
145
148
def test_directory(self):
146
149
"""Test --directory option"""
147
150
tree = self.make_branch_and_tree('tree')
148
self.build_tree_contents([('tree/a', 'initial\n')])
151
self.build_tree_contents([('tree/a', b'initial\n')])
150
153
tree.commit(message='committed')
151
self.build_tree_contents([('tree/a', 'initial\nmore\n')])
154
self.build_tree_contents([('tree/a', b'initial\nmore\n')])
152
155
self.run_bzr('shelve -d tree --all')
153
self.assertFileEqual('initial\n', 'tree/a')
156
self.assertFileEqual(b'initial\n', 'tree/a')
154
157
self.run_bzr('unshelve --directory tree')
155
self.assertFileEqual('initial\nmore\n', 'tree/a')
158
self.assertFileEqual(b'initial\nmore\n', 'tree/a')