"""Test jwt enabled services cache logic"""

import unittest
from unittest.mock import MagicMock


class jwtEnabledServices(unittest.TestCase):
    def setUp(self):
        self.jwt_enabled_service_client = MagicMock()

    def test_get_jwt_enabled_services(self):
        """Test for get_jwt_enabled_services."""

        mock_jwt_enabled_services = ['ows-account']
        self.jwt_enabled_service_client.getJwtEnabledServices = MagicMock(
            return_value=mock_jwt_enabled_services
        )
        result = self.jwt_enabled_service_client.getJwtEnabledServices()
        assert result == mock_jwt_enabled_services

    def test_cache_jwt_enabled_services(self):
        """Test caching of jwt_enabled_services."""

        expected_jwt_enabled_services = ['ows-account']
        self.jwt_enabled_service_client.setCache(
            'jwt_enabled_services', expected_jwt_enabled_services, 0
        )
        self.jwt_enabled_service_client.getCache = MagicMock(
            return_value=expected_jwt_enabled_services
        )
        result = self.jwt_enabled_service_client.getCache('jwt_enabled_services')
        assert result == expected_jwt_enabled_services

    def test_update_jwt_enabled_services_cache(self):
        """Test update cache."""

        expected_jwt_enabled_services = ['ows-account', 'ows-service']
        mock_jwt_enabled_services = ['ows-account']
        # Get existing cache data
        self.jwt_enabled_service_client.getCache = MagicMock(
            return_value=mock_jwt_enabled_services
        )
        result = self.jwt_enabled_service_client.getCache('jwt_enabled_services')
        # Append new item in a list
        result.append('ows-service')
        # Cache new jwt list
        self.jwt_enabled_service_client.setCache('jwt_enabled_services', result, 0)
        # Get the newly cached list
        self.jwt_enabled_service_client.getCache = MagicMock(return_value=result)
        result = self.jwt_enabled_service_client.getCache('jwt_enabled_services')
        assert result == expected_jwt_enabled_services