Skip to content

Commit

Permalink
pep8 changes
Browse files Browse the repository at this point in the history
Signed-off-by: Joe Julian <[email protected]>
  • Loading branch information
Joe Julian committed Nov 18, 2015
1 parent cb18a10 commit 9cbe132
Showing 1 changed file with 30 additions and 17 deletions.
47 changes: 30 additions & 17 deletions qa/workunits/ceph-disk/ceph-disk-test.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@

LOG = logging.getLogger('CephDisk')


class CephDisk:

def __init__(self):
Expand All @@ -64,10 +65,12 @@ def sh(command):
return output.strip()

def unused_disks(self, pattern='[vs]d.'):
names = filter(lambda x: re.match(pattern, x), os.listdir("/sys/block"))
names = filter(
lambda x: re.match(pattern, x), os.listdir("/sys/block"))
if not names:
return []
disks = json.loads(self.sh("ceph-disk list --format json " + " ".join(names)))
disks = json.loads(
self.sh("ceph-disk list --format json " + " ".join(names)))
unused = []
for disk in disks:
if 'partitions' not in disk:
Expand Down Expand Up @@ -106,9 +109,11 @@ def get_journal_partition(self, uuid):
for partition in disk['partitions']:
if partition['path'] == journal_dev:
if 'journal_for' in partition:
assert partition['journal_for'] == data_partition['path']
assert partition[
'journal_for'] == data_partition['path']
return partition
raise Exception("journal for uuid = " + uuid + " not found in " + str(disks))
raise Exception(
"journal for uuid = " + uuid + " not found in " + str(disks))

def destroy_osd(self, uuid):
id = self.sh("ceph osd create " + uuid)
Expand Down Expand Up @@ -160,6 +165,7 @@ def check_osd_status(self, uuid, have_journal=False):
journal_partition = self.get_journal_partition(uuid)
assert journal_partition


class TestCephDisk(object):

def setup_class(self):
Expand All @@ -179,7 +185,7 @@ def setup(self):

def test_deactivate_reactivate_osd(self):
c = CephDisk()
have_journal=True
have_journal = True
disk = c.unused_disks()[0]
osd_uuid = str(uuid.uuid1())
c.sh("ceph-disk zap " + disk)
Expand Down Expand Up @@ -258,7 +264,6 @@ def activate_reactivate_dmcrypt(self, type):
c.check_osd_status(osd_uuid, have_journal)
c.destroy_osd(osd_uuid)


def test_activate_dmcrypt_plain(self):
c = CephDisk()
c.conf['global']['osd dmcrypt type'] = 'plain'
Expand Down Expand Up @@ -326,7 +331,7 @@ def test_activate_with_journal_dev_is_symlink(self):
disk = c.unused_disks()[0]
osd_uuid = str(uuid.uuid1())
tempdir = tempfile.mkdtemp()
symlink = os.path.join(tempdir,'osd')
symlink = os.path.join(tempdir, 'osd')
os.symlink(disk, symlink)
c.sh("ceph-disk zap " + symlink)
c.sh("ceph-disk prepare --osd-uuid " + osd_uuid +
Expand All @@ -351,7 +356,7 @@ def test_activate_separated_journal(self):
data_disk = disks[0]
journal_disk = disks[1]
osd_uuid = self.activate_separated_journal(data_disk, journal_disk)
c.helper("pool_read_write 1") # 1 == pool size
c.helper("pool_read_write 1") # 1 == pool size
c.destroy_osd(osd_uuid)
c.sh("ceph-disk zap " + data_disk + " " + journal_disk)

Expand Down Expand Up @@ -381,7 +386,8 @@ def activate_separated_journal(self, data_disk, journal_disk):
c.sh("ceph-disk prepare --osd-uuid " + osd_uuid +
" " + data_disk + " " + journal_disk)
c.wait_for_osd_up(osd_uuid)
device = json.loads(c.sh("ceph-disk list --format json " + data_disk))[0]
device = json.loads(
c.sh("ceph-disk list --format json " + data_disk))[0]
assert len(device['partitions']) == 1
c.check_osd_status(osd_uuid, have_journal)
return osd_uuid
Expand All @@ -400,15 +406,17 @@ def test_activate_two_separated_journal(self):
other_data_disk = disks[1]
journal_disk = disks[2]
osd_uuid = self.activate_separated_journal(data_disk, journal_disk)
other_osd_uuid = self.activate_separated_journal(other_data_disk, journal_disk)
other_osd_uuid = self.activate_separated_journal(
other_data_disk, journal_disk)
#
# read/write can only succeed if the two osds are up because
# the pool needs two OSD
#
c.helper("pool_read_write 2") # 2 == pool size
c.helper("pool_read_write 2") # 2 == pool size
c.destroy_osd(osd_uuid)
c.destroy_osd(other_osd_uuid)
c.sh("ceph-disk zap " + data_disk + " " + journal_disk + " " + other_data_disk)
c.sh("ceph-disk zap " + data_disk + " " +
journal_disk + " " + other_data_disk)

#
# Create an OSD and reuse an existing journal partition
Expand All @@ -432,9 +440,10 @@ def test_activate_reuse_journal(self):
#
c.sh("ceph-disk prepare --osd-uuid " + osd_uuid +
" " + data_disk + " " + journal_path)
c.helper("pool_read_write 1") # 1 == pool size
c.helper("pool_read_write 1") # 1 == pool size
c.wait_for_osd_up(osd_uuid)
device = json.loads(c.sh("ceph-disk list --format json " + data_disk))[0]
device = json.loads(
c.sh("ceph-disk list --format json " + data_disk))[0]
assert len(device['partitions']) == 1
c.check_osd_status(osd_uuid)
journal_partition = c.get_journal_partition(osd_uuid)
Expand All @@ -448,15 +457,17 @@ def test_activate_reuse_journal(self):
def test_activate_multipath(self):
c = CephDisk()
if c.sh("lsb_release -si") != 'CentOS':
pytest.skip("see issue https://bugs.launchpad.net/ubuntu/+source/multipath-tools/+bug/1488688")
pytest.skip(
"see issue https://bugs.launchpad.net/ubuntu/+source/multipath-tools/+bug/1488688")
c.ensure_sd()
#
# Figure out the name of the multipath device
#
disk = c.unused_disks('sd.')[0]
c.sh("mpathconf --enable || true")
c.sh("multipath " + disk)
holders = os.listdir("/sys/block/" + os.path.basename(disk) + "/holders")
holders = os.listdir(
"/sys/block/" + os.path.basename(disk) + "/holders")
assert 1 == len(holders)
name = open("/sys/block/" + holders[0] + "/dm/name").read()
multipath = "/dev/mapper/" + name
Expand All @@ -468,7 +479,8 @@ def test_activate_multipath(self):
c.sh("ceph-disk prepare --osd-uuid " + osd_uuid +
" " + multipath)
c.wait_for_osd_up(osd_uuid)
device = json.loads(c.sh("ceph-disk list --format json " + multipath))[0]
device = json.loads(
c.sh("ceph-disk list --format json " + multipath))[0]
assert len(device['partitions']) == 2
data_partition = c.get_osd_partition(osd_uuid)
assert data_partition['type'] == 'data'
Expand All @@ -481,6 +493,7 @@ def test_activate_multipath(self):
c.sh("multipath -F")
c.unload_scsi_debug()


class CephDiskTest(CephDisk):

def main(self, argv):
Expand Down

0 comments on commit 9cbe132

Please sign in to comment.