Add tag check to for-each-ref replacementbenbrown/old-git-sha-contains

1 files changed, 5 insertions, 4 deletions

diff --git a/ybd/repos.py b/ybd/repos.py
index 1b4da71..0deb9b8 100644
--- a/ybd/repos.py
+++ b/ybd/repos.py
@@ -113,15 +113,16 @@ def ensure_ref_contains_sha(dn, gitdir, ref, sha):
         if ref == sha:
             return
 
-        def sha_in_ref():
+        def sha_in_ref(tag=False):
+            arg = "tag" if tag else "branch"
             try:
-                return bool(check_output(['git', 'branch', '--contains',
+                return bool(check_output(['git', arg, '--contains',
                                           sha, ref], stderr=fnull,
                                          universal_newlines=True).strip())
             except:
                 return False
 
-        if not sha_in_ref():
+        if not sha_in_ref() and not sha_in_ref(tag=True):
             # The sha may currently exist in another branch,
             # force an update of the provided ref.
             repo_url = get_repo_url(dn['repo'])
@@ -129,7 +130,7 @@ def ensure_ref_contains_sha(dn, gitdir, ref, sha):
                   '+refs/*/{0}:refs/*/{0}'.format(ref)],
                  stdout=fnull, stderr=fnull)
 
-        if not sha_in_ref():
+        if not sha_in_ref() and not sha_in_ref(tag=True):
             exit = check == 'exit'
             app.log(dn, 'WARNING: unable to find %s within' %
                     sha, ref, exit=exit)