80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
# Copyright (c) Twisted Matrix Laboratories.
|
|
# See LICENSE for details.
|
|
|
|
"""
|
|
Tests for L{twisted.tap.ftp}.
|
|
"""
|
|
|
|
from twisted.trial.unittest import TestCase
|
|
|
|
from twisted.cred import credentials, error
|
|
from twisted.tap.ftp import Options
|
|
from twisted.python import versions
|
|
from twisted.python.filepath import FilePath
|
|
|
|
|
|
|
|
class FTPOptionsTests(TestCase):
|
|
"""
|
|
Tests for the command line option parser used for C{twistd ftp}.
|
|
"""
|
|
|
|
usernamePassword = (b'iamuser', b'thisispassword')
|
|
|
|
def setUp(self):
|
|
"""
|
|
Create a file with two users.
|
|
"""
|
|
self.filename = self.mktemp()
|
|
f = FilePath(self.filename)
|
|
f.setContent(b':'.join(self.usernamePassword))
|
|
self.options = Options()
|
|
|
|
|
|
def test_passwordfileDeprecation(self):
|
|
"""
|
|
The C{--password-file} option will emit a warning stating that
|
|
said option is deprecated.
|
|
"""
|
|
self.callDeprecated(
|
|
versions.Version("Twisted", 11, 1, 0),
|
|
self.options.opt_password_file, self.filename)
|
|
|
|
|
|
def test_authAdded(self):
|
|
"""
|
|
The C{--auth} command-line option will add a checker to the list of
|
|
checkers
|
|
"""
|
|
numCheckers = len(self.options['credCheckers'])
|
|
self.options.parseOptions(['--auth', 'file:' + self.filename])
|
|
self.assertEqual(len(self.options['credCheckers']), numCheckers + 1)
|
|
|
|
|
|
def test_authFailure(self):
|
|
"""
|
|
The checker created by the C{--auth} command-line option returns a
|
|
L{Deferred} that fails with L{UnauthorizedLogin} when
|
|
presented with credentials that are unknown to that checker.
|
|
"""
|
|
self.options.parseOptions(['--auth', 'file:' + self.filename])
|
|
checker = self.options['credCheckers'][-1]
|
|
invalid = credentials.UsernamePassword(self.usernamePassword[0], 'fake')
|
|
return (checker.requestAvatarId(invalid)
|
|
.addCallbacks(
|
|
lambda ignore: self.fail("Wrong password should raise error"),
|
|
lambda err: err.trap(error.UnauthorizedLogin)))
|
|
|
|
|
|
def test_authSuccess(self):
|
|
"""
|
|
The checker created by the C{--auth} command-line option returns a
|
|
L{Deferred} that returns the avatar id when presented with credentials
|
|
that are known to that checker.
|
|
"""
|
|
self.options.parseOptions(['--auth', 'file:' + self.filename])
|
|
checker = self.options['credCheckers'][-1]
|
|
correct = credentials.UsernamePassword(*self.usernamePassword)
|
|
return checker.requestAvatarId(correct).addCallback(
|
|
lambda username: self.assertEqual(username, correct.username)
|
|
)
|