Store additional context

changes/changelog.py

@@ -87,3 +87,4 @@ def generate_changelog(context):
         dry_run=context.dry_run
     )
     log.info('Added content to CHANGELOG.md')
+    context.changelog_content = changelog_content

changes/config.py

@@ -3,19 +3,23 @@
 import click
 from giturlparse import parse
 from plumbum.cmd import git
+from plumbum import local
 import yaml
 
 CONFIG_FILE = '.changes'
 DEFAULTS = {
     'changelog': 'CHANGELOG.md',
     'readme': 'README.md',
+    'gh_token': None,
 }
 
 
 class CLI(object):
     test_command = None
     pypi = None
     skip_changelog = None
+    changelog_content = None
+    parsed_repo = None
 
     def __init__(self, module_name, dry_run, debug, no_input, requirements,
                  new_version, current_version, repo_url, version_prefix):
@@ -32,8 +36,28 @@ def __init__(self, module_name, dry_run, debug, no_input, requirements,
         self.current_version = current_version
         self.repo_url = repo_url
 
-        self.parsed_repo = parse(git('config --get remote.origin.url'.split(' ')))
 
+    @property
+    def repo(self):
+        return self.parsed_repo.repo
+
+    @property
+    def owner(self):
+        return self.parsed_repo.owner
+
+    @property
+    def github(self):
+        return self.parsed_repo.github
+
+    @property
+    def bitbucket(self):
+        return self.parsed_repo.bitbucket
+
+
+    @property
+    def parsed_repo(self):
+        with local.cwd(local.cwd / self.module_name):
+            return parse(git('config --get remote.origin.url'.split(' ')))
 
     @property
     def repo(self):

tests/__init__.py

@@ -10,6 +10,7 @@
 
 module_name = 'test_app'
 context = CLI(module_name, True, True, True, '%s/requirements.txt' % module_name, '0.0.2', '0.0.1', 'https://github.com/someuser/test_app', None)
+context.gh_token = 'foo'
 context.requirements = '%s/requirements.txt' % module_name
 context.tmp_file = '%s/__init__.py' % module_name
 context.initial_init_content = [
@@ -31,8 +32,10 @@ def setup():
     with open(context.requirements, 'w') as req_file:
         req_file.write('pytest')
 
-    with local.cwd(local.cwd / module_name):
-        git('init', module_name)
+    with local.cwd(local.cwd / context.module_name):
+        git('init')
+        git('remote', 'add', 'origin', 'https://github.com/michaeljoseph/test_app.git')
+
 
 def teardown():
     if os.path.exists(context.tmp_file):

tests/test_changelog.py

@@ -45,3 +45,4 @@ def test_replace_sha_with_commit_link():
 
 def test_generate_changelog():
     changelog.generate_changelog(context)
+    assert isinstance(context.changelog_content, list)

tests/test_config.py

@@ -27,5 +27,6 @@ def test_malformed_config_returns_none():
 
 def test_parsed_repo_url():
     assert context.owner == 'michaeljoseph'
+    assert context.repo == 'test_app'
     assert context.github
     assert not context.bitbucket