import unittest

from collections import namedtuple

Request = namedtuple("Request", ["arrived_at", "time_to_process"])
Response = namedtuple("Response", ["was_dropped", "started_at"])

import sources.network_packet
from sources.network_packet import process_requests


class MyTestCase(unittest.TestCase):
    def test(self):
        buffer_max_size = 1
        buf = sources.network_packet.Buffer(buffer_max_size)
        requests = []
        responses = process_requests(requests, buf)
        self.assertEqual(0, len(responses))

    def test1(self):
        buffer_max_size = 1
        buf = sources.network_packet.Buffer(buffer_max_size)
        requests = [Request(0, 1)]
        responses = process_requests(requests, buf)
        self.assertEqual(1, len(responses))
        self.assertFalse(responses[0].was_dropped)
        self.assertEqual(0, responses[0].started_at)

    def test2(self):
        buffer_max_size = 1
        buf = sources.network_packet.Buffer(buffer_max_size)
        requests = [Request(0, 1), Request(0, 1)]
        responses = process_requests(requests, buf)
        self.assertEqual(2, len(responses))
        self.assertFalse(responses[0].was_dropped)
        self.assertEqual(0, responses[0].started_at)
        self.assertTrue(responses[1].was_dropped)

    def test3(self):
        buffer_max_size = 1
        buf = sources.network_packet.Buffer(buffer_max_size)
        requests = [Request(0, 1), Request(1, 1)]
        responses = process_requests(requests, buf)
        self.assertEqual(2, len(responses))
        self.assertFalse(responses[0].was_dropped)
        self.assertEqual(0, responses[0].started_at)
        self.assertFalse(responses[1].was_dropped)
        self.assertEqual(1, responses[1].started_at)


if __name__ == '__main__':
    unittest.main()