test_client

Complexity

Total Complexity

44

Size/Duplication

Total Lines	316
Duplicated Lines	0 %

Importance

Changes

0

30 Functions

Rating	Name	Duplication	Size	Complexity
A	get_client()	0	3	1
A	get_client_with_option_show_not_selected_rules()	0	3	1
A	get_client_with_option_show_not_selected_rules_and_show_fail_rules()	0	8	1
A	test_client()	0	6	1
A	get_client_with_option_show_fail_rules()	0	3	1
A	test_search_rules_id()	0	5	1
A	test_search_rules_with_regex()	0	5	1
A	test_get_questions()	0	13	1
A	test_find_does_not_exist_rule()	0	6	2
A	get_client_tree()	0	3	1
A	get_client_on_web_browser()	0	3	1
A	test_search_rules_id_on_web_browser()	0	5	1
A	test_find_not_selected_rule()	0	6	2
A	test_get_questions_not_selected()	0	13	1
A	test_get_questions_not_selected_and_show_fail_rules()	0	14	1
A	test_prepare_graph()	0	10	1
A	load_tested_file()	0	11	5
A	test_prepare_tree()	0	10	1
A	test_search_not_selected_rule()	0	5	2
A	test_search_non_existent_rule()	0	5	2
A	test_prepare_graph_with_not_selected_rule()	0	4	1
A	test_if_not_installed_PyInquirer()	0	10	2
A	try_expection_for_prepare_graph()	0	5	2
A	test_prepare_graph_with_non_existent_rule()	0	4	1
A	test_if_not_installed_PyInquirer_with_option_show_not_selected_rules()	0	11	2
A	test_if_not_installed_PyInquirer_with_option_show_fail_rules()	0	10	2
A	test_get_wanted_rules()	0	16	1
A	test_if_not_installed_PyInquirer_with_option_show_not_selected_rules_and_show_fail_rules()	0	12	2
A	test_get_wanted_not_selected_rules()	0	14	1
A	test_get_questions_with_option_show_fail_rules()	0	13	2

import graph.client
import pytest
import tests.any_test_help
import re
import json
import mock
import sys


def get_client(src, rule):
    return graph.client.client(
        ["--off-web-browser", tests.any_test_help.get_src(src), rule])


def get_client_on_web_browser(src, rule):
    return graph.client.client(
        [tests.any_test_help.get_src(src), rule])


def get_client_tree(src, rule):
    return graph.client.client(
        ["--tree", "--off-web-browser", tests.any_test_help.get_src(src), rule])


def get_client_with_option_show_fail_rules(src, rule):
    return graph.client.client(
        ["--show-fail-rules", "--off-web-browser", tests.any_test_help.get_src(src), rule])


def get_client_with_option_show_not_selected_rules(src, rule):
    return graph.client.client(
        ["--show-not-selected-rules", "--off-web-browser", tests.any_test_help.get_src(src), rule])


def get_client_with_option_show_not_selected_rules_and_show_fail_rules(
        src,
        rule):
    return graph.client.client(["--show-not-selected-rules",
                                "--show-fail-rules",
                                "--off-web-browser",
                                tests.any_test_help.get_src(src),
                                rule])


def test_client():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    rule = 'rule'
    client = get_client(src, rule)
    assert client.source_filename == tests.any_test_help.get_src(src)
    assert client.rule_name == rule


def test_search_rules_id():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    part_of_id_rule = 'xccdf_org.ssgproject.'
    client = get_client(src, part_of_id_rule)
    assert len(client.search_rules_id()) == 184


def test_search_rules_id_on_web_browser():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    part_of_id_rule = 'xccdf_org.ssgproject.'
    client = get_client_on_web_browser(src, part_of_id_rule)
    assert len(client.search_rules_id()) == 184


def test_find_does_not_exist_rule():
    rule = 'random_rule_which_doest_exist'
    src = 'test_data/ssg-fedora-ds-arf.xml'
    client = get_client(src, rule)
    with pytest.raises(Exception, match="err- 404 rule not found!"):
        assert client.search_rules_id()


