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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
# Copyright (C) 2007 Giampaolo Rodola' <g.rodola@gmail.com>.
# Use of this source code is governed by MIT license that can be
# found in the LICENSE file.
import io
import os
import warnings
from unittest.mock import patch
import pytest
import pyftpdlib
from pyftpdlib import __ver__
from pyftpdlib.__main__ import main
from pyftpdlib.authorizers import DummyAuthorizer
from pyftpdlib.servers import FTPServer
from . import PyftpdlibTestCase
class TestCommandLineParser(PyftpdlibTestCase):
"""Test command line parser."""
def setUp(self):
super().setUp()
class DummyFTPServer(FTPServer):
"""An overridden version of FTPServer class which forces
serve_forever() to return immediately.
"""
def serve_forever(self, *args, **kwargs):
self.close_all()
self.devnull = io.BytesIO()
self.original_ftpserver_class = FTPServer
self.clog = patch("pyftpdlib.__main__.config_logging")
self.clog.start()
pyftpdlib.__main__.FTPServer = DummyFTPServer
def tearDown(self):
self.clog.stop()
pyftpdlib.servers.FTPServer = self.original_ftpserver_class
super().tearDown()
def test_interface_opt(self):
# no param
with pytest.raises(SystemExit):
main(["-i"])
with pytest.raises(SystemExit):
main(["--interface"])
main(["--interface", "127.0.0.1", "-p", "0"])
def test_port_opt(self):
# no param
with pytest.raises(SystemExit):
main(["-p"])
# not an int
with pytest.raises(SystemExit):
main(["-p", "foo"])
main(["-p", "0"])
main(["--port", "0"])
def test_write_opt(self):
with warnings.catch_warnings():
warnings.filterwarnings("error")
with pytest.raises(RuntimeWarning):
main(["-w", "-p", "0"])
with warnings.catch_warnings():
warnings.filterwarnings("ignore")
ftpd = main(["-w", "-p", "0"])
perms = ftpd.handler.authorizer.get_perms("anonymous")
assert (
perms
== DummyAuthorizer.read_perms + DummyAuthorizer.write_perms
)
# unexpected argument
with warnings.catch_warnings():
with pytest.raises(SystemExit):
main(["-w", "foo", "-p", "0"])
def test_directory_opt(self):
dirname = self.get_testfn()
os.mkdir(dirname)
ftpd = main(["-d", dirname, "-p", "0"])
ftpd = main(["--directory", dirname, "-p", "0"])
assert ftpd.handler.authorizer.get_home_dir(
"anonymous"
) == os.path.abspath(dirname)
# without argument
with pytest.raises(SystemExit):
main(["-d"])
# no such directory
with pytest.raises(ValueError, match="no such directory"):
main(["-d", "?!?", "-p", "0"])
def test_nat_address_opt(self):
ftpd = main(["-n", "127.0.0.1", "-p", "0"])
assert ftpd.handler.masquerade_address == "127.0.0.1"
ftpd.close_all()
ftpd = main(["--nat-address", "127.0.0.1", "-p", "0"])
ftpd.close_all()
assert ftpd.handler.masquerade_address == "127.0.0.1"
# without argument
with pytest.raises(SystemExit):
main(["-n", "-p", "0"])
def test_range_opt(self):
ftpd = main(["-r", "60000-61000", "-p", "0"])
assert ftpd.handler.passive_ports == list(range(60000, 61000 + 1))
# without arg
with pytest.raises(SystemExit):
main(["-r"])
# wrong arg
with pytest.raises(SystemExit):
main(["-r", "yyy-zzz"])
def test_debug_opt(self):
main(["-D", "-p", "0"])
main(["--debug", "-p", "0"])
# with arg
with pytest.raises(SystemExit):
main(["-D", "xxx"])
def test_version_opt(self):
for opt in ("-v", "--version"):
with pytest.raises(SystemExit) as cm:
main([opt, "-p", "0"])
assert str(cm.value) == f"pyftpdlib {__ver__}"
def test_verbose_opt(self):
for opt in ("-V", "--verbose"):
main([opt, "-p", "0"])
def test_username_and_password_opt(self):
ftpd = main(["--username", "foo", "--password", "bar", "-p", "0"])
assert ftpd.handler.authorizer.has_user("foo")
# no --password
with pytest.raises(SystemExit):
main(["--username", "foo"])
|
