"""test daemon handler""" import pytest from unittest.mock import MagicMock from unittest.mock import NonCallableMagicMock from unittest.mock import patch from daemonJwtHandler import daemon_handler from daemonJwtHandler.logic.jwt_services import JwtServices from daemonJwtHandler.constants import auth mock_jwt_enabled_services = ['ows-account'] mock_values = { auth.JWT_ENABLED_SERVICES_SECRET_KEY: mock_jwt_enabled_services, auth.JWT_SECRET_KEY: 'Jwttoken' } def test_daemon_handler_auth(): """Test daemon handler to forward jwt""" service_name = 'ows-account' daemon_handler.JwtServices = MockCaching() result = daemon_handler.daemon_handler(service_name) expected_values = auth.AUTHORIZATION_FORMAT.format( jwt_token=mock_values[auth.JWT_SECRET_KEY]) print(expected_values) print(result) assert result == expected_values def test_daemon_handler_none(): """Test daemon handler to forward jwt""" service_name = 'ows-payee' daemon_handler.JwtServices = MockCaching() result = daemon_handler.daemon_handler(service_name) assert not result class MockCaching(object): """Mock for utcnow and strptime.""" @staticmethod def getInstance(): """Set mock value for getCache.""" return MockCaching() @staticmethod def getCache(key): """Set mock value for getCache.""" return mock_values[key]