add support for codebuild github PR refs pattern

src/build.py

@@ -27,9 +27,14 @@ def __init__(self, build_event):
 
     def get_pr_id(self):
         """If this build was for a PR branch, returns the PR ID, otherwise returns None."""
-        matches = re.match(r'^pr\/(\d+)', self._get_build_details().get('sourceVersion', ""))
+        source_version = self._get_build_details().get('sourceVersion', "")
+        matches = re.match(r'^pr\/(\d+)', source_version)
         if not matches:
-            return None
+            # check for refs pattern
+            pattern = r'refs/pull/(\d+)/head'
+            matches = re.search(pattern, source_version)
+            if not matches:
+                return None
         return int(matches.group(1))
 
     @property

test/unit/test_build.py

@@ -52,6 +52,12 @@ def test_get_pr_id_is_pr(mocker, mock_codebuild):
     assert build_obj.get_pr_id() == 123
 
 
+def test_get_pr_id_is_pr_refs_pattern(mocker, mock_codebuild):
+    _mock_build_details('refs/pull/123/head^{2adff25fccedf7c08117b3fc93a7eca896c19060}')
+    build_obj = build.Build(_mock_build_event())
+    assert build_obj.get_pr_id() == 123
+
+
 def test_is_pr_build_not_pr(mocker, mock_codebuild):
     _mock_build_details('master')
     build_obj = build.Build(_mock_build_event())