from tracker import json_utils, db
from tracker.test import test_data

from tracker import artists

from . import login_to_test_client as _login


def test_lookup(session):
    from tracker.test import add_and_flush
    sc_user, in_user, tracked_entity, *rest = add_and_flush(session,
        db.ScUser(scid=1, name="Find Me First", data=test_data.make_soundcloud_user_json(username='findmefirst')),
        db.InUser(inid='findmetoo', page_data=test_data.make_insta_user_json(full_name='Find Me Too')),
        db.TrackedEntity(eid=1, name="Find Me Yes", twitter_data=test_data.make_twitter_user_json(screen_name='findmeyes')),

        db.ScUser(scid=2, name="I'm not returned", data=test_data.make_soundcloud_user_json(username='notreturned')),
        db.InUser(inid='notfound', page_data=test_data.make_insta_user_json(full_name="Im not returned")),
        db.TrackedEntity(eid=2, name="Not found", twitter_data=test_data.make_twitter_user_json(screen_name='notfound')),
        db.TrackedEntity(eid=100, name="Influencer"),
        db.Following(influencer_eid=100, followee_eid=1),
    )
    artist = artists.create_association(sc_user=sc_user, in_user=in_user, tracked_entity=tracked_entity)
    arid = artist.arid

    session.execute("refresh materialized view entity_names")
    session.commit()
    session.execute("refresh materialized view twitter_artist_activity")
    session.commit()


    from api import api
    app = api.app.test_client()
    _login(app)

    rv = app.get('/api/artists/a/{}'.format(arid))

    data = json_utils.loads(rv.data)

    # print(json_utils.dumps(data, indent=2))
    assert len(data['results']) == 3