import secret_loader


def test_init():
    loader = secret_loader.SecretLoader('TEST', hardcoded={'KEY': 'VALUE'})
    assert loader.load('KEY') == 'VALUE'

def test_lookup_unknown():
    loader = secret_loader.SecretLoader('TEST', hardcoded={})
    assert loader.load('UNKNOWN') is None