def test_find_not_selected_rule():
    rule = 'xccdf_org.ssgproject.content_rule_ntpd_specify_remote_server'
    src = 'test_data/ssg-fedora-ds-arf.xml'
    client = get_client(src, rule)
    with pytest.raises(Exception, match=rule):
        assert client.search_rules_id()


def test_search_rules_with_regex():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client(src, regex)
    assert len(client.search_rules_id()) == 2


def test_get_questions():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client(src, regex)
    from PyInquirer import Separator

    out = client.get_questions(
        Separator('= The rules ID ='),
        Separator('= The not selected rules ID ='))
    rule1 = 'xccdf_org.ssgproject.content_rule_package_abrt_removed'
    rule2 = 'xccdf_org.ssgproject.content_rule_package_sendmail_removed'
    assert out[0]['choices'][1]['name'] == rule1
    assert out[0]['choices'][2]['name'] == rule2


def test_get_questions_not_selected():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client_with_option_show_not_selected_rules(src, regex)
    from PyInquirer import Separator

    out = client.get_questions(
        Separator('= The rules ID ='),
        Separator('= The not selected rules ID ='))
    rule1 = 'xccdf_org.ssgproject.content_rule_package_setroubleshoot_removed'
    rule2 = 'xccdf_org.ssgproject.content_rule_package_mcstrans_removed'
    assert out[0]['choices'][-2]['name'] == rule1
    assert out[0]['choices'][-1]['name'] == rule2


def test_get_questions_not_selected_and_show_fail_rules():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client_with_option_show_not_selected_rules_and_show_fail_rules(
        src, regex)
    from PyInquirer import Separator

    out = client.get_questions(
        Separator('= The rules ID ='),
        Separator('= The not selected rules ID ='))
    rule1 = 'xccdf_org.ssgproject.content_rule_package_abrt_removed'
    rule2 = 'xccdf_org.ssgproject.content_rule_package_mcstrans_removed'
    assert out[0]['choices'][1]['name'] == rule1
    assert out[0]['choices'][-1]['name'] == rule2


def test_get_questions_with_option_show_fail_rules():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client_with_option_show_fail_rules(src, regex)
    from PyInquirer import Separator

    out = client.get_questions(
        Separator('= The rules ID ='),
        Separator('= The not selected rules ID ='))
    rule1 = 'xccdf_org.ssgproject.content_rule_package_abrt_removed'
    assert out[0]['choices'][1]['name'] == rule1
    with pytest.raises(Exception, match="list index out of range"):
        assert out[0]['choices'][2]['name'] is None


def test_get_wanted_not_selected_rules():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client(src, regex)

    out = [
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_nis_removed'},
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_ntpdate_removed'},
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_telnetd_removed'},
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_gdm_removed'},
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_setroubleshoot_removed'},
        {'id_rule': 'xccdf_org.ssgproject.content_rule_package_mcstrans_removed'}]

    assert out == client._get_wanted_not_selected_rules()


def test_get_wanted_rules():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    regex = r'_package_\w+_removed'
    client = get_client(src, regex)

    out = [
        {'href': '#oval0',
         'id_def': 'oval:ssg-package_abrt_removed:def:1',
         'id_rule': 'xccdf_org.ssgproject.content_rule_package_abrt_removed',
         'result': 'fail'},
        {'href': '#oval0',
         'id_def': 'oval:ssg-package_sendmail_removed:def:1',
         'id_rule': 'xccdf_org.ssgproject.content_rule_package_sendmail_removed',
         'result': 'pass'}]

    assert out == client._get_wanted_rules()


def test_search_non_existent_rule():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    non_existent_rule = 'non-existent_rule'
    with pytest.raises(Exception, match="err- 404 rule not found!"):
        assert get_client(src, non_existent_rule).search_rules_id()


def test_search_not_selected_rule():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    non_existent_rule = 'xccdf_org.ssgproject.content_rule_package_nis_removed'
    with pytest.raises(Exception, match=non_existent_rule):
        assert get_client(src, non_existent_rule).search_rules_id()


def test_prepare_graph():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    rule = 'xccdf_org.ssgproject.content_rule_package_abrt_removed'
    client = get_client(src, rule)
    rules = {'rules': [rule]}
    client.prepare_data(rules)
    result = load_tested_file('../graph_html_interpreter/data.js')
    referenc_result = load_tested_file(
        'test_data/referenc_result_data_graph.js')
    assert result == referenc_result


