1
=== modified file 'tests/autopilot/music_app/tests/__init__.py'
2
--- tests/autopilot/music_app/tests/__init__.py	2014-03-07 20:10:34 +0000
3
+++ tests/autopilot/music_app/tests/__init__.py	2014-04-30 15:25:17 +0000
4
@@ -8,7 +8,6 @@
5
8
"""Music app autopilot tests."""
8
"""Music app autopilot tests."""
6
9
9
7
10
import tempfile
10
import tempfile
8
11
9
12
try:
11
try:
10
13
    from unittest import mock
12
    from unittest import mock
11
14
except ImportError:
13
except ImportError:
12
@@ -28,7 +27,8 @@
13
28
27
14
29
from ubuntuuitoolkit import (
28
from ubuntuuitoolkit import (
15
30
    base,
29
    base,
17
31
    emulators as toolkit_emulators
30
    emulators as toolkit_emulators,
18
31
    environment
19
32
)
32
)
20
33
33
21
34
34
22
@@ -50,9 +50,6 @@
23
50
    local_location_dir = os.path.dirname(os.path.dirname(working_dir))
50
    local_location_dir = os.path.dirname(os.path.dirname(working_dir))
24
51
    local_location = local_location_dir + "/music-app.qml"
51
    local_location = local_location_dir + "/music-app.qml"
25
52
    installed_location = "/usr/share/music-app/music-app.qml"
52
    installed_location = "/usr/share/music-app/music-app.qml"
26
53
    sqlite_dir = os.path.expanduser(
27
54
        "~/.local/share/com.ubuntu.music/Databases")
28
55
    backup_dir = sqlite_dir + ".backup"
29
56
53
30
57
    def setup_environment(self):
54
    def setup_environment(self):
31
58
        if os.path.exists(self.local_location):
55
        if os.path.exists(self.local_location):
32
@@ -67,15 +64,8 @@
33
67
        return launch, test_type
64
        return launch, test_type
34
68
65
35
69
    def setUp(self):
66
    def setUp(self):
36
70
        #backup and wipe db's before testing
37
71
        self.temp_move_sqlite_db()
38
72
        self.addCleanup(self.restore_sqlite_db)
39
73
40
74
        launch, self.test_type = self.setup_environment()
67
        launch, self.test_type = self.setup_environment()
45
75
        if self.test_type != 'click':
68
        self.home_dir = self._patch_home()
42
76
            self.home_dir = self._patch_home()
43
77
        else:
44
78
            self.home_dir = self._save_home()
46
79
        self._create_music_library()
69
        self._create_music_library()
47
80
        self.pointing_device = Pointer(self.input_device_class.create())
70
        self.pointing_device = Pointer(self.input_device_class.create())
48
81
        super(MusicTestCase, self).setUp()
71
        super(MusicTestCase, self).setUp()
49
@@ -103,25 +93,10 @@
50
103
            "com.ubuntu.music",
93
            "com.ubuntu.music",
51
104
            emulator_base=toolkit_emulators.UbuntuUIToolkitEmulatorBase)
94
            emulator_base=toolkit_emulators.UbuntuUIToolkitEmulatorBase)
52
105
95
53
106
    def _save_home(self):
54
107
        logger.debug('Saving HOME')
55
108
        home_dir = os.environ['HOME']
56
109
        backup_list = ('Music', )  # '.cache/mediascanner')
57
110
        backup_path = [os.path.join(home_dir, i) for i in backup_list]
58
111
        backups = [(i, '%s.bak' % i) for i in backup_path if os.path.exists(i)]
59
112
        for b in backups:
60
113
            logger.debug('backing up %s to %s' % b)
61
114
            try:
62
115
                shutil.rmtree(b[1])
63
116
            except:
64
117
                pass
65
118
            shutil.move(b[0], b[1])
66
119
            #self.addCleanup(shutil.move(b[1], b[0]))
67
120
        return home_dir
68
121
69
122
    def _patch_home(self):
96
    def _patch_home(self):
70
123
        #make a temp dir
97
        #make a temp dir
71
124
        temp_dir = tempfile.mkdtemp()
98
        temp_dir = tempfile.mkdtemp()
72
99
73
125
        #delete it, and recreate it to the length
100
        #delete it, and recreate it to the length
74
126
        #required so our patching the db works
101
        #required so our patching the db works
75
127
        #require a length of 25
102
        #require a length of 25
76
@@ -130,6 +105,7 @@
77
130
        os.mkdir(temp_dir)
105
        os.mkdir(temp_dir)
78
131
        logger.debug("Created fake home directory " + temp_dir)
106
        logger.debug("Created fake home directory " + temp_dir)
79
132
        self.addCleanup(shutil.rmtree, temp_dir)
107
        self.addCleanup(shutil.rmtree, temp_dir)
80
108
81
133
        #if the Xauthority file is in home directory
109
        #if the Xauthority file is in home directory
82
134
        #make sure we copy it to temp home, otherwise do nothing
110
        #make sure we copy it to temp home, otherwise do nothing
83
135
        xauth = os.path.expanduser(os.path.join('~', '.Xauthority'))
111
        xauth = os.path.expanduser(os.path.join('~', '.Xauthority'))
84
@@ -138,10 +114,17 @@
85
138
            shutil.copyfile(
114
            shutil.copyfile(
86
139
                os.path.expanduser(os.path.join('~', '.Xauthority')),
115
                os.path.expanduser(os.path.join('~', '.Xauthority')),
87
140
                os.path.join(temp_dir, '.Xauthority'))
116
                os.path.join(temp_dir, '.Xauthority'))
90
141
        patcher = mock.patch.dict('os.environ', {'HOME': temp_dir})
