Python init_table Examples

Example #1

File: test_iptables.py Project: MaelStor/connord

def test_init_table_when_user_is_not_root(mocker):
    table_name = "filter"
    mocked_user = mocker.patch("connord.iptables.user.is_root")
    mocked_user.return_value = False

    from connord import iptables

    try:
        iptables.init_table(table_name)
        assert False
    except PermissionError as error:
        assert str(error) == "Function 'init_table' needs root access."

    mocked_user.assert_called_once()

Example #2

File: test_iptables.py Project: MaelStor/connord

def test_init_table_when_table_is_valid_ipv4(mocker):
    table_name = "filter"
    mocked_user = mocker.patch("connord.iptables.user.is_root")
    mocked_user.return_value = True

    from connord import iptables

    table_mock = mocker.Mock()
    mocked_table = mocker.patch.object(iptables, "Table", table_mock)
    mocker.patch.object(iptables.Table, "ALL", ["filter"])

    iptables.init_table(table_name)

    mocked_user.assert_called_once()
    mocked_table.assert_called_once_with("filter")

Example #3

File: test_iptables.py Project: MaelStor/connord

def test_init_table_when_table_is_invalid_ipv4(mocker):
    table_name = "invalid"
    mocked_user = mocker.patch("connord.iptables.user.is_root")
    mocked_user.return_value = True

    from connord import iptables

    table_mock = mocker.Mock()
    mocked_table = mocker.patch.object(iptables, "Table", table_mock)
    mocker.patch.object(iptables.Table, "ALL", ["filter"])

    try:
        iptables.init_table(table_name)
        assert False
    except TypeError as error:
        assert str(error) == "Error: 'invalid' is not a valid table."

    mocked_user.assert_called_once()
    mocked_table.assert_not_called()