def test_can_transmit(self):
Node.random_ranges = [8, 16, 32]
Node.max_attempts = 10
node_a = Node()
node_a.back_off = 1
node_b = Node()
node_b.back_off = 3
self.assertEqual(node_a.can_transmit(), False)
self.assertEqual(node_b.can_transmit(), False)
node_a.tick()
node_b.tick()
self.assertEqual(node_a.can_transmit(), True)
self.assertEqual(node_b.can_transmit(), False)
def test_send_packet(self):
Node.random_ranges = [8, 16, 32]
Node.max_attempts = 3
node_a = Node()
node_a.current_attempt = 2
node_a.back_off = 0
node_a.send_packet()
self.assertEqual(node_a.current_attempt, 0)
self.assertEqual(node_a.num_transmits, 1)