← Back to branch summary

~cloud-init-dev/cloud-init/trunk

« back to all changes in this revision

Viewing changes to tests/unittests/test_datasource/test_altcloud.py

Enable flake8 and fix a large amount of reported issues

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/unittests/test_datasource/test_altcloud.py
134
134
        '''
135
135
        util.read_dmi_data = _dmi_data('RHEV')
136
136
        dsrc = DataSourceAltCloud({}, None, self.paths)
137
 
        self.assertEquals('RHEV', dsrc.get_cloud_type())
 
137
        self.assertEqual('RHEV', dsrc.get_cloud_type())
138
138
 
139
139
    def test_vsphere(self):
140
140
        '''
143
143
        '''
144
144
        util.read_dmi_data = _dmi_data('VMware Virtual Platform')
145
145
        dsrc = DataSourceAltCloud({}, None, self.paths)
146
 
        self.assertEquals('VSPHERE', dsrc.get_cloud_type())
 
146
        self.assertEqual('VSPHERE', dsrc.get_cloud_type())
147
147
 
148
148
    def test_unknown(self):
149
149
        '''
152
152
        '''
153
153
        util.read_dmi_data = _dmi_data('Unrecognized Platform')
154
154
        dsrc = DataSourceAltCloud({}, None, self.paths)
155
 
        self.assertEquals('UNKNOWN', dsrc.get_cloud_type())
 
155
        self.assertEqual('UNKNOWN', dsrc.get_cloud_type())
156
156
 
157
157
 
158
158
class TestGetDataCloudInfoFile(TestCase):
187
187
        _write_cloud_info_file('RHEV')
188
188
        dsrc = DataSourceAltCloud({}, None, self.paths)
189
189
        dsrc.user_data_rhevm = lambda: True
190
 
        self.assertEquals(True, dsrc.get_data())
 
190
        self.assertEqual(True, dsrc.get_data())
191
191
 
192
192
    def test_vsphere(self):
193
193
        '''Success Test module get_data() forcing VSPHERE.'''
195
195
        _write_cloud_info_file('VSPHERE')
196
196
        dsrc = DataSourceAltCloud({}, None, self.paths)
197
197
        dsrc.user_data_vsphere = lambda: True
198
 
        self.assertEquals(True, dsrc.get_data())
 
198
        self.assertEqual(True, dsrc.get_data())
199
199
 
200
200
    def test_fail_rhev(self):
201
201
        '''Failure Test module get_data() forcing RHEV.'''
203
203
        _write_cloud_info_file('RHEV')
204
204
        dsrc = DataSourceAltCloud({}, None, self.paths)
205
205
        dsrc.user_data_rhevm = lambda: False
206
 
        self.assertEquals(False, dsrc.get_data())
 
206
        self.assertEqual(False, dsrc.get_data())
207
207
 
208
208
    def test_fail_vsphere(self):
209
209
        '''Failure Test module get_data() forcing VSPHERE.'''
211
211
        _write_cloud_info_file('VSPHERE')
212
212
        dsrc = DataSourceAltCloud({}, None, self.paths)
213
213
        dsrc.user_data_vsphere = lambda: False
214
 
        self.assertEquals(False, dsrc.get_data())
 
214
        self.assertEqual(False, dsrc.get_data())
215
215
 
216
216
    def test_unrecognized(self):
217
217
        '''Failure Test module get_data() forcing unrecognized.'''
218
218
 
219
219
        _write_cloud_info_file('unrecognized')
220
220
        dsrc = DataSourceAltCloud({}, None, self.paths)
221
 
        self.assertEquals(False, dsrc.get_data())
 
221
        self.assertEqual(False, dsrc.get_data())
222
222
 
223
223
 
224
224
class TestGetDataNoCloudInfoFile(TestCase):
250
250
        util.read_dmi_data = _dmi_data('RHEV Hypervisor')
251
251
        dsrc = DataSourceAltCloud({}, None, self.paths)
252
252
        dsrc.user_data_rhevm = lambda: True
253
 
        self.assertEquals(True, dsrc.get_data())
 
253
        self.assertEqual(True, dsrc.get_data())
254
254
 
255
255
    def test_vsphere_no_cloud_file(self):
256
256
        '''Test No cloud info file module get_data() forcing VSPHERE.'''
258
258
        util.read_dmi_data = _dmi_data('VMware Virtual Platform')
259
259
        dsrc = DataSourceAltCloud({}, None, self.paths)
260
260
        dsrc.user_data_vsphere = lambda: True
261
 
        self.assertEquals(True, dsrc.get_data())
 
261
        self.assertEqual(True, dsrc.get_data())
262
262
 
263
263
    def test_failure_no_cloud_file(self):
264
264
        '''Test No cloud info file module get_data() forcing unrecognized.'''
265
265
 
266
266
        util.read_dmi_data = _dmi_data('Unrecognized Platform')
267
267
        dsrc = DataSourceAltCloud({}, None, self.paths)
268
 
        self.assertEquals(False, dsrc.get_data())
 
268
        self.assertEqual(False, dsrc.get_data())
269
269
 
270
270
 
271
271
class TestUserDataRhevm(TestCase):
305
305
 
306
306
        dsrc = DataSourceAltCloud({}, None, self.paths)
307
307
 
308
 
        self.assertEquals(False, dsrc.user_data_rhevm())
 
308
        self.assertEqual(False, dsrc.user_data_rhevm())
309
309
 
310
310
    def test_modprobe_fails(self):
311
311
        '''Test user_data_rhevm() where modprobe fails.'''
315
315
 
316
316
        dsrc = DataSourceAltCloud({}, None, self.paths)
317
317
 
318
 
        self.assertEquals(False, dsrc.user_data_rhevm())
 
318
        self.assertEqual(False, dsrc.user_data_rhevm())
319
319
 
320
320
    def test_no_modprobe_cmd(self):
321
321
        '''Test user_data_rhevm() with no modprobe command.'''
325
325
 
326
326
        dsrc = DataSourceAltCloud({}, None, self.paths)
327
327
 
328
 
        self.assertEquals(False, dsrc.user_data_rhevm())
 
328
        self.assertEqual(False, dsrc.user_data_rhevm())
329
329
 
330
330
    def test_udevadm_fails(self):
331
331
        '''Test user_data_rhevm() where udevadm fails.'''
335
335
 
336
336
        dsrc = DataSourceAltCloud({}, None, self.paths)
337
337
 
338
 
        self.assertEquals(False, dsrc.user_data_rhevm())
 
338
        self.assertEqual(False, dsrc.user_data_rhevm())
339
339
 
340
340
    def test_no_udevadm_cmd(self):
341
341
        '''Test user_data_rhevm() with no udevadm command.'''
345
345
 
346
346
        dsrc = DataSourceAltCloud({}, None, self.paths)
347
347
 
348
 
        self.assertEquals(False, dsrc.user_data_rhevm())
 
348
        self.assertEqual(False, dsrc.user_data_rhevm())
349
349
 
350
350
 
351
351
class TestUserDataVsphere(TestCase):
380
380
 
381
381
        dsrc = DataSourceAltCloud({}, None, self.paths)
382
382
 
383
 
        self.assertEquals(False, dsrc.user_data_vsphere())
 
383
        self.assertEqual(False, dsrc.user_data_vsphere())
384
384
 
385
385
 
386
386
class TestReadUserDataCallback(TestCase):
408
408
    def test_callback_both(self):
409
409
        '''Test read_user_data_callback() with both files.'''
410
410
 
411
 
        self.assertEquals('test user data',
412
 
                          read_user_data_callback(self.mount_dir))
 
411
        self.assertEqual('test user data',
 
412
                         read_user_data_callback(self.mount_dir))
413
413
 
414
414
    def test_callback_dc(self):
415
415
        '''Test read_user_data_callback() with only DC file.'''
418
418
                                dc_file=False,
419
419
                                non_dc_file=True)
420
420
 
421
 
        self.assertEquals('test user data',
422
 
                          read_user_data_callback(self.mount_dir))
 
421
        self.assertEqual('test user data',
 
422
                         read_user_data_callback(self.mount_dir))
423
423
 
424
424
    def test_callback_non_dc(self):
425
425
        '''Test read_user_data_callback() with only non-DC file.'''
428
428
                                dc_file=True,
429
429
                                non_dc_file=False)
430
430
 
431
 
        self.assertEquals('test user data',
432
 
                          read_user_data_callback(self.mount_dir))
 
431
        self.assertEqual('test user data',
 
432
                         read_user_data_callback(self.mount_dir))
433
433
 
434
434
    def test_callback_none(self):
435
435
        '''Test read_user_data_callback() no files are found.'''
436
436
 
437
437
        _remove_user_data_files(self.mount_dir)
438
 
        self.assertEquals(None, read_user_data_callback(self.mount_dir))
 
438
        self.assertEqual(None, read_user_data_callback(self.mount_dir))
439
439
 
440
440
 
441
441
def force_arch(arch=None):