111 lines
3.1 KiB
Python
111 lines
3.1 KiB
Python
import mock
|
|
from mock import patch
|
|
import pytest
|
|
from tinydb.storages import MemoryStorage
|
|
from tinydb import TinyDB, Query
|
|
import bumper
|
|
import os
|
|
import datetime, time
|
|
import platform
|
|
import json
|
|
import asyncio
|
|
from testfixtures import LogCapture
|
|
import sys
|
|
|
|
|
|
@patch("bumper.firstrun_input")
|
|
@patch("bumper.create_certs")
|
|
def test_firstrun(mock_input, mock_create):
|
|
with LogCapture() as l:
|
|
|
|
bumper.firstrun_input.return_value = "n"
|
|
bumper.first_run()
|
|
l.check_present(
|
|
(
|
|
"root",
|
|
"CRITICAL",
|
|
"Can't continue without certificates, please create some then try again.",
|
|
)
|
|
)
|
|
|
|
bumper.firstrun_input.return_value = "y"
|
|
bumper.first_run()
|
|
assert mock_create.called == True
|
|
|
|
|
|
def mock_subrun(*args):
|
|
return args
|
|
|
|
@patch("bumper.start")
|
|
def test_argparse(mock_start):
|
|
bumper.ca_cert = "tests/test_certs/ca.crt"
|
|
bumper.server_cert = "tests/test_certs/bumper.crt"
|
|
bumper.server_key = "tests/test_certs/bumper.key"
|
|
|
|
bumper.main(["--debug"])
|
|
assert bumper.bumper_debug == True
|
|
assert mock_start.called == True
|
|
|
|
bumper.main(["--listen", "127.0.0.1"])
|
|
assert bumper.bumper_listen == "127.0.0.1"
|
|
assert mock_start.called == True
|
|
|
|
bumper.main(["--announce", "127.0.0.1"])
|
|
assert bumper.bumper_announce_ip == "127.0.0.1"
|
|
assert mock_start.called == True
|
|
|
|
bumper.main(["--debug", "--listen", "127.0.0.1", "--announce", "127.0.0.1"])
|
|
assert bumper.bumper_debug == True
|
|
assert bumper.bumper_announce_ip == "127.0.0.1"
|
|
assert bumper.bumper_listen == "127.0.0.1"
|
|
assert mock_start.called == True
|
|
|
|
|
|
@patch("subprocess.run")
|
|
@patch("platform.system.lower")
|
|
@patch("platform.machine")
|
|
@patch("os.execv")
|
|
def test_createcert(mock_run, mock_platform, mock_machine, mock_exec):
|
|
mock_run.side_effect = mock_subrun
|
|
platform.system.return_value = "darwin"
|
|
bumper.create_certs()
|
|
assert mock_run.called == True
|
|
assert "../create_certs/create_certs_osx" in mock_exec.call_args.args[0]
|
|
|
|
platform.system.return_value = "windows"
|
|
bumper.create_certs()
|
|
assert mock_run.called == True
|
|
assert "../create_certs/create_certs_windows.exe" in mock_exec.call_args.args[0]
|
|
|
|
platform.system.return_value = "linux"
|
|
bumper.create_certs()
|
|
assert mock_run.called == True
|
|
assert "../create_certs/create_certs_linux" in mock_exec.call_args.args[0]
|
|
|
|
platform.system.return_value = "linux"
|
|
platform.machine.return_value = "arm"
|
|
bumper.create_certs()
|
|
assert mock_run.called == True
|
|
assert "../create_certs/create_certs_rpi" in mock_exec.call_args.args[0]
|
|
|
|
with LogCapture() as l:
|
|
platform.system.return_value = "nixbad"
|
|
bumper.create_certs()
|
|
|
|
l.check_present(
|
|
(
|
|
"root",
|
|
"CRITICAL",
|
|
"Can't determine platform. Create certs manually and try again.",
|
|
)
|
|
)
|
|
|
|
|
|
|
|
@patch("bumper.first_run")
|
|
def test_main(mock_firstrun):
|
|
bumper.ca_cert = "sf"
|
|
bumper.main()
|
|
assert mock_firstrun.called == True
|
|
bumper.ca_cert = "tests/test_certs/ca.crt"
|
|
|