import json

import pytest

import os

from cleaner import app

def test_config_parser():
    os.environ["ENDPOINT_EXCLUDE_TAG"] = '{ "Key": "env", "Value": "prod" }'
    os.environ["NOTEBOOK_EXCLUDE_TAG"] = '{ "Key": "project", "Value": "ciritical" }'
    os.environ["MAX_COUNT"] = '100'
    
    config = app.parse_config()
    
    assert config["ENDPOINT_EXCLUDE_TAG"] ==  { 'Key': 'env', 'Value': 'prod' }
    assert config["NOTEBOOK_EXCLUDE_TAG"] ==  { 'Key': 'project', 'Value': 'ciritical' }
    assert config["MAX_COUNT"] ==  100
    
def test_config_parser_has_default():
    del os.environ["ENDPOINT_EXCLUDE_TAG"]
    del os.environ["NOTEBOOK_EXCLUDE_TAG"]
    del os.environ["MAX_COUNT"]
    
    config = app.parse_config()
    
    assert config["ENDPOINT_EXCLUDE_TAG"] == None 
    assert config["NOTEBOOK_EXCLUDE_TAG"] == None
    assert config["MAX_COUNT"] == 100
    
def test_try_parse_env():
    os.environ["ENDPOINT_EXCLUDE_TAG"] = '{ "Key": "env", "Value": "prod" }'
    oneEnv = app.try_parse_env("ENDPOINT_EXCLUDE_TAG")
    assert oneEnv == { 'Key': 'env', 'Value': 'prod' }
    
def test_try_parse_env_invalid_json():
    os.environ["ENDPOINT_EXCLUDE_TAG"] = "MAL_FORMED_JSON:{}"
    with pytest.raises(Exception):
        oneEnv = app.try_parse_env("ENDPOINT_EXCLUDE_TAG")
        
def test_is_serverless():
    class MockSageMakerClient:
        def describe_endpoint(*args, **kwargs):
            assert kwargs["EndpointName"] == "mock-endpoint"
            return {
                "EndpointConfigName": "mock-endpoint-config"
            }
        def describe_endpoint_config(*args, **kwargs):
            assert kwargs["EndpointConfigName"] == "mock-endpoint-config"
            return {
                "ProductionVariants": [
                    {
                        "VariantName": "mock-variant-name",
                        "ServerlessConfig": {}
                    }
                ]
            }
                        
    client = MockSageMakerClient()
    assert app.is_serverless_endpoint(client, "mock-endpoint") == True