117
91
142
        patcher.start()
118
        #click can use initctl env (upstart), but desktop still requires mock
92
119
        if self.test_type == 'click':
93
120
            environment.set_initctl_env_var('HOME', temp_dir)
94
121
            self.addCleanup(environment.unset_initctl_env_var, 'HOME')
95
122
        else:
96
123
            patcher = mock.patch.dict('os.environ', {'HOME': temp_dir})
97
124
            patcher.start()
98
125
            self.addCleanup(patcher.stop)
99
126
100
143
        logger.debug("Patched home to fake home directory " + temp_dir)
127
        logger.debug("Patched home to fake home directory " + temp_dir)
101
144
        self.addCleanup(patcher.stop)
102
145
        return temp_dir
128
        return temp_dir
103
146
129
104
147
    def _create_music_library(self):
130
    def _create_music_library(self):
105
@@ -159,29 +142,19 @@
106
159
142
107
160
        logger.debug("Content dir set to %s" % content_dir)
143
        logger.debug("Content dir set to %s" % content_dir)
108
161
144
109
162
        #stop media scanner
110
163
        #if self.test_type == 'click':
111
164
        #    subprocess.check_call(['stop', 'mediascanner'])
112
165
113
166
        #copy content
145
        #copy content
114
167
        shutil.copy(os.path.join(content_dir, '1.ogg'), musicpath)
146
        shutil.copy(os.path.join(content_dir, '1.ogg'), musicpath)
115
168
        shutil.copy(os.path.join(content_dir, '2.ogg'), musicpath)
147
        shutil.copy(os.path.join(content_dir, '2.ogg'), musicpath)
116
169
        shutil.copy(os.path.join(content_dir, '3.mp3'), musicpath)
148
        shutil.copy(os.path.join(content_dir, '3.mp3'), musicpath)
120
170
        if self.test_type != 'click':
149
        shutil.copytree(
121
171
            shutil.copytree(os.path.join(content_dir, 'mediascanner'),
150
            os.path.join(content_dir, 'mediascanner'), mediascannerpath)
119
172
                            mediascannerpath)
122
173
151
123
174
        logger.debug("Music copied, files " + str(os.listdir(musicpath)))
152
        logger.debug("Music copied, files " + str(os.listdir(musicpath)))
124
175
153
134
176
        if self.test_type != 'click':
154
        self._patch_mediascanner_home(mediascannerpath)
135
177
            self._patch_mediascanner_home(mediascannerpath)
155
        logger.debug(
136
178
            logger.debug(
156
            "Mediascanner database copied, files " +
137
179
                "Mediascanner database copied, files " +
157
            str(os.listdir(mediascannerpath)))
129
180
                str(os.listdir(mediascannerpath)))
130
181
131
182
        #start media scanner
132
183
        #if self.test_type == 'click':
133
184
        #    subprocess.check_call(['start', 'mediascanner'])
138
185
158
139
186
    def _patch_mediascanner_home(self, mediascannerpath):
159
    def _patch_mediascanner_home(self, mediascannerpath):
140
187
        #do some inline db patching
160
        #do some inline db patching
141
@@ -204,10 +177,10 @@
142
204
        #replace all occurences of string find with string replace
177
        #replace all occurences of string find with string replace
143
205
        #in the given file
178
        #in the given file
144
206
        out_filename = in_filename + ".tmp"
179
        out_filename = in_filename + ".tmp"
149
207
        infile = open(in_filename, 'r')
180
        infile = open(in_filename, 'rb')
150
208
        outfile = open(out_filename, 'w')
181
        outfile = open(out_filename, 'wb')
151
209
        for s in infile.xreadlines():
182
        for line in infile:
152
210
            outfile.write(s.replace(find, replace))
183
            outfile.write(line.replace(str.encode(find), str.encode(replace)))
153
211
        infile.close()
184
        infile.close()
154
212
        outfile.close()
185
        outfile.close()
155
213
186
156
@@ -215,35 +188,6 @@
157
215
        os.remove(in_filename)
188
        os.remove(in_filename)
158
216
        os.rename(out_filename, in_filename)
189
        os.rename(out_filename, in_filename)
159
217
190
160
218
    def temp_move_sqlite_db(self):
161
219
        try:
162
220
            shutil.rmtree(self.backup_dir)
163
221
        except:
164
222
            pass
165
223
        else:
166
224
            logger.warning("Prexisting backup database found and removed")
167
225
168
226
        try:
169
227
            shutil.move(self.sqlite_dir, self.backup_dir)
170
228
        except:
171
229
            logger.warning("No current database found")
172
230
        else:
173
231
            logger.debug("Backed up database")
174
232
175
233
    def restore_sqlite_db(self):
176
234
        if os.path.exists(self.backup_dir):
177
235
            if os.path.exists(self.sqlite_dir):
178
236
                try:
179
237
                    shutil.rmtree(self.sqlite_dir)
180
238
                except:
181
239
                    logger.error("Failed to remove test database and restore" /
182
240
                                 "database")
183
241
                    return
184
242
            try:
185
243
                shutil.move(self.backup_dir, self.sqlite_dir)
186
244
            except:
187
245
                logger.error("Failed to restore database")
188
246
189
247
    @property
191
    @property
190
248
    def player(self):
192
    def player(self):
191
249
        return self.main_view.get_player()
193
        return self.main_view.get_player()
Reviewer	Review Type	Date Requested	Status
Victor Thompson		2014-03-18	Approve on 2014-04-30
Ubuntu Phone Apps Jenkins Bot	continuous-integration		Approve on 2014-04-30
Review via email: mp+211637@code.launchpad.net