import os

from datetime import datetime, timedelta

from . import login_to_test_client as _login

from tracker import json_utils, db

from api import api



def test_can_list_teams_with_users_and_influencer_information(session):
    session.add(db.User(username='epic1', team='Epic'))
    session.add(db.User(username='epic2', team='Epic'))
    session.add(db.User(username='france1', team='Sony France'))

    session.add(db.SpyPlaylist(
        spyid='spy1',
        user='test',
        name='1 Playlist',
    ))
    session.add(db.SpyPlaylist(
        spyid='spy2',
        user='test',
        name='2 Playlist',
    ))
    session.add(db.UsersSpyPlaylists(username='epic1', spyid="spy1"))
    session.add(db.UsersSpyPlaylists(username='epic1', spyid="spy2"))

    session.add(db.ScUser(scid=1, name='sc1',))
    session.add(db.ScUser(scid=2, name='sc1',))
    session.add(db.UsersScInfluencers(username='epic1', scid=1))
    session.add(db.UsersScInfluencers(username='epic1', scid=2))

    session.add(db.YtChannel(ytid='yt1', name='yt1',))
    session.add(db.YtChannel(ytid='yt2', name='yt2',))
    session.add(db.YtChannel(ytid='yt3', name='yt3',))
    session.add(db.UsersYtChannels(username='epic1', ytid="yt1"))
    session.add(db.UsersYtChannels(username='epic1', ytid="yt2"))
    session.add(db.UsersYtChannels(username='epic1', ytid="yt3"))


    session.add(db.TrackedEntity(eid=1, name='twitter1'))
    session.add(db.TrackedEntity(eid=2, name='twitter2'))
    session.add(db.UsersTwInfluencers(username='epic1', eid=1))
    session.add(db.UsersTwInfluencers(username='epic1', eid=2))

    session.add(db.InUser(inid='in1'))
    session.add(db.InUser(inid='in2'))
    session.add(db.UsersInInfluencers(username='epic1', inid="in1"))
    session.add(db.UsersInInfluencers(username='epic1', inid="in2"))

    session.commit()
    app = api.app.test_client()
    _login(app)

    rv = app.get('/api/admin/users')

    data = json_utils.loads(rv.data)

    print(json_utils.dumps(data, indent=2))

    assert rv.status_code == 200

    assert len(data['users_influencers']) == 11

    assert len(data['users']) == 3

    epic1 = [u for u in data['users'] if u['username'] == 'epic1'][0]
    epic2 = [u for u in data['users'] if u['username'] == 'epic2'][0]
    france1 = [u for u in data['users'] if u['username'] == 'france1'][0]

    assert epic1['team'] == 'Epic'
    assert epic2['team'] == 'Epic'
    assert france1['team'] == 'Sony France'