fastly-py/test/api_test.py

import unittest

import fastly


class APITest(unittest.TestCase):

    def setUp(self):
        self.api = fastly.API()

    def test_purge(self):
        self.assertTrue(self.api.purge_url('test.com', '/'))

    def test_purge_by_key(self):
        self.api.deauthenticate()
        self.api.authenticate_by_key('TESTAPIKEY')
        self.assertTrue(self.api.purge_key('test.com', 'foo'))

    def test_cookie_purge_by_key(self):
        self.api.deauthenticate()
        self.api.authenticate_by_password('foo@example.com', 'password')
        with self.assertRaises(fastly.AuthenticationError):
            self.api.purge_key('test.com', 'foo')

    def test_auth_error(self):
        self.api.deauthenticate()
        with self.assertRaises(fastly.AuthenticationError):
            self.api.conn.request('GET', '/current_customer')

    def test_auth_key_success(self):
        self.api.deauthenticate()
        self.api.authenticate_by_key('TESTAPIKEY')
        self.api.conn.request('GET', '/current_customer')

    def test_auth_session_success(self):
        self.api.deauthenticate()
        self.api.authenticate_by_password('foo@example.com', 'password')
        self.api.conn.request('GET', '/current_customer')

if __name__ == '__main__':
    unittest.main()