Module NetworkTest
[hide private]
[frames] | no frames]

Source Code for Module NetworkTest

 1  ##################################################################### 
 2  # -*- coding: iso-8859-1 -*-                                        # 
 3  #                                                                   # 
 4  # Frets on Fire                                                     # 
 5  # Copyright (C) 2006 Sami Kyöstilä                                  # 
 6  #                                                                   # 
 7  # This program is free software; you can redistribute it and/or     # 
 8  # modify it under the terms of the GNU General Public License       # 
 9  # as published by the Free Software Foundation; either version 2    # 
10  # of the License, or (at your option) any later version.            # 
11  #                                                                   # 
12  # This program is distributed in the hope that it will be useful,   # 
13  # but WITHOUT ANY WARRANTY; without even the implied warranty of    # 
14  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the     # 
15  # GNU General Public License for more details.                      # 
16  #                                                                   # 
17  # You should have received a copy of the GNU General Public License # 
18  # along with this program; if not, write to the Free Software       # 
19  # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,        # 
20  # MA  02110-1301, USA.                                              # 
21  ##################################################################### 
22   
23  import unittest 
24  import Network 
25  import time 
26   
27 -class TestConnection(Network.Connection):
28 - def handlePacket(self, packet):
29 self.packet = packet
30
31 -class TestServer(Network.Server):
32 - def createConnection(self, sock):
33 return TestConnection(sock)
34
35 -class NetworkTest(unittest.TestCase):
36 - def testHandshake(self):
37 s = TestServer() 38 c = TestConnection() 39 c.connect("localhost") 40 41 c.sendPacket("moikka") 42 43 Network.communicate(100) 44 client = s.clients.values()[0] 45 assert client.packet == "moikka" 46 assert client.id == 1
47
48 - def tearDown(self):
50 51 if __name__ == "__main__": 52 unittest.main() 53