class Test(unittest.TestCase):
def setUp(self):
self.camRede = CamadaRede()
self.camTransOrigem = CamadaTransporte(1, self.camRede)
self.camTransDestino = CamadaTransporte(2, self.camRede)
self.camRede.origem = self.camTransOrigem
self.camRede.destino = self.camTransDestino
self.camAplicOrigem = CamadaAplicacao(1, self.camTransOrigem)
self.camAplicDestino = CamadaAplicacao(2, self.camTransDestino)
self.camAplicOrigem.aplicacao= self
self.camAplicDestino.aplicacao = self
self.camTransOrigem.camadaAplicacao = self.camAplicOrigem
self.camTransDestino.camadaAplicacao = self.camAplicDestino
self.camAplicDestino.start()
self.camAplicOrigem.start()
def tearDown(self):
self.camRede = None
self.camTransOrigem = None
self.camTransDestino = None
self.camAplicOrigem = None
self.camAplicDestino = None
def testeConexaoOrigemDestino(self):
self.camAplicOrigem.conectar(2)
self.assertTrue(self.camAplicOrigem.cid > 0)
self.assertTrue(self.camAplicDestino.cid > 0)
def testeConexaoDestinoOrigem(self):
self.camAplicDestino.conectar(1)
self.assertTrue(self.camAplicOrigem.cid > 0)
self.assertTrue(self.camAplicDestino.cid > 0)
def testeEnviarPacoteOrigemDestino(self):
self.camAplicOrigem.conectar(2)
self.camAplicOrigem.enviarMensagem("Teste")
self.assertEquals(("Teste",2), self.msgRecebida)
def testeEnviarPacoteDestinoOrigem(self):
self.camAplicDestino.conectar(1)
self.camAplicDestino.enviarMensagem("Teste")
self.assertEquals(("Teste",1), self.msgRecebida)
def testeDesconectarOrigem(self):
self.camAplicOrigem.conectar(2)
self.assertEquals(0, self.camAplicOrigem.fecharConexao())
def testeDesconectarDestino(self):
self.camAplicDestino.conectar(1)
self.assertEquals(0, self.camAplicOrigem.fecharConexao())
def receberMensagem(self, msg, id):
self.msgRecebida = (msg, id)
#Criacao da camada de rede
camRede = CamadaRede()
#Criacao das camadas de transporte utilizadas no programa.
camTransOrigem = CamadaTransporte(1, camRede)
camTransDestino = CamadaTransporte(2, camRede)
#Associacao da camada de rede com as camadas de transporte.
camRede.origem = camTransOrigem
camRede.destino = camTransDestino
#Criacao das camadas de aplicacao utilizadas no programa.
camAplicOrigem = CamadaAplicacao(1, camTransOrigem)
camAplicDestino = CamadaAplicacao(2, camTransDestino)
#Associacao das camadas de transporte com as suas camadas de aplicacao
#correspondentes.
camTransOrigem.camadaAplicacao = camAplicOrigem
camTransDestino.camadaAplicacao = camAplicDestino
#Inicializacao das camadas de aplizacao
camAplicDestino.start()
camAplicOrigem.start()
#Criacao da interface grafica do programa.
interface = Interface(camAplicOrigem, camAplicDestino)
#Inicializacao da interface grafica.
interface.iniciar()
