mirror of https://github.com/scrapy/scrapy.git
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import pytest
|
|
from twisted.conch.telnet import ITelnetProtocol
|
|
from twisted.cred import credentials
|
|
|
|
from scrapy.extensions.telnet import TelnetConsole
|
|
from scrapy.utils.test import get_crawler
|
|
from tests.utils.decorators import inline_callbacks_test
|
|
|
|
pytestmark = pytest.mark.requires_reactor # TelnetConsole requires a reactor
|
|
|
|
|
|
class TestTelnetExtension:
|
|
def _get_console_and_portal(self, settings=None):
|
|
crawler = get_crawler(settings_dict=settings)
|
|
console = TelnetConsole(crawler)
|
|
|
|
# This function has some side effects we don't need for this test
|
|
console._get_telnet_vars = dict
|
|
|
|
console.start_listening()
|
|
protocol = console.protocol()
|
|
portal = protocol.protocolArgs[0]
|
|
|
|
return console, portal
|
|
|
|
@inline_callbacks_test
|
|
def test_bad_credentials(self):
|
|
console, portal = self._get_console_and_portal()
|
|
creds = credentials.UsernamePassword(b"username", b"password")
|
|
d = portal.login(creds, None, ITelnetProtocol)
|
|
with pytest.raises(ValueError, match="Invalid credentials"):
|
|
yield d
|
|
console.stop_listening()
|
|
|
|
@inline_callbacks_test
|
|
def test_good_credentials(self):
|
|
console, portal = self._get_console_and_portal()
|
|
creds = credentials.UsernamePassword(
|
|
console.username.encode("utf8"), console.password.encode("utf8")
|
|
)
|
|
d = portal.login(creds, None, ITelnetProtocol)
|
|
yield d
|
|
console.stop_listening()
|
|
|
|
@inline_callbacks_test
|
|
def test_custom_credentials(self):
|
|
settings = {
|
|
"TELNETCONSOLE_USERNAME": "user",
|
|
"TELNETCONSOLE_PASSWORD": "pass",
|
|
}
|
|
console, portal = self._get_console_and_portal(settings=settings)
|
|
creds = credentials.UsernamePassword(b"user", b"pass")
|
|
d = portal.login(creds, None, ITelnetProtocol)
|
|
yield d
|
|
console.stop_listening()
|