1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
# Copyright (C) 2011 Curtis Hovey <sinzui.is@verizon.net>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Test the BranchTreeModel functionality."""
from time import (
strftime,
localtime,
)
from bzrlib import (
tests,
)
from bzrlib.plugins.gtk.branchview.treemodel import BranchTreeModel
class BranchTreeModelTestCase(tests.TestCaseWithMemoryTransport):
def make_test_branch(self, revid, message=None, committer=None):
builder = self.make_branch_builder('test')
builder.build_snapshot(
revid, None, [('add', ('', 'root-id', 'directory', None))],
message=message, committer=committer)
return builder.get_branch()
def test_init(self):
branch = self.make_test_branch('A')
revision = branch.repository.get_revision('A')
branch.tags.set_tag('2.0', revision.revision_id)
model = BranchTreeModel(branch, [])
self.assertEqual(branch, model.branch)
self.assertEqual(branch.repository, model.repository)
self.assertEqual({'A': [u'2.0']}, model.tags)
self.assertEqual({}, model.revisions)
self.assertEqual([], model.line_graph_data)
def test_add_tag_create(self):
branch = self.make_test_branch('A')
revision = branch.repository.get_revision('A')
model = BranchTreeModel(branch, [])
model.add_tag('2.0', revision.revision_id)
self.assertEqual({'A': [u'2.0']}, model.tags)
def test_add_tag_append(self):
branch = self.make_test_branch('A')
revision = branch.repository.get_revision('A')
branch.tags.set_tag('2.0', revision.revision_id)
model = BranchTreeModel(branch, [])
model.add_tag('2.1', revision.revision_id)
self.assertEqual({'A': ['2.0', '2.1']}, model.tags)
def test_line_graph_item_to_model_row(self):
branch = self.make_test_branch(
'A', message='badger', committer='fnord')
revision = branch.repository.get_revision('A')
branch.tags.set_tag('2.0', revision.revision_id)
model = BranchTreeModel(branch, [])
data_item = (
revision.revision_id, (1, 0), (5, 6, 0.5), ('B'), ('C'), [12, 34])
row = model._line_graph_item_to_model_row(0, data_item)
self.assertEqual(revision.revision_id, row[0], 'Wrong revid.')
self.assertEqual((1, 0), row[1], 'Wrong node.')
self.assertEqual((5, 6, 0.5), row[2], 'Wrong lines.')
self.assertEqual([], row[3], 'Wrong last_lines.')
self.assertEqual('12.34', row[4], 'Wrong revno.')
self.assertEqual('badger', row[5], 'Wrong summary.')
self.assertEqual('badger', row[6], 'Wrong message.')
self.assertEqual('fnord', row[7], 'wrong committer.')
self.assertEqual(
strftime("%Y-%m-%d %H:%M", localtime(revision.timestamp)),
row[8], 'Wrong timestamp.')
self.assertEqual(revision, row[9], 'Wrong revision.')
self.assertEqual(('B'), row[10], 'Wrong parents.')
self.assertEqual(('C'), row[11], 'Wrong children.')
self.assertEqual(['2.0'], row[12], 'Wrong tags.')
self.assertEqual('fnord', row[13], 'Wrong authors.')
def test_set_line_graph_data(self):
branch = self.make_test_branch(
'A', message='badger', committer='fnord')
rev_id = branch.repository.get_revision('A').revision_id
model = BranchTreeModel(branch, [])
data = [(rev_id, (2, 0), (3, 4, 0.5), None, None, [1, 2])]
model.set_line_graph_data(data)
self.assertEqual(data, model.line_graph_data)
tree_iter = model.get_iter_first()
self.assertEqual(rev_id, model.get_value(tree_iter, 0))
|