import pathlib
import unittest


class BeeDataHandlerServiceTests(unittest.TestCase):
    def test_listener_restarts_after_clean_redis_disconnect(self):
        service_path = (
            pathlib.Path(__file__).parents[1]
            / "deploy"
            / "systemd"
            / "bee_data_handler.service"
        )
        service = service_path.read_text(encoding="utf-8")

        self.assertIn("Restart=always", service)
        self.assertIn("RestartSec=5s", service)
        self.assertIn("EnvironmentFile=/opt/.env", service)


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