#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2009 Pierre-André Saulais <*****@*****.**>
#
# This file is part of the Spark File-transfer Tool.
#
# Spark is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# Spark is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Spark; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import unittest
import logging
import spark
from spark.tests.common import run_tests
if __name__ == "__main__":
modules = ["spark.tests.%s" % name for name in spark.tests.__all__]
run_tests(modules)
from spark.core import Future, Event
from spark.tests.common import run_tests, assertMatch, assertNoMatch
from spark.tests.ProtocolTest import testRequest, testResponse, testNotification, testBlock
class MessageMatchingTest(unittest.TestCase):
def testString(self):
""" match() should properly match string patterns """
assertMatch("foo", "foo")
assertNoMatch("foo", "bar")
def testTuples(self):
""" match() should properly match tuple patterns """
assertMatch(("foo", int), ("foo", 1))
assertNoMatch(("foo", int), ("bar", 1))
def testMessages(self):
""" match() should properly match message patterns """
assertMatch(Request("swap", str, str), Request("swap", "foo", "bar").withID(1))
assertMatch(Event("listening", None), Event('listening', ('127.0.0.1', 4550)))
assertMatch(Block, Block(0, 1, 'foo'))
assertNoMatch(Request("paws"), Request("swap", "foo", "bar").withID(1))
assertNoMatch(Request("swap", "foo", "bar").withID(1), Request("swap"))
assertNoMatch(('disconnect', ), Event("protocol-negociated", "SPARKv1"))
def testMatchSubclass(self):
""" match() should match a type if it is one of its parent types """
assertMatch(basestring, u"foo")
if __name__ == '__main__':
run_tests()
def initPatterns(self, loop, state):
super(TestServer, self).initPatterns(loop, state)
loop.suscribeTo(state.messenger.disconnected, result=False)
loop.addHandler(Request("swap", basestring, basestring), self)
def requestSwap(self, req, transID, a, b, state):
self.sendResponse(state, req, b, a)
class ProcessIntegrationTest(unittest.TestCase):
@processTimeout(1.0)
def testTcpSession(self):
with TestServer() as server:
server.listening.suscribe()
Process.send(server.pid, Command("bind", (BIND_ADDRESS, BIND_PORT)))
assertMatch(Event("listening", None), Process.receive())
with TcpMessenger() as client:
client.protocolNegociated.suscribe()
client.disconnected.suscribe()
client.connect((BIND_ADDRESS, BIND_PORT))
assertMatch(Event("protocol-negociated", basestring), Process.receive())
client.send(Request("swap", "foo", "bar").withID(1))
assertMatch(Response("swap", "bar", "foo").withID(1), Process.receive())
assertMatch(Event("disconnected"), Process.receive())
if __name__ == "__main__":
import logging
run_tests(level=logging.INFO)