arestclient_test.py

#!/usr/bin/env python
# Copyright 2012-2016 Jonathan Paugh and contributors
# See COPYING for license details
from arestclient.GitHub import GitHub
from arestclient.base import IncompleteRequest
import unittest


class Client(object):
    http_methods = ('demo', 'test')

    def __init__(self, username=None, password=None, token=None,
                 connection_properties=None):
        pass

    def setConnectionProperties(self, props):
        pass

    def demo(self, *args, **params):
        return self.methodCalled('demo', *args, **params)

    def test(self, *args, **params):
        return self.methodCalled('test', *args, **params)

    def methodCalled(self, methodName, *args, **params):
        return {
            'methodName': methodName,
            'args': args,
            'params': params
        }


class TestGitHubObjectCreation(unittest.TestCase):
    def test_user_pw(self):
        gh = GitHub('korfuri', '1234')
        self.assertTrue(gh is not None)

        gh = GitHub(username='korfuri', password='1234')
        self.assertTrue(gh is not None)

    def test_token(self):
        gh = GitHub(username='korfuri', token='deadbeef')
        self.assertTrue(gh is not None)

        gh = GitHub(token='deadbeef')
        self.assertTrue(gh is not None)

    def test_token_password(self):
        with self.assertRaises(TypeError):
            GitHub(username='korfuri', password='1234', token='deadbeef')


class TestIncompleteRequest(unittest.TestCase):

    def newIncompleteRequest(self):
        return IncompleteRequest(Client())

    def test_pathByGetAttr(self):
        rb = self.newIncompleteRequest()
        rb.hug.an.octocat
        self.assertEqual(rb.url, "/hug/an/octocat")

    def test_callMethodDemo(self):
        rb = self.newIncompleteRequest()
        self.assertEqual(
            rb.path.demo(),
            {
                "methodName": "demo",
                "args": (),
                "params": {"url": "/path"}
            }
        )

    def test_pathByGetItem(self):
        rb = self.newIncompleteRequest()
        rb["hug"][1]["octocat"]
        self.assertEqual(rb.url, "/hug/1/octocat")

    def test_callMethodTest(self):
        rb = self.newIncompleteRequest()
        self.assertEqual(
            rb.path.demo(),
            {
                "methodName": "demo",
                "args": (),
                "params": {"url": "/path"}
            }
        )


def test_github():
    g = GitHub()
    status, data = g.users.octocat.get()
    assert data.get('name') == 'The Octocat'
    assert status == 200
    # Workaround to https://github.com/mozilla/agithub/issues/67
    response_headers = dict([(x.lower(), y) for x, y in g.getheaders()])
    assert (
            response_headers.get('Content-Type'.lower()) ==
            'application/json; charset=utf-8')


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