← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/tests/test_controllers.py

  • Committer: Michael Hudson
  • Date: 2008-09-12 04:36:13 UTC
  • Revision ID: michael.hudson@canonical.com-20080912043613-d2550i1udbl2g86b
our tests aren't the greatest, but let's keep them passing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
loggerhead/tests/test_controllers.py
1
 
from loggerhead.apps.branch import BranchWSGIApp
2
 
from loggerhead.controllers.annotate_ui import AnnotateUI
3
 
from loggerhead.controllers.inventory_ui import InventoryUI
4
 
from loggerhead.controllers.revision_ui import RevisionUI
5
 
from loggerhead.tests.test_simple import BasicTests
6
 
from loggerhead import util
7
 
 
8
 
 
9
 
class TestInventoryUI(BasicTests):
10
 
 
11
 
    def make_bzrbranch_and_inventory_ui_for_tree_shape(self, shape):
12
 
        tree = self.make_branch_and_tree('.')
13
 
        self.build_tree(shape)
14
 
        tree.smart_add([])
15
 
        tree.commit('')
16
 
        tree.branch.lock_read()
17
 
        self.addCleanup(tree.branch.unlock)
18
 
        branch_app = BranchWSGIApp(tree.branch)
19
 
        return tree.branch, InventoryUI(branch_app, branch_app.get_history)
20
 
 
21
 
    def test_get_filelist(self):
22
 
        bzrbranch, inv_ui = self.make_bzrbranch_and_inventory_ui_for_tree_shape(
23
 
            ['filename'])
24
 
        inv = bzrbranch.repository.get_inventory(bzrbranch.last_revision())
25
 
        self.assertEqual(1, len(inv_ui.get_filelist(inv, '', 'filename')))
26
 
 
27
 
 
28
 
class TestRevisionUI(BasicTests):
29
 
 
30
 
    def make_bzrbranch_and_revision_ui_for_tree_shapes(self, shape1, shape2):
31
 
        tree = self.make_branch_and_tree('.')
32
 
        self.build_tree_contents(shape1)
33
 
        tree.smart_add([])
34
 
        tree.commit('')
35
 
        self.build_tree_contents(shape2)
36
 
        tree.smart_add([])
37
 
        tree.commit('')
38
 
        tree.branch.lock_read()
39
 
        self.addCleanup(tree.branch.unlock)
40
 
        branch_app = BranchWSGIApp(tree.branch)
41
 
        branch_app._environ = {
42
 
            'wsgi.url_scheme':'',
43
 
            'SERVER_NAME':'',
44
 
            'SERVER_PORT':'80',
45
 
            }
46
 
        branch_app._url_base = ''
47
 
        branch_app.friendly_name = ''
48
 
        return tree.branch, RevisionUI(branch_app, branch_app.get_history)
49
 
 
50
 
    def test_get_values(self):
51
 
        branch, rev_ui = self.make_bzrbranch_and_revision_ui_for_tree_shapes(
52
 
            [], [])
53
 
        rev_ui.args = ['2']
54
 
        util.set_context({})
55
 
        self.assertIsInstance(
56
 
            rev_ui.get_values('', {}, []),
57
 
            dict)
58
 
 
59
 
 
60
 
class TestAnnotateUI(BasicTests):
61
 
 
62
 
    def make_annotate_ui_for_file_history(self, file_id, rev_ids_texts):
63
 
        tree = self.make_branch_and_tree('.')
64
 
        open('filename', 'w').write('')
65
 
        tree.add(['filename'], [file_id])
66
 
        for rev_id, text in rev_ids_texts:
67
 
            open('filename', 'w').write(text)
68
 
            tree.commit(rev_id=rev_id, message='.')
69
 
        tree.branch.lock_read()
70
 
        self.addCleanup(tree.branch.unlock)
71
 
        branch_app = BranchWSGIApp(tree.branch)
72
 
        return AnnotateUI(branch_app, branch_app.get_history)
73
 
 
74
 
    def test_annotate_file(self):
75
 
        history = [('rev1', 'old\nold\n'), ('rev2', 'new\nold\n')]
76
 
        ann_ui = self.make_annotate_ui_for_file_history('file_id', history)
77
 
        annotated = list(ann_ui.annotate_file('file_id', 'rev2'))
78
 
        self.assertEqual(2, len(annotated))
79
 
        self.assertEqual('2', annotated[0].change.revno)
80
 
        self.assertEqual('1', annotated[1].change.revno)