import datetime

from spotify_charts_notifications import redis_db


def test_remove_old_keys(redis_client):
    today = datetime.date.today().strftime("%Y-%m-%d")
    redis_client.mset({
        redis_db.get_key(f'lock'): '',
        redis_db.get_key(f'us/{today}/foo/bar'): '',
        redis_db.get_key(f'gb/{today}/foo/bar'): '',
        'another/key/foo/bar': '',
        # should be removed
        redis_db.get_key('us/2019-09-09/foo/bar'): '',
        redis_db.get_key('us/2019-08-06/foo/bar'): '',
    })
    redis_db.remove_old_keys(redis_client, today, 'us')
    keys = redis_client.keys()
    assert len(keys) == 4
    assert redis_db.get_key(f'us/{today}/foo/bar').encode() in keys
    assert redis_db.get_key(f'gb/{today}/foo/bar').encode() in keys
    assert redis_db.get_key('lock').encode() in keys
    assert 'another/key/foo/bar'.encode() in keys