# What is this?
## This tests the batch update spend logic on the proxy server


import asyncio
import os
import random
import sys
import time
import traceback
from datetime import datetime

from dotenv import load_dotenv
from fastapi import Request

load_dotenv()

sys.path.insert(
    0, os.path.abspath("../..")
)  # Adds the parent directory to the system path

import logging
from litellm.proxy.management_endpoints.sso_helper_utils import (
    check_is_admin_only_access,
    has_admin_ui_access,
)
from litellm.proxy._types import LitellmUserRoles


def test_check_is_admin_only_access():
    assert check_is_admin_only_access("admin_only") is True
    assert check_is_admin_only_access("user_only") is False


def test_has_admin_ui_access():
    assert has_admin_ui_access(LitellmUserRoles.PROXY_ADMIN.value) is True
    assert has_admin_ui_access(LitellmUserRoles.PROXY_ADMIN_VIEW_ONLY.value) is True
    assert has_admin_ui_access(LitellmUserRoles.INTERNAL_USER_VIEW_ONLY.value) is False