1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import sys
from pathlib import Path
from tempfile import NamedTemporaryFile
from unittest.mock import patch
from apt_mirror.apt_mirror import get_config_file, is_alternative_binary_path
from apt_mirror.config import Config
from tests.base import BaseTest
class TestMain(BaseTest):
def test_config_file_argument(self):
with (
NamedTemporaryFile() as tmp_file,
NamedTemporaryFile() as tmp_file2,
patch.object(Config, "DEFAULT_CONFIGFILE", tmp_file.name),
patch.object(Config, "DEFAULT_CONFIGFILE2", tmp_file2.name),
):
with patch.object(sys, "argv", ["apt-mirror"]):
self.assertEqual(str(get_config_file()), Config.DEFAULT_CONFIGFILE)
with patch.object(sys, "argv", ["apt-mirror2"]):
self.assertEqual(str(get_config_file()), Config.DEFAULT_CONFIGFILE2)
with (
patch.object(sys, "argv", ["apt-mirror2", "/~~~nonexistent"]),
self.assertRaises(SystemExit),
):
self.assertEqual(str(get_config_file()), Config.DEFAULT_CONFIGFILE)
with (
patch.object(Config, "DEFAULT_CONFIGFILE2", "/~~~nonexistent"),
patch.object(sys, "argv", ["apt-mirror2"]),
self.assertRaises(SystemExit),
):
self.assertEqual(str(get_config_file()), Config.DEFAULT_CONFIGFILE2)
def test_alternative_path(self):
with patch.object(sys, "argv", ["apt-mirror"]):
self.assertFalse(is_alternative_binary_path())
with patch.object(sys, "argv", ["apt-mirror3"]):
self.assertFalse(is_alternative_binary_path())
with patch.object(sys, "argv", ["apt-mirror2"]):
self.assertTrue(is_alternative_binary_path())
def test_alternative_config_base_path(self):
with NamedTemporaryFile() as tmp_file:
config = Config(Path(tmp_file.name), Config.DEFAULT_BASE_PATH)
self.assertEqual(config.base_path, Path(Config.DEFAULT_BASE_PATH))
config = Config(Path(tmp_file.name), Config.DEFAULT_BASE_PATH2)
self.assertEqual(config.base_path, Path(Config.DEFAULT_BASE_PATH2))
|
