def setUp(self): self.helper = TestHelper() self.env = simpy.Environment() self.network = util.Network(self.env, self.helper.settings) self.peer_a_id = bs.Bits(uint=0, length=16) self.peer_b_id = bs.Bits(uint=1, length=16) self.peer_a = unittest.mock.Mock(spec=peer.Peer) self.peer_b = unittest.mock.Mock(spec=peer.Peer) self.peer_a_address = self.network.register(self.peer_a) self.peer_b_address = self.network.register(self.peer_b)