Update tests to check for base branches which have dashes in them.

python · Jun 16, 2018 · 04e31d5 · 04e31d5
1 parent c5f7faf
commit 04e31d5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/cherry_picker/cherry_picker/test.py b/cherry_picker/cherry_picker/test.py
@@ -33,15 +33,16 @@ def changedir(d):
 
 
 def test_get_base_branch():
+    # The format of cherry-pick branches we create are "backport-{SHA}-{base_branch}"
     cherry_pick_branch = 'backport-afc23f4-2.7'
     result = get_base_branch(cherry_pick_branch)
     assert result == '2.7'
 
 
-def test_get_base_branch_without_dash():
-    cherry_pick_branch ='master'
+def test_get_base_branch_which_has_dashes():
+    cherry_pick_branch ='backport-afc23f4-baseprefix-2.7-basesuffix'
     result = get_base_branch(cherry_pick_branch)
-    assert result == 'master'
+    assert result == 'prefix-2.7-suffix'
 
 
 @mock.patch('subprocess.check_output')