class Test_MCCP:
def setUp(self):
self.tc = TelnetClient(TelnetClientFactory(None, 'ascii', None))
self.tc.transport = self.t = FakeTransport()
def test_agree_to_enable_COMPRESS2(self):
assert self.tc.enableRemote(COMPRESS2)
def test_doesnt_agree_to_random_nexed_options(self):
assert not self.tc.enableRemote('\xA2')
def test_doesnt_enable_MCCP_without_COMPRESS2(self):
self.tc.negotiationMap[COMPRESS2]('')
assert not self.t.their_mccp_active
def test_does_enable_MCCP_with_COMPRESS2(self):
self.tc.enableRemote(COMPRESS2)
self.tc.negotiationMap[COMPRESS2]('')
assert self.t.their_mccp_active
def test_enable_disable_leaves_disabled(self):
self.tc.enableRemote(COMPRESS2)
self.tc.disableRemote(COMPRESS2)
self.tc.negotiationMap[COMPRESS2]('')
assert not self.t.their_mccp_active
def test_close_loses_connection(self):
self.tc.close()
assert self.t.lost_connection
def test_connectionMade_sends_connection_opened_to_the_outputs():
f = TelnetClientFactory(None, 'ascii', None)
telnet = TelnetClient(f)
r = f.realm = Mock(spec = RootRealm)
telnet.connectionMade()
assert r.connectionMade.called
class Test_sending_lines:
def setUp(self):
self.f = TelnetClientFactory(None, 'ascii', None)
self.tc = TelnetClient(self.f)
self.tc.transport = self.transport = FakeTransport()
def test_sendLine_appends_crlf(self):
self.tc.sendLine("foo")
assert self.transport.written == 'foo\r\n'
def test_doubles_IAC(self):
self.f.encoding = 'cp1250'
#crock a unicode sequence that has 0xFF in it as CP1250
self.tc.sendLine(u'foo\u02D9bar')
assert self.transport.written == 'foo\xff\xffbar\r\n'
def test_with_interesting_encoding(self):
self.f.encoding = 'utf-8'
self.tc.sendLine(u'foo')
assert self.transport.written == 'foo\r\n'
def test_with_interesting_encoding_and_interesting_characters(self):
self.f.encoding = 'cp1250'
self.tc.sendLine(u'foo\u2019bar')
assert self.transport.written == 'foo\x92bar\r\n'
def setUp(self):
self.f = TelnetClientFactory(None, 'ascii', None)
self.f.realm = self.e = Mock(spec = RootRealm)
self.tc = TelnetClient(self.f)
self.tc.transport = FakeTransport()
self.fores = RunLengthList([(0, fg_code(WHITE, False))])
self.backs = RunLengthList([(0, bg_code(BLACK))])
class Test_LineReceiver_aspects:
def setUp(self):
self.received = []
self.tc = TelnetClient(TelnetClientFactory(None, 'ascii', None))
self.tc.transport = FakeTransport()
self.tc.lineReceived = self.lr
def lr(self, line):
self.received.append(line)
def test_boring(self):
self.tc.dataReceived("foo\r\nbar\r\nbaz")
assert self.received == ['foo', 'bar'], self.received
def test_closing_flushes_buffer(self):
self.tc.dataReceived("bar")
self.tc.connectionLost(None)
assert self.received == ['bar']
def setUp(self):
self.tc = TelnetClient(TelnetClientFactory(None, 'ascii', None))
self.tc.transport = self.t = FakeTransport()
def setUp(self):
self.received = []
self.tc = TelnetClient(TelnetClientFactory(None, 'ascii', None))
self.tc.transport = FakeTransport()
self.tc.lineReceived = self.lr
class Test_receiving_lines:
def setUp(self):
self.f = TelnetClientFactory(None, 'ascii', None)
self.f.realm = self.e = Mock(spec = RootRealm)
self.tc = TelnetClient(self.f)
self.tc.transport = FakeTransport()
self.fores = RunLengthList([(0, fg_code(WHITE, False))])
self.backs = RunLengthList([(0, bg_code(BLACK))])
def test_ga_received_sends_line_on(self):
expected = [Metaline('foo', self.fores, self.backs, wrap = True,
line_end = 'soft')]
self.tc.dataReceived("foo" + IAC + GA)
print expected
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_ga_received_flushes_out_the_buffer(self):
expected = [Metaline("foo", self.fores, self.backs, wrap = True,
line_end = 'soft'),
Metaline('', self.fores, self.backs, wrap = True,
line_end = 'soft')]
self.tc.dataReceived('foo' + IAC + GA + IAC + GA)
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_lineReceived_sends_line_on(self):
self.tc.lineReceived("foo")
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == [Metaline('foo', self.fores, self.backs, wrap = True)]
def test_lineReceived_parses_colours(self):
expected = [Metaline('foo', RunLengthList([(0, fg_code(RED, False))]),
self.backs, wrap = True)]
self.tc.lineReceived('\x1b[31mfoo')
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_lineReceived_works_via_dataReceived(self):
expected = [Metaline('foo', self.fores, self.backs, wrap = True)]
self.tc.dataReceived('foo\r\n')
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_lineReceived_cleans_out_VT100_stuff(self):
expected = [Metaline('foo', self.fores, self.backs, wrap = True)]
self.tc.lineReceived('fooQ' + BS + VT)
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_lineReceived_decodes_data(self):
#real-ish example that booted me :(
self.f.encoding = 'cp1250'
expected = [Metaline(u"bar\u2019baz", self.fores, self.backs,
wrap = True)]
self.tc.lineReceived('bar\x92baz')
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_receives_repeated_normal_CR_LF_in_broken_godwars_mode_fine(self):
self.tc.fix_broken_godwars_line_endings = True
self.tc.dataReceived("foo\r\n\r\n")
expected = [simpleml("foo", fg_code(WHITE, False), bg_code(BLACK)),
simpleml("", fg_code(WHITE, False), bg_code(BLACK))]
for ml in expected:
ml.wrap = True
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_fixes_LF_CR_normally(self):
self.tc.fix_broken_godwars_line_endings = True
self.tc.dataReceived("foo\n\r")
expected = [simpleml("foo", fg_code(WHITE, False), bg_code(BLACK))]
expected[0].wrap = True
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
assert lines == expected, lines
def test_fixes_LF_CR_at_start(self):
self.tc.fix_broken_godwars_line_endings = True
self.tc.dataReceived("\n\r")
expected = [simpleml("", fg_code(WHITE, False), bg_code(BLACK))]
expected[0].wrap = True
lines = [line for ((line,), kwargs)
in self.e.metalineReceived.call_args_list]
print expected
assert lines == expected, lines
def setUp(self):
self.f = TelnetClientFactory(None, 'ascii', None)
self.f.gui = Mock()
self.tc = TelnetClient(self.f)
self.tc.transport = FakeTransport()
class Test_server_echo_setting:
def setUp(self):
self.f = TelnetClientFactory(None, 'ascii', None)
self.f.gui = Mock()
self.tc = TelnetClient(self.f)
self.tc.transport = FakeTransport()
def test_allows_enabling(self):
res = self.tc.enableRemote(ECHO)
assert res
def test_enable_sets_server_echo_to_True(self):
self.tc.enableRemote(ECHO)
assert self.f.realm.server_echo
def test_enable_hides_input_box_but_grabs_focus(self):
self.tc.enableRemote(ECHO)
assert self.f.gui.command_line.method_calls == \
[('set_visibility', (False,), {})]
def test_disable_sets_server_echo_to_False(self):
self.tc.enableRemote(ECHO)
self.tc.disableRemote(ECHO)
assert not self.f.realm.server_echo
def test_disable_shows_input_box_and_grabs_focus(self):
self.tc.disableRemote(ECHO)
assert self.f.gui.command_line.method_calls == \
[('set_visibility', (True,), {})]
