Add git branch finder and test

deploystream/providers/git_plugin/plugin.py → deploystream/providers/git_provider/__init__.py

@@ -1,10 +1,13 @@
+import os
+from os.path import join, exists
+
 import git
 
 
-class GitPlugin(object):
+class GitProvider(object):
 
-    def __init__(self, **configuration):
-        pass
+    def __init__(self, code_dir='.'):
+        self.code_dir = code_dir
 
     def get_repo_branches_involved(self, feature_id):
         """
@@ -20,7 +23,17 @@ def get_repo_branches_involved(self, feature_id):
                 1: branch name
                 2: latest commit
         """
-        return []  # no info available for now
+        # Every folder in side self.code_dir that is a git repo will be looked
+        # at
+        repo_branches = []
+        for repo_name in os.listdir(self.code_dir):
+            repo_location = join(self.code_dir, repo_name)
+            if exists(join(repo_location, ".git")):
+                branches = self.get_branches_involved(repo_location,
+                                                      feature_id)
+                repo_branches.extend([(repo_name, ) + branch
+                                            for branch in branches])
+        return repo_branches
 
     def get_branches_involved(self, repo_location, feature_id):
         """
@@ -42,8 +55,9 @@ def get_branches_involved(self, repo_location, feature_id):
                         .format(repo_location=repo_location))
         remote = git.remote.Remote(repo, 'origin')
         affected = []
+        lower_feature = feature_id.lower()
         for remote_ref in remote.refs:
-            if feature_id in remote_ref.remote_head:
+            if lower_feature in remote_ref.remote_head.lower():
                 affected.append((remote_ref.remote_head,
                                  str(remote_ref.commit)))
 

deploystream/settings.py

@@ -1,8 +1,8 @@
-from local_settings import GITHUB_CONFIG
+from local_settings import GITHUB_CONFIG, GIT_CONFIG
 
 
 SOURCE_CODE_PLUGINS = [
-    ('deploystream.providers.git_plugin.plugin.GitPlugin', {}),
+    ('deploystream.providers.git_provider.GitProvider', GIT_CONFIG),
 ]
 
 PLANNING_PLUGINS = [

local_settings.py

@@ -5,3 +5,8 @@
     'repositories': [('pretenders', 'deploystream')],
     'token': TOKEN,
 }
+
+
+GIT_CONFIG = {
+    'code_dir': None
+}

tests/test_providers/data/__init__.py

@@ -0,0 +1 @@
+# This folder is used to house data for tests.

tests/test_providers/test_git_provider.py

@@ -0,0 +1,39 @@
+import os
+from os.path import join, exists, dirname
+
+from nose.tools import assert_equal
+
+from deploystream.providers.git_provider import GitProvider
+
+DUMMY_CODE_DIR = join(dirname(__file__), 'data')
+
+
+def ensure_dummy_clone_available():
+    """
+    Check that we have access to pretenderers' dummyrepo
+    """
+    folder_name = join(DUMMY_CODE_DIR, 'dummyrepo')
+    if not exists(folder_name):
+        os.system('git clone git+ssh://[email protected]/pretenders/dummyrepo {0}'
+                  .format(folder_name))
+    else:
+        os.system('git --git-dir={0} fetch'.format(folder_name))
+
+
+def test_git_plugin_finds_branches_across_repos():
+    """
+    Test that the GitPlugin finds branches in repos in the dir given.
+
+    Clone the dummyrepo into the data folder if not already there.
+
+    The data in this test is found by looking at the dummyrepo and getting
+    the branch names and latest commit of any branches that match "FeAtUrE".
+    """
+    ensure_dummy_clone_available()
+    provider = GitProvider(code_dir=DUMMY_CODE_DIR)
+    branches = provider.get_repo_branches_involved('FeAtUrE')
+
+    assert_equal([
+        ('dummyrepo', 'my/feature_branch',
+         'cf9130d3c07b061a88569153f10a7c7779338cfa'),
+        ], branches)