import unittest
from apidirectory import ApiDirectory


# Test calling the API tirectory.
class ApiDirectoryTest(unittest.TestCase):
    client_id = "some-guid-goes-here"
    secret = "secret-string-goes-here"
    scope = "iamgroups"
    token_url = "https://apigw-tst.it.umich.edu/um/inst/oauth2/token"

    # Test the construtor.
    def test_constructor(self):
        api = ApiDirectory(self.client_id, self.secret, self.scope, self.token_url)
        self.assertEqual(api.client_id, self.client_id)
        self.assertEqual(api.secret, self.secret)
        self.assertEqual(api.scope, self.scope)
        self.assertEqual(api.token_url, self.token_url)
        self.assertEqual(api.timeout, 10)

    # Test the header.
    def test_build_header(self):
        api = ApiDirectory(self.client_id, self.secret, self.scope, self.token_url)

        header = api.build_headers()
        self.assertEqual(header['x-ibm-client-id'], self.client_id)
        self.assertTrue('authorization' in header)