from unittest import TestCase

from ipc_unix import pubsub
from tests import get_temp_file_path


class PubSubTestCase(TestCase):
    def setUp(self):
        self.socket_path = get_temp_file_path()
        self.publisher = pubsub.Publisher(self.socket_path)
        self.subscriber = pubsub.Subscriber(self.socket_path)

    def tearDown(self):
        self.publisher.close()
        self.subscriber.close()

    def test_transmits(self):
        self.publisher.write({"foo": "bar"})
        response = self.subscriber.get_message()
        self.assertEqual(response, {"foo": "bar"})

    def test_buffers_messages(self):
        for i in range(5):
            self.publisher.write(i)
        messages = []
        for i in range(5):
            messages.append(self.subscriber.get_message())
        self.assertEqual(messages, [0, 1, 2, 3, 4])