← Back to branch summary

~xubuntu-dev/ubiquity/lp1437180_feh

« back to all changes in this revision

Viewing changes to tests/test_misc.py

  • Committer: Colin Watson
  • Date: 2012-09-07 11:32:25 UTC
  • Revision ID: cjwatson@canonical.com-20120907113225-glzjy68vf5qru6i4
Remove Python 2 support.  We aren't going back now.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_misc.py
1
1
#!/usr/bin/python3
2
2
 
3
3
import grp
 
4
import io
4
5
import os
5
6
import pwd
6
 
try:
7
 
    from test.support import EnvironmentVarGuard, run_unittest
8
 
except ImportError:
9
 
    from test.test_support import EnvironmentVarGuard, run_unittest
 
7
from test.support import EnvironmentVarGuard, run_unittest
10
8
import unittest
11
9
 
12
10
# These tests require Mock 0.7.0
13
11
import mock
14
12
 
15
13
from ubiquity import misc
16
 
from helpers import builtin_patch, text_file_type
17
14
 
18
15
_proc_swaps = [
19
16
    'Filename\t\t\t\tType\t\tSize\tUsed\tPriority',
59
56
    def setUp(self):
60
57
        misc.get_release.release_info = None
61
58
 
62
 
    @builtin_patch('open')
 
59
    @mock.patch('builtins.open')
63
60
    def test_is_swap(self, mock_open):
64
 
        magic = mock.MagicMock(spec=text_file_type)
 
61
        magic = mock.MagicMock(spec=io.TextIOBase)
65
62
        mock_open.return_value = magic
66
63
        magic.__enter__.return_value = magic
67
64
        magic.__iter__.return_value = iter(_proc_swaps)
68
65
        self.assertTrue(misc.is_swap('/dev/sda5'))
69
66
        self.assertFalse(misc.is_swap('/dev/sda'))
70
67
 
71
 
    @builtin_patch('open')
 
68
    @mock.patch('builtins.open')
72
69
    def test_is_swap_fail(self, mock_open):
73
70
        mock_open.side_effect = Exception('Ka-blamo!')
74
71
        self.assertFalse(misc.is_swap('/dev/sda5'))
94
91
        # os.access, raise an exception as a side effect.
95
92
        pass
96
93
 
97
 
    @builtin_patch('open')
 
94
    @mock.patch('builtins.open')
98
95
    def test_get_release(self, mock_open):
99
 
        magic = mock.MagicMock(spec=text_file_type)
 
96
        magic = mock.MagicMock(spec=io.TextIOBase)
100
97
        magic.__enter__.return_value = magic
101
98
        mock_open.return_value = magic
102
99
        magic.readline.return_value = _disk_info
104
101
        self.assertEqual(release.name, 'Ubuntu-Server')
105
102
        self.assertEqual(release.version, '10.04.1 LTS')
106
103
 
107
 
    @builtin_patch('open')
 
104
    @mock.patch('builtins.open')
108
105
    def test_get_release_fail(self, mock_open):
109
106
        mock_open.side_effect = Exception('Pow!')
110
107
        release = misc.get_release()
128
125
 
129
126
    #    self.assertEqual(misc.windows_startup_folder('/tmp/tmp.XXXXXX'))
130
127
 
131
 
    @builtin_patch('open')
 
128
    @mock.patch('builtins.open')
132
129
    def test_mount_info(self, mock_open):
133
 
        magic = mock.MagicMock(spec=text_file_type)
 
130
        magic = mock.MagicMock(spec=io.TextIOBase)
134
131
        magic.__enter__.return_value = magic
135
132
        mock_open.return_value = magic
136
133
        # TODO come up with better mountpoints.