import datetime
from twython import Twython, TwythonError, TwythonAuthError

from .config import (
    app_key, app_secret, oauth_token, oauth_token_secret,
    protected_twitter_1, protected_twitter_2, screen_name,
    test_tweet_id, test_list_slug, test_list_owner_screen_name,
    access_token, test_tweet_object, test_tweet_html, test_account_id, test_funding_instrument_id, unittest
)

import time
from twython.api_ads import TwythonAds


class TwythonEndpointsTestCase(unittest.TestCase):
    def setUp(self):

        client_args = {
            'headers': {
                'User-Agent': '__twython__ Test'
            },
            'allow_redirects': False
        }

        # This is so we can hit coverage that Twython sets
        # User-Agent for us if none is supplied
        oauth2_client_args = {
            'headers': {}
        }

        self.api = TwythonAds(app_key, app_secret,
                           oauth_token, oauth_token_secret,
                           client_args=client_args)

        self.oauth2_api = Twython(app_key, access_token=access_token,
                                  client_args=oauth2_client_args)

    def test_get_accounts(self):
        accounts = self.api.get_accounts()
        self.assertTrue(len(accounts) > 0)

    def test_get_account(self):
        account = self.api.get_account(test_account_id)
        self.assertEqual(account['id'], test_account_id)
        with self.assertRaises(TwythonError):
            self.api.get_account('1234')

    def test_get_funding_instruments(self):
        funding_instruments = self.api.get_funding_instruments(test_account_id)
        self.assertTrue(len(funding_instruments) > 0)

    def test_get_funding_instrument(self):
        funding_instrument = self.api.get_funding_instrument(test_account_id, test_funding_instrument_id)
        self.assertEqual(funding_instrument['id'], test_funding_instrument_id)
        self.assertEqual(funding_instrument['account_id'], test_account_id)
        with self.assertRaises(TwythonError):
            self.api.get_funding_instrument('1234', '1234')

    def test_get_campaigns(self):
        campaigns = self.api.get_campaigns(test_account_id)
        self.assertTrue(len(campaigns) > 0)

    def test_create_campaign(self):
        new_campaign = {
            'name': 'Test Twitter campaign - Twython',
            'funding_instrument_id': test_funding_instrument_id,
            'start_time': datetime.datetime.now().strftime('%Y-%m-%dT%H:%M:%SZ'),
            'daily_budget_amount_local_micro': 10 * 1000000
        }
        campaign = self.api.create_campaign(test_account_id, **new_campaign)
        self.assertEqual(campaign['account_id'], test_account_id)
        self.assertIsNotNone(campaign['id'])