import base

import github3

from base import expect

from github3.users import User, Plan, Key

from github3.events import Event

from warnings import catch_warnings, simplefilter

class TestUser(base.BaseTest):

def __init__(self, methodName='runTest'):

super(TestUser, self).__init__(methodName)

self.user = github3.user(self.sigm)

def test_user(self):

expect(self.user).isinstance(User)

def test_disk_usage(self):

expect(self.user.disk_usage) >= 0

def test_for_hire(self):

with catch_warnings():

simplefilter('ignore')

self.user.for_hire

def test_hireable(self):

expect(self.user.hireable).isinstance(bool)

def test_is_assignee_on(self):

expect(self.user.is_assignee_on(self.sigm, self.todo)).is_True()

expect(self.user.is_assignee_on(self.kr, 'requests')).is_False()

def test_list_events(self):

self.expect_list_of_class(self.user.list_events(), Event)

def test_list_followers(self):

self.expect_list_of_class(self.user.list_followers(), User)

def test_list_following(self):

self.expect_list_of_class(self.user.list_following(), User)

def test_list_received_events(self):

self.expect_list_of_class(self.user.list_received_events(), Event)

def test_owned_private_repos(self):

expect(self.user.owned_private_repos) >= 0

def test_total_private_gists(self):

expect(self.user.total_private_gists) >= 0

def test_private_gists(self):

with catch_warnings():

simplefilter('ignore')

self.user.private_gists

def test_plan(self):

if self.user.plan:

expect(self.user.plan).isinstance(Plan)

def test_public_gists(self):

expect(self.user.public_gists) >= 0

def test_total_private_repos(self):

expect(self.user.total_private_repos) >= 0

def test_requires_auth(self):

with expect.raises(github3.GitHubError):

self.user.add_email_addresses(['foo@example.com',

'graff@colmin.gov'])

self.user.delete_email_addresses(['foo@example.com'])

self.user.list_org_events(self.gh3py)

self.user.update()

def test_with_auth(self):

if not self.auth:

return

user = self._g.user()

email = 'new_email@gmail.com'

addresses = user.add_email_addresses([email])

expect(addresses).isinstance(list)

expect(user.delete_email_addresses([email])).is_True()

addresses = user.add_email_address(email)

expect(addresses).isinstance(list)

expect(user.delete_email_address(email)).is_True()

try:

ev = user.list_org_events(self.gh3py)

self.expect_list_of_class(ev, Event)

except github3.GitHubError:

pass

expect(user.update(user.name, user.email, user.blog, user.company,

user.location, user.hireable, user.bio)).isinstance(bool)

class TestKey(base.BaseTest):

def __init__(self, methodName='runTest'):

super(TestKey, self).__init__(methodName)

if self.auth:

self.key = self._g.list_keys()[0]

else:

json = {

'url': 'https://api.github.com/user/keys/id',

'verified': True,

'id': 999999,

'key': 'ssh-rsa AAAAB4...',

'title': 'fake'

}

self.key = Key(json)

def test_key(self):

expect(self.key).isinstance(Key)

def test_pubkey(self):

expect(self.key.key).isinstance(base.str_test)

def test_id(self):

expect(self.key.id) > 0

def test_title(self):

expect(self.key.title).isinstance(base.str_test)

def test_requires_auth(self):

if not self.auth:

with expect.raises(github3.GitHubError):

self.key.update('title', 'ssha-rsa AAAAB2...')

self.key.delete()