16 lines
562 B
Python
16 lines
562 B
Python
import os
|
|
import pytest
|
|
import updater
|
|
from unittest import mock
|
|
|
|
@pytest.mark.parametrize("system", [("Windows", "Linux", "Darwin")])
|
|
def test_find_datafiles(system):
|
|
with mock.patch("platform.system", return_value=system):
|
|
result = updater.find_datafiles()
|
|
if system == "Windows":
|
|
assert "bootstrap.exe" in result[0][1]
|
|
assert os.path.exists(result[0][1][0])
|
|
else:
|
|
assert len(result[0][1]) == 2
|
|
assert os.path.exists(result[0][1][0])
|
|
assert os.path.exists(result[0][1][1]) |