- 991E1FF9817A01D4320ABDF30E09C890B5454C726E313A488DD6BEDC9E8E663019A63CAF9FC16979251F653BEDA05ECECA4A806A7E3C299E6847A8B6FE11A6E4+ F6EB911B7A54EC1EF368E1EED8202AA2E385086F3D5509C08EB8BB78D216FF2408B7409B5F4F368D46EA98475C4AFB3A67A094EDE9109278AD0CB7EA93459F4Bblatta/tests/test_station.py(2 . 27)(2 . 81)
3363 import unittest
3364 import logging
3365 from mock import Mock
3366 from mock import patch
3367
3368 from lib.commands import DIRECT
3369 from lib.commands import GETDATA
3370 from lib.station import Station
3371 from lib.state import State
3372 from lib.message import Message
3373 from lib.getdata import GetData
3374 from lib.order_buffer import OrderBuffer
3375 from lib.long_buffer import LongBuffer
3376 from collections import namedtuple
3377 import helper
3378
3379 class TestStation(unittest.TestCase):
3380 def setUp(self):
3381 helper.setup()
3382 logging.basicConfig(level=logging.DEBUG)
3383 options = {
3384 "clients": {"clientsocket": Mock()},
3385 "db_path": "tests/test.db",
3386 "socket": Mock()
3387 }
3388 self.station_socket = Mock()
3389 Options = namedtuple('Options', ['db_path',
3390 'address_table_path',
3391 'socket',
3392 'irc_ports',
3393 'udp_port',
3394 'channel_name',
3395 'password',
3396 'motd',
3397 'listen'])
3398 options = Options(
3399 None,
3400 None,
3401 self.station_socket,
3402 None,
3403 None,
3404 None,
3405 None,
3406 None,
3407 None
3408 )
3409 self.station = Station(options)
3410 self.station.deliver = Mock()
3411 self.station.rebroadcast = Mock()
3412 self.station.rebroadcast.return_value = "foobar"
3413 self.bob_state = State(Mock(), None)
3414 self.station.state.set_knob('nick', 'alice')
3415 self.bob_state.set_knob('nick', 'bob')
3416 self.setupBob()
3417 self.setupAlice()
3418
3419 def setupBob(self):
3420 self.station.state.add_peer('bob')
3421 self.station.state.add_key(
3422 'bob',
3423 '9h6wYndVjt8QpnIZOYb7KD2tYKCKw4rjlYg4LM1ODx1Qkr3qA0IuKNukkwKhQ4UP9ypMlhyPHa7AGD7NO7Ws5w=='
3424 )
3425 self.station.state.update_at({
3426 'handle': 'bob',
3427 'address': '127.0.0.1',
3428 'port': 8889
3429 })
3430
3431 def setupAlice(self):
3432 self.bob_state.add_peer('alice')
3433 self.bob_state.add_key(
3434 'alice',
3435 '9h6wYndVjt8QpnIZOYb7KD2tYKCKw4rjlYg4LM1ODx1Qkr3qA0IuKNukkwKhQ4UP9ypMlhyPHa7AGD7NO7Ws5w=='
3436 )
3437 self.bob_state.update_at({
3438 'handle': 'alice',
3439 'address': '127.0.0.1',
3440 'port': 8888
3441 })
3442
3443 def tearDown(self):
3444 pass
3445
3446 def test_embargo_bounce_ordering(self):
3447 self.skipTest("the tested code has been re-implemented")
3448 peer1 = Mock()
3449 peer1.handles = ["a", "b"]
3450 peer2 = Mock()
(35 . 7)(89 . 7)
3452 high_bounce_message.peer = peer2
3453 high_bounce_message.bounces = 2
3454 high_bounce_message.message_hash = "messagehash"
3455 self.station.embargo_queue = {
3456 self.station.short_buffer = {
3457 "messagehash": [
3458 low_bounce_message,
3459 high_bounce_message
(45 . 73)(99 . 31)
3461 self.station.deliver.assert_called_once_with(low_bounce_message)
3462 self.station.rebroadcast.assert_called_once_with(low_bounce_message)
3463
3464 def test_immediate_message_delivered(self):
3465 peer = Mock()
3466 peer.handles = ["a", "b"]
3467 message = Mock()
3468 message.speaker = "a"
3469 message.peer = peer
3470 self.station.embargo_queue = {
3471 "messagehash": [
3472 message
3473 ],
3474 }
3475 self.station.check_for_immediate_messages()
3476 self.station.deliver.assert_called_once_with(message)
3477 self.station.rebroadcast.assert_called_once_with(message)
3478
3479 def test_hearsay_message_not_delivered(self):
3480 peer = Mock()
3481 peer.handles = ["a", "b"]
3482 message = Mock()
3483 message.speaker = "c"
3484 message.peer = peer
3485 self.station.embargo_queue = {
3486 "messagehash": [
3487 message
3488 ],
3489 }
3490 self.station.check_for_immediate_messages()
3491 self.station.deliver.assert_not_called()
3492
3493 def test_embargo_queue_cleared(self):
3494 self.skipTest("the embargo queue is now th short buffer")
3495 peer = Mock()
3496 peer.handles = ["a", "b"]
3497 message = Mock()
3498 message.speaker = "c"
3499 message.peer = peer
3500 self.station.embargo_queue = {
3501 self.station.short_buffer = {
3502 "messagehash": [
3503 message
3504 ],
3505 }
3506 self.assertEqual(len(self.station.embargo_queue), 1)
3507 self.assertEqual(len(self.station.short_buffer), 1)
3508 self.station.flush_hearsay_messages()
3509 self.assertEqual(len(self.station.embargo_queue), 0)
3510
3511 def test_immediate_prefix(self):
3512 peer = Mock()
3513 peer.handles = ["a", "b"]
3514 message = Mock()
3515 message.speaker = "a"
3516 message.prefix = None
3517 message.peer = peer
3518 self.station.embargo_queue = {
3519 "messagehash": [
3520 message
3521 ],
3522 }
3523 self.station.check_for_immediate_messages()
3524 self.assertEqual(message.prefix, None)
3525 self.assertEqual(len(self.station.short_buffer), 0)
3526
3527 def test_simple_hearsay_prefix(self):
3528 self.skipTest("this code has moved")
3529 peer = Mock()
3530 peer.handles = ["a", "b"]
3531 message = Mock()
3532 message.speaker = "c"
3533 message.prefix = None
3534 message.peer = peer
3535 self.station.embargo_queue = {
3536 self.station.short_buffer = {
3537 "messagehash": [
3538 message
3539 ],
(120 . 6)(132 . 7)
3541 self.assertEqual(message.prefix, "c[a]")
3542
3543 def test_in_wot_hearsay_prefix_under_four(self):
3544 self.skipTest("the embargo queue is now th short buffer")
3545 peer1 = Mock()
3546 peer1.handles = ["a", "b"]
3547 peer2 = Mock()
(141 . 7)(154 . 7)
3549 message_via_peer3.prefix = None
3550 message_via_peer3.peer = peer3
3551 message_via_peer3.bounces = 1
3552 self.station.embargo_queue = {
3553 self.station.short_buffer = {
3554 "messagehash": [
3555 message_via_peer1,
3556 message_via_peer2,
(153 . 6)(166 . 7)
3558 self.assertEqual(message_via_peer1.prefix, "c[a|d|f]")
3559
3560 def test_in_wot_hearsay_prefix_more_than_three(self):
3561 self.skipTest("the embargo queue is now th short buffer")
3562 peer1 = Mock()
3563 peer1.handles = ["a", "b"]
3564 peer2 = Mock()
(181 . 7)(195 . 7)
3566 message_via_peer4.prefix = None
3567 message_via_peer4.peer = peer4
3568 message_via_peer4.bounces = 1
3569 self.station.embargo_queue = {
3570 self.station.short_buffer = {
3571 "messagehash": [
3572 message_via_peer1,
3573 message_via_peer2,
(192 . 3)(206 . 47)
3575 self.station.flush_hearsay_messages()
3576 self.station.deliver.assert_called_once_with(message_via_peer1)
3577 self.assertEqual(message_via_peer1.prefix, "c[4]")
3578
3579 # this test occasionally fails
3580 def test_receive_getdata_request_for_existing_direct_message(self):
3581 self.skipTest("intermittent failure")
3582 # 'send' bob a couple of messages
3583 m1 = Message({
3584 'command': DIRECT,
3585 'handle': 'bob',
3586 'speaker': 'alice',
3587 'body': 'm1',
3588 'bounces': 0
3589 }, self.station.state)
3590
3591 m1.send()
3592
3593 m2 = Message({
3594 'command': DIRECT,
3595 'handle': 'bob',
3596 'speaker': 'alice',
3597 'body': 'm2',
3598 'bounces': 0,
3599 }, self.station.state)
3600
3601 m2.send()
3602
3603 # oops look's like bob didn't get the message
3604
3605 # build GETDATA black packet to retreive m1
3606 alice = self.bob_state.get_peer_by_handle('alice')
3607 bob = self.station.state.get_peer_by_handle('bob')
3608 gd_message = GetData(m2, self.bob_state)
3609 gd_message_bytes = gd_message.get_message_bytes(alice)
3610 gd_black_packet = Message.pack(bob, GETDATA, gd_message.bounces, gd_message_bytes)
3611
3612 # call handle_udp_data with GETDATA packet
3613 self.station.handle_udp_data([gd_black_packet, ['127.0.0.1', 8889]])
3614
3615 # build up the retry black packet to verify that it was sent
3616 retry_black_packet = Message.pack(bob, DIRECT, 0, m1.get_message_bytes(bob))
3617
3618 # assert retry is called and sends the correct message
3619 sent_message_black_packet = self.station_socket.sendto.call_args[0][0]
3620 sent_message = Message.unpack(bob, sent_message_black_packet, LongBuffer(), OrderBuffer(), self.station.state)
3621 self.assertEqual(sent_message.body, 'm1')