bumblebee-status/tests/modules/contrib/test_network.py

from unittest import TestCase, mock
import pytest

import core.config
import core.widget
import modules.contrib.network

import socket

pytest.importorskip("netifaces")


def build_module():
    config = core.config.Config([])
    return modules.contrib.network.Module(config=config, theme=None)


def wireless_default():
    return {"default": {1: ("10.0.1.12", "wlan3")}}


def wired_default():
    return {"default": {18: ("10.0.1.12", "eth3")}}


def exec_side_effect_valid(*args, **kwargs):
    if args[0] == "iwgetid":
        return "ESSID: bumblefoo"
    if "iwconfig" in args[0]:
        return "level=-30"
    return mock.DEFAULT


def exec_side_effect_invalid(*args, **kwargs):
    return "invalid gibberish, can't parse for info"


class TestNetworkUnit(TestCase):
    def test_load_module(self):
        __import__("modules.contrib.network")

    @pytest.mark.allow_hosts(["127.0.0.1"])
    def test_no_internet(self):
        module = build_module()
        assert module.widgets()[0].full_text() == "No connection"

    @mock.patch("util.cli.execute")
    @mock.patch("netifaces.gateways")
    @mock.patch("socket.create_connection")
    @mock.patch("netifaces.AF_INET", 1)
    @mock.patch("builtins.open", mock.mock_open(read_data="wlan3"))
    def test_valid_wireless_connection(self, socket_mock, gateways_mock, execute_mock):
        socket_mock.return_value = mock.MagicMock()
        fake_ssid = "bumblefoo"
        gateways_mock.return_value = wireless_default()
        execute_mock.side_effect = exec_side_effect_valid

        module = build_module()

        assert fake_ssid in module.widgets()[0].full_text()

    @mock.patch("netifaces.gateways")
    @mock.patch("socket.create_connection")
    @mock.patch("netifaces.AF_INET", 18)
    @mock.patch("builtins.open", mock.mock_open(read_data="wlan3"))
    def test_valid_wired_connection(self, socket_mock, gateways_mock):
        gateways_mock.return_value = wired_default()
        socket_mock.return_value = mock.MagicMock()

        module = build_module()

        assert module.widgets()[0].full_text() == "Ethernet"

    @mock.patch("netifaces.gateways")
    @mock.patch("socket.create_connection")
    def test_invalid_gateways(self, socket_mock, gateways_mock):
        socket_mock.return_value = mock.Mock()
        gateways_mock.return_value = {"xyz": "abc"}

        module = build_module()
        assert module.widgets()[0].full_text() == "No connection"

    @mock.patch("util.cli.execute")
    @mock.patch("socket.create_connection")
    @mock.patch("netifaces.gateways")
    @mock.patch("netifaces.AF_INET", 1)
    @mock.patch("builtins.open", mock.mock_open(read_data="wlan3"))
    def test_invalid_execs(self, gateways_mock, socket_mock, execute_mock):
        execute_mock.side_effect = exec_side_effect_invalid
        socket_mock.return_value = mock.MagicMock()
        gateways_mock.return_value = wireless_default()

        module = build_module()

        assert module.widgets()[0].full_text() == "Unknown ?%"

    @mock.patch("builtins.open", **{"return_value.raiseError.side_effect": Exception()})
    @mock.patch("socket.create_connection")
    @mock.patch("netifaces.gateways")
    @mock.patch("netifaces.AF_INET", 18)
    @mock.patch("builtins.open", mock.mock_open(read_data="wlan3"))
    def test_no_wireless_file(self, gateways_mock, socket_mock, mock_open):
        gateways_mock.return_value = wired_default()
        socket_mock.return_value = mock.MagicMock()
        module = build_module()

        assert module.widgets()[0].full_text() == "Ethernet"