open.source/magic-wormhole-transit-relay
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix more tests

Browse Source
This commit is contained in:
meejah 2021-02-12 00:35:52 -07:00
parent 1ab5e4ffb3
commit 9dfc410aa8
1 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/wormhole_transit_relay/test/test_transit_server.py
View File

@ -320,8 +320,7 @@ class Usage(ServerBase, unittest.TestCase):
# that will log the "empty" usage event
self.assertEqual(len(self._usage), 1, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[0]
self.assertEqual(result, "empty", self._usage)
self.assertEqual(self._usage[0]["mood"], "empty", self._usage)
def test_short(self):
p1 = self.new_protocol()
@ -340,8 +339,7 @@ class Usage(ServerBase, unittest.TestCase):
# that will log the "errory" usage event, then drop the connection
p1.transport.loseConnection()
self.assertEqual(len(self._usage), 1, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[0]
self.assertEqual(result, "errory", self._usage)
self.assertEqual(self._usage[0]["mood"], "errory", self._usage)
def test_lonely(self):
p1 = self.new_protocol()
@ -353,9 +351,8 @@ class Usage(ServerBase, unittest.TestCase):
p1.transport.loseConnection()
self.assertEqual(len(self._usage), 1, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[0]
self.assertEqual(result, "lonely", self._usage)
self.assertIdentical(waiting_time, None)
self.assertEqual(self._usage[0]["mood"], "lonely", self._usage)
self.assertIdentical(self._usage[0]["waiting_time"], None)
def test_one_happy_one_jilted(self):
p1 = self.new_protocol()
@ -375,9 +372,8 @@ class Usage(ServerBase, unittest.TestCase):
p1.transport.loseConnection()
self.assertEqual(len(self._usage), 1, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[0]
self.assertEqual(result, "happy", self._usage)
self.assertEqual(total_bytes, 20)
self.assertEqual(self._usage[0]["mood"], "happy", self._usage)
self.assertEqual(self._usage[0]["total_bytes"], 20)
self.assertNotIdentical(waiting_time, None)
def test_redundant(self):
@ -399,18 +395,15 @@ class Usage(ServerBase, unittest.TestCase):
p1c.transport.loseConnection()
self.assertEqual(len(self._usage), 1, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[0]
self.assertEqual(result, "lonely", self._usage)
self.assertEqual(self._usage[0]["mood"], "lonely")
p2.dataReceived(handshake(token1, side=side2))
self.assertEqual(len(self._transit_server._pending_requests), 0)
self.assertEqual(len(self._usage), 2, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[1]
self.assertEqual(result, "redundant", self._usage)
self.assertEqual(self._usage[1]["mood"], "redundant")
# one of the these is unecessary, but probably harmless
p1a.transport.loseConnection()
p1b.transport.loseConnection()
self.assertEqual(len(self._usage), 3, self._usage)
(started, result, total_bytes, total_time, waiting_time) = self._usage[2]
self.assertEqual(result, "happy", self._usage)
self.assertEqual(self._usage[2]["mood"], "happy")