def test_prepare_tree():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    rule = 'xccdf_org.ssgproject.content_rule_package_abrt_removed'
    client = get_client_tree(src, rule)
    rules = {'rules': [rule]}
    client.prepare_data(rules)
    result = load_tested_file('../tree_html_interpreter/data.js')
    referenc_result = load_tested_file(
        'test_data/referenc_result_data_tree.js')
    assert result == referenc_result


def load_tested_file(src):
    with open(tests.any_test_help.get_src(src), 'r') as f:
        data = f.readlines()
    out = []
    edge = False
    for row in data:
        if row == '    "edges": [\n':
            edge = True
        if not edge:
            out.append(row)
    return out


def try_expection_for_prepare_graph(src, rule, err):
    client = get_client(src, rule)
    rules = {'rules': [rule]}
    with pytest.raises(Exception, match=err):
        assert client.prepare_data(rules)


def test_prepare_graph_with_non_existent_rule():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    rule = 'non-existent_rule'
    try_expection_for_prepare_graph(src, rule, '404')


def test_prepare_graph_with_not_selected_rule():
    src = 'test_data/ssg-fedora-ds-arf.xml'
    rule = 'xccdf_org.ssgproject.content_rule_package_nis_removed'
    try_expection_for_prepare_graph(src, rule, 'not selected')


def test_if_not_installed_PyInquirer(capsys):
    with mock.patch.dict(sys.modules, {'PyInquirer': None}):
        src = 'test_data/ssg-fedora-ds-arf.xml'
        regex = r'_package_\w+_removed'
        client = get_client(src, regex)
        out = client.run_gui_and_return_answers()
        assert out is None
        captured = capsys.readouterr()
        assert captured.out == (
            '== The Rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_abrt_removed\\b\n'
            'xccdf_org.ssgproject.content_rule_package_sendmail_removed\\b\n')


def test_if_not_installed_PyInquirer_with_option_show_fail_rules(capsys):
    with mock.patch.dict(sys.modules, {'PyInquirer': None}):
        src = 'test_data/ssg-fedora-ds-arf.xml'
        regex = r'_package_\w+_removed'
        client = get_client_with_option_show_fail_rules(src, regex)
        out = client.run_gui_and_return_answers()
        assert out is None
        captured = capsys.readouterr()
        assert captured.out == (
            '== The Rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_abrt_removed\\b\n')


def test_if_not_installed_PyInquirer_with_option_show_not_selected_rules(
        capsys):
    with mock.patch.dict(sys.modules, {'PyInquirer': None}):
        src = 'test_data/ssg-fedora-ds-arf.xml'
        regex = r'_package_\w+_removed'
        client = get_client_with_option_show_not_selected_rules(src, regex)
        out = client.run_gui_and_return_answers()
        assert out is None
        captured = capsys.readouterr()
        assert captured.out == (
            '== The Rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_abrt_removed\\b\n'
            'xccdf_org.ssgproject.content_rule_package_sendmail_removed\\b\n'
            '== The not selected rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_nis_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_ntpdate_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_telnetd_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_gdm_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_setroubleshoot_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_mcstrans_removed(Not selected)\n')


def test_if_not_installed_PyInquirer_with_option_show_not_selected_rules_and_show_fail_rules(
        capsys):
    with mock.patch.dict(sys.modules, {'PyInquirer': None}):
        src = 'test_data/ssg-fedora-ds-arf.xml'
        regex = r'_package_\w+_removed'
        client = get_client_with_option_show_not_selected_rules_and_show_fail_rules(
            src, regex)
        out = client.run_gui_and_return_answers()
        assert out is None
        captured = capsys.readouterr()
        assert captured.out == (
            '== The Rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_abrt_removed\\b\n'
            '== The not selected rule IDs ==\n'
            'xccdf_org.ssgproject.content_rule_package_nis_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_ntpdate_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_telnetd_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_gdm_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_setroubleshoot_removed(Not selected)\n'
            'xccdf_org.ssgproject.content_rule_package_mcstrans_removed(Not selected)\n')