~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_library_state.py

• browse files at revision 2520.4.6
• compare with another revision
• download diff
• download tarball
• view history from revision 2520.4.6

• reverse the comparison (2520.4.6 to 6379.6.7)
• stop comparing with revision 6379.6.7

Show diffs side-by-side

added

removed

bzrlib/tests/test_library_state.py

1

 

# Copyright (C) 2010, 2011 Canonical Ltd

2

 

#

3

 

# This program is free software; you can redistribute it and/or modify

4

 

# it under the terms of the GNU General Public License as published by

5

 

# the Free Software Foundation; either version 2 of the License, or

6

 

# (at your option) any later version.

7

 

#

8

 

# This program is distributed in the hope that it will be useful,

9

 

# but WITHOUT ANY WARRANTY; without even the implied warranty of

10

 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

11

 

# GNU General Public License for more details.

12

 

#

13

 

# You should have received a copy of the GNU General Public License

14

 

# along with this program; if not, write to the Free Software

15

 

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

16

 

 

17

 

"""Tests for BzrLibraryState."""

18

 

 

19

 

from bzrlib import (

20

 

    library_state,

21

 

    tests,

22

 

    ui as _mod_ui

23

 

    )

24

 

from bzrlib.tests import fixtures

25

 

 

26

 

 

27

 

# TODO: once sufficiently cleaned up this should be able to be TestCase.

28

 

class TestLibraryState(tests.TestCaseWithTransport):

29

 

 

30

 

    def test_ui_is_used(self):

31

 

        ui = _mod_ui.SilentUIFactory()

32

 

        state = library_state.BzrLibraryState(

33

 

            ui=ui, trace=fixtures.RecordingContextManager())

34

 

        orig_ui = _mod_ui.ui_factory

35

 

        state.__enter__()

36

 

        try:

37

 

            self.assertEqual(ui, _mod_ui.ui_factory)

38

 

        finally:

39

 

            state.__exit__(None, None, None)

40

 

            self.assertEqual(orig_ui, _mod_ui.ui_factory)

41

 

 

42

 

    def test_trace_context(self):

43

 

        tracer = fixtures.RecordingContextManager()

44

 

        ui = _mod_ui.SilentUIFactory()

45

 

        state = library_state.BzrLibraryState(ui=ui, trace=tracer)

46

 

        state.__enter__()

47

 

        try:

48

 

            self.assertEqual(['__enter__'], tracer._calls)

49

 

        finally:

50

 

            state.__exit__(None, None, None)

51

 

            self.assertEqual(['__enter__', '__exit__'], tracer._calls)

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1