from inventory_provider.tasks import worker
from inventory_provider.tasks.common import _get_redis


def backend_db():
    return _get_redis({
        'redis': {
            'hostname': None,
            'port': None
        },
        'redis-databases': [0, 7]
    }).db


def test_clear_classifier_cache(
        router,
        mocked_redis,
        data_config,
        classifier_cache_test_entries):
    worker.InventoryTask.config = data_config
    backend_db().update(classifier_cache_test_entries)
    worker.clear_cached_classifier_responses(router)
    for k in backend_db():
        assert not k.startswith('classifier-cache:%s:' % router)