Skip to content

Instantly share code, notes, and snippets.

@funkytaco

funkytaco/test.py

Forked from michaelgugino/test.py
Created June 2, 2020 19:15
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save funkytaco/2dbd3d636cfbd200dcc4c4430ff6e113 to your computer and use it in GitHub Desktop.
Save funkytaco/2dbd3d636cfbd200dcc4c4430ff6e113 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
test.py
import os
import sys
import pytest
try:
# python3, mock is built in.
from unittest.mock import patch
except ImportError:
# In python2, mock is installed via pip.
from mock import patch
MODULE_PATH = os.path.realpath(os.path.join(__file__, os.pardir, os.pardir, 'library'))
sys.path.insert(1, MODULE_PATH)
import glusterfs_check_cns # noqa
NODE_LIST_STD_OUT_1 = ("""
NAME STATUS ROLES AGE VERSION
fedora1.openshift.io Ready compute,infra,master 1d v1.11.0+d4cacc0
fedora2.openshift.io Ready infra 1d v1.11.0+d4cacc0
fedora3.openshift.io Ready infra 1d v1.11.0+d4cacc0
""")
NODE_LIST_STD_OUT_2 = ("""
NAME STATUS ROLES AGE VERSION
fedora1.openshift.io Ready compute,infra,master 1d v1.11.0+d4cacc0
fedora2.openshift.io NotReady infra 1d v1.11.0+d4cacc0
fedora3.openshift.io Ready infra 1d v1.11.0+d4cacc0
""")
NODE_LIST_STD_OUT_3 = ("""
NAME STATUS ROLES AGE VERSION
fedora1.openshift.io Ready compute,infra,master 1d v1.11.0+d4cacc0
fedora2.openshift.io NotReady infra 1d v1.11.0+d4cacc0
fedora3.openshift.io Invalid infra 1d v1.11.0+d4cacc0
""")
POD_SELECT_STD_OUT = ("""NAME READY STATUS RESTARTS AGE IP NODE
glusterblock-storage-provisioner-dc-1-ks5zt 1/1 Running 0 1d 10.130.0.5 fedora3.openshift.io
glusterfs-storage-fzdn2 1/1 Running 0 1d 192.168.124.175 fedora1.openshift.io
glusterfs-storage-mp9nk 1/1 Running 4 1d 192.168.124.233 fedora2.openshift.io
glusterfs-storage-t9c6d 1/1 Running 0 1d 192.168.124.50 fedora3.openshift.io
heketi-storage-1-rgj8b 1/1 Running 0 1d 10.130.0.4 fedora3.openshift.io""")
# Need to ensure we have extra empty lines in this output;
# thus the quotes are one line above and below the text.
VOLUME_LIST_STDOUT = ("""
heketidbstorage
volume1
""")
VOLUME_HEAL_INFO_GOOD = ("""
Brick 192.168.124.233:/var/lib/heketi/mounts/vg_936ddf24061d55788f50496757d2f3b2/brick_9df1b6229025ea45521ab1b370d24a06/brick
Status: Connected
Number of entries: 0
Brick 192.168.124.175:/var/lib/heketi/mounts/vg_95975e77a6dc7a8e45586eac556b0f24/brick_172b6be6704a3d9f706535038f7f2e52/brick
Status: Connected
Number of entries: 0
Brick 192.168.124.50:/var/lib/heketi/mounts/vg_6523756fe1becfefd3224d3082373344/brick_359e4cf44cd1b82674f7d931cb5c481e/brick
Status: Connected
Number of entries: 0
""")
VOLUME_HEAL_INFO_BAD = ("""
Brick 192.168.124.233:/var/lib/heketi/mounts/vg_936ddf24061d55788f50496757d2f3b2/brick_9df1b6229025ea45521ab1b370d24a06/brick
Status: Connected
Number of entries: 0
Brick 192.168.124.175:/var/lib/heketi/mounts/vg_95975e77a6dc7a8e45586eac556b0f24/brick_172b6be6704a3d9f706535038f7f2e52/brick
Status: Connected
Number of entries: 0
Brick 192.168.124.50:/var/lib/heketi/mounts/vg_6523756fe1becfefd3224d3082373344/brick_359e4cf44cd1b82674f7d931cb5c481e/brick
Status: Connected
Number of entries: -
""")
class DummyModule(object):
def exit_json(*args, **kwargs):
return 0
def fail_json(*args, **kwargs):
raise Exception(kwargs['msg'])
@patch('glusterfs_check_cns.call_or_fail')
def test_get_valid_nodes(call_mock):
module = DummyModule()
oc_exec = []
exclude_node = "fedora1.openshift.io"
call_mock.return_value = NODE_LIST_STD_OUT_1
valid_nodes = glusterfs_check_cns.get_valid_nodes(module, oc_exec, exclude_node)
assert valid_nodes == ['fedora2.openshift.io', 'fedora3.openshift.io']
call_mock.return_value = NODE_LIST_STD_OUT_2
valid_nodes = glusterfs_check_cns.get_valid_nodes(module, oc_exec, exclude_node)
assert valid_nodes == ['fedora3.openshift.io']
call_mock.return_value = NODE_LIST_STD_OUT_3
with pytest.raises(Exception) as err:
valid_nodes = glusterfs_check_cns.get_valid_nodes(module, oc_exec, exclude_node)
assert 'Exception: Unable to find suitable node in get nodes output' in str(err)
@patch('glusterfs_check_cns.call_or_fail')
def test_select_pod(call_mock):
module = DummyModule()
oc_exec = []
cluster_name = "storage"
valid_nodes = ["fedora2.openshift.io", "fedora3.openshift.io"]
call_mock.return_value = POD_SELECT_STD_OUT
# Should select first valid podname in call_or_fail output.
pod_name = glusterfs_check_cns.select_pod(module, oc_exec, cluster_name, valid_nodes)
assert pod_name == 'glusterfs-storage-mp9nk'
with pytest.raises(Exception) as err:
pod_name = glusterfs_check_cns.select_pod(module, oc_exec, "does not exist", valid_nodes)
assert 'Exception: Unable to find suitable pod in get pods output' in str(err)
@patch('glusterfs_check_cns.call_or_fail')
def test_get_volume_list(call_mock):
module = DummyModule()
oc_exec = []
pod_name = ''
call_mock.return_value = VOLUME_LIST_STDOUT
volume_list = glusterfs_check_cns.get_volume_list(module, oc_exec, pod_name)
assert volume_list == ['heketidbstorage', 'volume1']
@patch('glusterfs_check_cns.call_or_fail')
def test_check_volume_health_info(call_mock):
module = DummyModule()
oc_exec = []
pod_name = ''
volume = 'somevolume'
call_mock.return_value = VOLUME_HEAL_INFO_GOOD
# this should just complete quietly.
glusterfs_check_cns.check_volume_health_info(module, oc_exec, pod_name, volume)
call_mock.return_value = VOLUME_HEAL_INFO_BAD
expected_error = 'volume {} is not ready'.format(volume)
with pytest.raises(Exception) as err:
glusterfs_check_cns.check_volume_health_info(module, oc_exec, pod_name, volume)
assert expected_error in str(err)
if __name__ == '__main__':
test_get_valid_nodes()
test_select_pod()
test_get_volume_list()
test_check_volume_health_info()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment