1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
|
from builtins import next
import datetime
from unittest import TestCase
from configparser import RawConfigParser
import pytz
import responses
from bugwarrior.services.github import GithubService
from .base import ServiceTest, AbstractServiceTest
class TestGithubIssue(AbstractServiceTest, ServiceTest):
maxDiff = None
SERVICE_CONFIG = {
'github.login': 'arbitrary_login',
'github.password': 'arbitrary_password',
'github.username': 'arbitrary_username',
}
arbitrary_created = (
datetime.datetime.utcnow() - datetime.timedelta(hours=1)
).replace(tzinfo=pytz.UTC, microsecond=0)
arbitrary_updated = datetime.datetime.utcnow().replace(
tzinfo=pytz.UTC, microsecond=0)
arbitrary_issue = {
'title': 'Hallo',
'html_url': 'http://whanot.com/',
'url': 'https://api.github.com/repos/arbitrary_username/arbitrary_repo/issues/1',
'number': 10,
'body': 'Something',
'user': {'login': 'arbitrary_login'},
'milestone': {'title': 'alpha'},
'labels': [{'name': 'bugfix'}],
'created_at': arbitrary_created.isoformat(),
'updated_at': arbitrary_updated.isoformat(),
'repo': 'ralphbean/bugwarrior',
}
arbitrary_extra = {
'project': 'one',
'type': 'issue',
'annotations': [],
}
def setUp(self):
super(TestGithubIssue, self).setUp()
self.service = self.get_mock_service(GithubService)
def test_normalize_label_to_tag(self):
issue = self.service.get_issue_for_record(
self.arbitrary_issue,
self.arbitrary_extra
)
self.assertEqual(issue._normalize_label_to_tag('needs work'),
'needs_work')
def test_to_taskwarrior(self):
self.service.import_labels_as_tags = True
issue = self.service.get_issue_for_record(
self.arbitrary_issue,
self.arbitrary_extra
)
expected_output = {
'project': self.arbitrary_extra['project'],
'priority': self.service.default_priority,
'annotations': [],
'tags': ['bugfix'],
issue.URL: self.arbitrary_issue['html_url'],
issue.REPO: self.arbitrary_issue['repo'],
issue.TYPE: self.arbitrary_extra['type'],
issue.TITLE: self.arbitrary_issue['title'],
issue.NUMBER: self.arbitrary_issue['number'],
issue.UPDATED_AT: self.arbitrary_updated,
issue.CREATED_AT: self.arbitrary_created,
issue.BODY: self.arbitrary_issue['body'],
issue.MILESTONE: self.arbitrary_issue['milestone']['title'],
issue.USER: self.arbitrary_issue['user']['login'],
}
actual_output = issue.to_taskwarrior()
self.assertEqual(actual_output, expected_output)
@responses.activate
def test_issues(self):
self.add_response(
'https://api.github.com/user/repos?per_page=100',
json=[{
'name': 'some_repo',
'owner': {'login': 'some_username'}
}])
self.add_response(
'https://api.github.com/users/arbitrary_username/repos?per_page=100',
json=[{
'name': 'arbitrary_repo',
'owner': {'login': 'arbitrary_username'}
}])
self.add_response(
'https://api.github.com/repos/arbitrary_username/arbitrary_repo/issues?per_page=100',
json=[self.arbitrary_issue])
self.add_response(
'https://api.github.com/user/issues?per_page=100',
json=[self.arbitrary_issue])
self.add_response(
'https://api.github.com/repos/arbitrary_username/arbitrary_repo/issues/10/comments?per_page=100',
json=[{
'user': {'login': 'arbitrary_login'},
'body': 'Arbitrary comment.'
}])
issue = next(self.service.issues())
expected = {
'annotations': [u'@arbitrary_login - Arbitrary comment.'],
'description': u'(bw)Is#10 - Hallo .. http://whanot.com/',
'githubbody': u'Something',
'githubcreatedon': self.arbitrary_created,
'githubmilestone': u'alpha',
'githubnumber': 10,
'githubrepo': 'arbitrary_username/arbitrary_repo',
'githubtitle': u'Hallo',
'githubtype': 'issue',
'githubupdatedat': self.arbitrary_updated,
'githuburl': u'http://whanot.com/',
'githubuser': u'arbitrary_login',
'priority': 'M',
'project': 'arbitrary_repo',
'tags': []}
self.assertEqual(issue.get_taskwarrior_record(), expected)
class TestGithubService(TestCase):
def test_token_authorization_header(self):
config = RawConfigParser()
config.interactive = False
config.add_section('general')
config.add_section('mygithub')
config.set('mygithub', 'service', 'github')
config.set('mygithub', 'github.login', 'tintin')
config.set('mygithub', 'github.username', 'tintin')
config.set('mygithub', 'github.token',
'@oracle:eval:echo 1234567890ABCDEF')
service = GithubService(config, 'general', 'mygithub')
self.assertEqual(service.client.session.headers['Authorization'],
"token 1234567890ABCDEF")
|