class Test_metalineReceived:
def setUp(self):
self.realm = RootRealm(Mock())
self.p = Mock()
self.realm.addProtocol(self.p)
self.realm.telnet = Mock()
self.ml = simpleml('foo', sentinel.fores, sentinel.backs)
self.ml2 = simpleml("bar", sentinel.fores, sentinel.backs)
self.ml2_written = self.ml2.copy()
self.ml2_written.insert(0, '\n')
@property
def lines_gotten(self):
return [line for ((line,), kwargs) in
self.p.metalineReceived.call_args_list]
def test_sends_to_screen_normally(self):
self.realm.metalineReceived(self.ml)
assert self.lines_gotten == [self.ml]
@binding_trigger("foo")
def trigger_1(self, match, realm):
realm.write(self.ml2)
def test_write_writes_afterwards(self):
self.realm.triggers.append(self.trigger_1)
self.realm.metalineReceived(self.ml)
assert self.lines_gotten == [self.ml, self.ml2_written]
@binding_trigger("foo")
def trigger_2(self, match, realm):
realm.display_line = False
def test_doesnt_display_if_asked_not_to(self):
self.realm.triggers.append(self.trigger_2)
self.realm.metalineReceived(self.ml)
assert self.lines_gotten == []
@binding_alias("spam")
def bar_writing_alias(self, match, realm):
realm.write("BAR BAR BAR")
realm.send_to_mud = False
@binding_trigger('foo')
def spam_sending_trigger(self, match, realm):
realm.send("spam")
def test_aliases_inside_triggers_write_after_trigger_writes(self):
self.realm.triggers.append(self.spam_sending_trigger)
self.realm.aliases.append(self.bar_writing_alias)
noteline = Metaline("\nBAR BAR BAR",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]))
self.realm.metalineReceived(self.ml)
assert self.lines_gotten == [self.ml, noteline]
class Test_addProtocol:
def setUp(self):
self.factory = TelnetClientFactory(None, 'ascii', None)
self.realm = RootRealm(self.factory)
self.realm.telnet = self.telnet = Mock(spec = TelnetClient)
self.receiver = Mock()
self.realm.addProtocol(self.receiver)
def test_passes_on_connection_lost(self):
self.realm.connectionLost()
assert self.receiver.connectionLost.called
@patch('pymudclient.realms', 'time')
def test_connection_lost_writes_message(self, our_time):
our_time.strftime.return_value = 'FOOBAR'
self.realm.write = Mock()
self.realm.connectionLost()
assert self.realm.write.called
ml = self.realm.write.call_args[0][0]
assert ml.line == 'FOOBAR'
assert ml.fores.items() == [(0, HexFGCode(0xFF, 0xAA, 0x00))]
assert ml.backs.items() == [(0, bg_code(BLACK))]
@patch('pymudclient.realms', 'time')
def test_connection_made_writes_message(self, our_time):
our_time.strftime.return_value = 'FOOBAR'
self.realm.write = Mock()
self.realm.connectionMade()
assert self.realm.write.called
ml = self.realm.write.call_args[0][0]
assert ml.line == 'FOOBAR'
assert ml.fores.items() == [(0, HexFGCode(0xFF, 0xAA, 0x00))]
assert ml.backs.items() == [(0, bg_code(BLACK))]
def test_passes_on_connection_made(self):
self.realm.connectionMade()
assert self.receiver.connectionMade.called
def test_sends_connection_lost_and_close_in_right_order(self):
self.realm.close()
#simulate Twisted's 'throw-it-over-the-wall' anti-guarantee
self.realm.connectionLost()
calls = [mname for (mname, args, kws) in self.receiver.method_calls]
#the metalineReceived is the closing note
assert calls == ['metalineReceived', 'connectionLost', 'close'], calls
def test_connection_lost_then_close_works(self):
self.realm.connectionLost()
self.realm.close()
calls = [mname for (mname, args, kws) in self.receiver.method_calls]
assert calls == ['metalineReceived', 'connectionLost', 'close'], calls
class TestRuneguardTriggers(unittest.TestCase):
@property
def lines_gotten(self):
return [line for ((line,), kwargs) in
self.protocol.metalineReceived.call_args_list]
@property
def telnet_lines_gotten(self):
return [line for ((line,),kwargs) in
self.realm.telnet.sendLine.call_args_list]
def setUp(self):
self.realm=RootRealm(None)
self.realm.factory=TestFactory()
self.realm.module_settings_dir='/home/dmitry/muds/settings'
import sys
sys.path.append('/home/dmitry/dev/modules')
from runeguard import Runeguard
#sr = ShieldRez(self.realm)
self.realm.load_module(Runeguard)
self.realm.telnet = TestPrinter()
self.protocol = TestPrinter()
self.realm.addProtocol(self.protocol)
self.realm.state['target']='Iniar'
def test_rune_fired(self):
self.realm.state['target']='Emily'
self.realm.metalineReceived(simpleml("You concentrate on the sowulu rune on an ink-stained snowy marble tablet, and its searing image suddenly flares on Emily's skin.",None,None))
self.realm.receive_gui_line('rv')
self.realm.metalineReceived(simpleml("You concentrate on the pithakhan rune on an ink-stained snowy marble tablet, and its searing image suddenly flares on Emily's skin.",None,None))
self.realm.receive_gui_line('rv')
self.realm.metalineReceived(simpleml("The residual effects of the sowulu rune around Emily fade.",None,None))
self.realm.receive_gui_line('rv')
def test_rebounding_on(self):
self.realm.state['last_command_type']='attack'
self.realm.metalineReceived(simpleml('You suddenly perceive the vague outline of an aura of rebounding around Iniar.',None,None))
self.realm.state['last_command_type']='raze'
self.realm.metalineReceived(simpleml('A shimmering translucent shield forms around Iniar.',None,None))
self.realm.metalineReceived(simpleml('The shimmering translucent shield around Iniar fades away.',None,None))
self.realm.metalineReceived(simpleml('Iniar\'s aura of weapons rebounding disappears.',None,None))
#print([l.line for l in self.lines_gotten])
#print([l for l in self.telnet_lines_gotten])
class TestTriggers(unittest.TestCase):
def setUp(self):
self.realm=RootRealm(None)
self.realm.load_module(TestModule, True)
self.realm.telnet = Mock()
self.protocol = Mock()
self.realm.addProtocol(self.protocol)
@property
def lines_gotten(self):
return [line for ((line,), kwargs) in
self.protocol.metalineReceived.call_args_list]
def test_simple_trigger(self):
self.realm.metalineReceived(simpleml('abc', None, None))
return_line = self.lines_gotten[1].line
assert self.lines_gotten[1].line == '\nsuccess'
def test_list_trigger(self):
self.realm.metalineReceived(simpleml('highjump', None, None))
return_line = self.lines_gotten[1].line
assert return_line=='\nsuccess2'
class Test_write_wrapping:
def setUp(self):
self.realm = RootRealm(Mock())
self.p = Mock()
self.realm.addProtocol(self.p)
def test_line_insertion_with_hard_line_end_no_sls(self):
self.realm._last_line_end = 'hard'
self.realm.write(simpleml("foo", sentinel.fore,
sentinel.back))
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [simpleml("\nfoo", sentinel.fore, sentinel.back)]
def test_line_insertion_with_hard_line_end_sls(self):
self.realm._last_line_end = 'hard'
ml = simpleml("foo", sentinel.fore, sentinel.back)
ml.soft_line_start = True
expected = ml.copy()
expected.insert(0, '\n')
self.realm.write(ml)
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [expected]
def test_line_insertion_sle_no_sls(self):
self.realm._last_line_end = 'soft'
self.realm.write(simpleml("foo", sentinel.fore,
sentinel.back))
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [simpleml("\nfoo", sentinel.fore, sentinel.back)]
def test_line_no_insertion_sle_sls(self):
self.realm._last_line_end = 'soft'
ml = simpleml("foo", sentinel.fore, sentinel.back)
ml.soft_line_start = True
expected = ml.copy()
self.realm.write(ml)
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [expected]
def test_respects_no_line_end(self):
self.realm._last_line_end = None
ml = simpleml("foo", sentinel.fore, sentinel.back)
expected = ml
self.realm.write(ml)
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [expected], rcvd
def test_last_line_end_setting(self):
ml = simpleml("foo", sentinel.fore, sentinel.back)
ml.line_end = sentinel.line_end
self.realm.write(ml)
assert self.realm._last_line_end == sentinel.line_end
def test_last_line_end_is_defaultly_None(self):
assert self.realm._last_line_end is None
def test_with_wrap(self):
ml = simpleml("bar", sentinel.fore1, sentinel.fore2)
ml.wrapped = Mock()
ml.wrapped.return_value = ml2 = simpleml("foo", sentinel.fore,
sentinel.back)
self.realm.write(ml)
assert ml.wrapped.called
rcvd = [ml for ((ml,), _) in self.p.metalineReceived.call_args_list]
assert rcvd == [ml2]
class Test_write:
def setUp(self):
self.realm = RootRealm(Mock())
self.realm.telnet = Mock()
self.p = Mock()
self.realm.addProtocol(self.p)
self.noting_line = simpleml("foo", Mock(), Mock())
def writer(self, match, realm):
print 'writer called!'
realm.write(self.noting_line)
@property
def lines_gotten(self):
return [line for ((line,), kwargs) in
self.p.metalineReceived.call_args_list]
def test_from_not_a_string(self):
self.realm.write(42)
assert len(self.lines_gotten) == 1
assert self.lines_gotten[0].line == '42'
def test_from_string(self):
self.realm.write('spam')
assert len(self.lines_gotten) == 1
assert self.lines_gotten[0].line == 'spam'
def test_from_metaline(self):
ml = Metaline('foo', None, None)
self.realm.write(ml)
assert self.lines_gotten == [ml]
def test_no_colourbleed_fg(self):
self.realm.write("eggs")
cols = self.lines_gotten[0].fores.items()
expected = [(0, fg_code(WHITE, False))]
assert cols == expected, (cols, expected)
def test_no_colourbleed_bg(self):
self.realm.write("eggs")
cols = self.lines_gotten[0].backs.items()
assert cols == [(0, bg_code(BLACK))], cols
def test_passes_on_wrap_default(self):
self.realm.write("eggs")
assert not self.lines_gotten[0].wrap
def test_soft_line_start_default_is_off(self):
self.realm.write("barbaz")
assert not self.lines_gotten[0].soft_line_start
def test_passes_on_soft_line_start(self):
self.realm.write('foo', soft_line_start = True)
assert self.lines_gotten[0].soft_line_start
noting_trigger = binding_trigger('bar')(writer)
noting_alias = binding_alias('bar')(writer)
def test_write_writes_after_during_matching_triggers(self):
self.realm.triggers.append(self.noting_trigger)
inline = Metaline('bar', set(), set())
self.realm.metalineReceived(inline)
assert self.lines_gotten == [inline, self.noting_line], \
self.lines_gotten
def test_write_writes_after_during_alias_matching(self):
self.realm.aliases.append(self.noting_alias)
inline = Metaline('bar', RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)
self.realm.send('bar')
print self.lines_gotten
print
expected = [inline, self.noting_line]
print expected
assert self.lines_gotten == expected
def tracer(self, match, realm):
realm.trace("Foo")
tracing_trigger = binding_trigger("baz")(tracer)
tracing_alias = binding_alias("baz")(tracer)
def test_trace_writes_after_during_matching_triggers(self):
self.realm.tracing = True
self.realm.triggers.append(self.tracing_trigger)
inline = Metaline('baz', set(), set())
self.realm.metalineReceived(inline)
expected_lines = [simpleml("\nTRACE: %s matched!" % self.tracing_trigger,
fg_code(WHITE, False), bg_code(BLACK)),
simpleml("\nTRACE: Foo", fg_code(WHITE, False),
bg_code(BLACK))]
print self.lines_gotten
print
expected = [inline] + expected_lines
print expected
assert self.lines_gotten == expected
def test_trace_writes_after_during_alias_matching(self):
self.realm.tracing = True
self.realm.aliases.append(self.tracing_alias)
inline = Metaline('baz', RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)
self.realm.send('baz')
expected_lines = [simpleml("\nTRACE: %s matched!" % self.tracing_alias,
fg_code(WHITE, False), bg_code(BLACK)),
simpleml("\nTRACE: Foo", fg_code(WHITE, False),
bg_code(BLACK))]
print self.lines_gotten
print
expected = [inline] + expected_lines
print expected
assert self.lines_gotten == expected
class Test_send:
def setUp(self):
self.realm = RootRealm(Mock())
self.p = Mock()
self.realm.addProtocol(self.p)
self.realm.telnet = Mock()
@property
def lines_gotten(self):
return [line for ((line,), kwargs) in
self.p.metalineReceived.call_args_list]
def test_send_sends_to_the_mud(self):
self.realm.send("bar")
assert self.realm.telnet.sendLine.call_args_list == [(('bar',), {})]
def test_send_echos_by_default(self):
self.realm.send("bar")
expected = Metaline('bar',
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)
assert self.lines_gotten == [expected]
def test_send_doesnt_echo_if_told_not_to(self):
self.realm.send("bar", False)
assert self.lines_gotten == []
def test_send_uses_echoes_with_soft_line_start(self):
self.realm.send("spam")
expected = Metaline('spam',
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)
assert self.lines_gotten == [expected]
#TODO: (not tested yet)
# - test calling send in a trigger and echoing then
# - test recursive calls to send properly
# - actually test matching properly
@binding_alias("bar")
def our_alias_1(self, match, realm):
print 'sending after!'
realm.send_after("foo")
def test_send_after_sends_afterwards(self):
self.realm.aliases.append(self.our_alias_1)
self.realm.send('bar')
assert self.realm.telnet.sendLine.call_args_list == [(('bar',), {}),
(('foo',), {})]
def test_send_after_default_echoing_is_off(self):
self.realm.aliases.append(self.our_alias_1)
self.realm.send("bar")
expected = [Metaline('bar',
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)]
assert self.lines_gotten == expected
@binding_alias('baz')
def our_alias_2(self, match, realm):
realm.send_after("eggs", echo = False)
def test_send_after_doesnt_echo_if_asked_not_to(self):
self.realm.aliases.append(self.our_alias_2)
self.realm.send("baz")
expected = [Metaline('baz',
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)]
assert self.lines_gotten == expected
@binding_alias("foo")
def foo_alias_sends_bar(self, match, realm):
print 'Foo alias going!'
realm.send('bar', echo = True)
@binding_alias("bar")
def bar_alias_sends_baz(self, match, realm):
print 'Bar alias going'
realm.send('baz', echo = True)
def test_sends_and_writes_in_a_consistent_order(self):
self.realm.aliases.append(self.foo_alias_sends_bar)
self.realm.aliases.append(self.bar_alias_sends_baz)
self.realm.send("foo", echo = True)
expect_write = [Metaline("baz",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True),
Metaline("\nbar",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True),
Metaline("\nfoo",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True)]
expect_send = ['baz', 'bar', 'foo']
sent = [line for ((line,), kwargs)
in self.realm.telnet.sendLine.call_args_list]
assert self.lines_gotten == expect_write
assert sent == expect_send
@binding_alias('spam')
def noisy_alias(self, match, realm):
realm.write("FOO FOO FOO")
def test_writes_come_after_echoing(self):
self.realm.aliases.append(self.noisy_alias)
self.realm.send("spam")
expecting = [Metaline("spam",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]),
soft_line_start = True),
Metaline("\nFOO FOO FOO",
RunLengthList([(0, fg_code(WHITE, False))]),
RunLengthList([(0, bg_code(BLACK))]))]
assert self.lines_gotten == expecting
def test_server_echo_defaultly_False(self):
assert not self.realm.server_echo
def test_doesnt_echo_if_server_echo_is_True(self):
self.realm.server_echo = True
self.realm.send("Foo")
assert self.lines_gotten == []