Switch helper to the true method

libgit2 · Jun 19, 2017 · 8c09e8b · 8c09e8b
1 parent 558a039
commit 8c09e8b
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/ObjectiveGitTests/GTUtilityFunctions.m b/ObjectiveGitTests/GTUtilityFunctions.m
@@ -40,11 +40,10 @@
 #pragma mark - Branch
 
 BranchBlock localBranchWithName = ^ GTBranch * (NSString *branchName, GTRepository *repo) {
-	NSString *reference = [GTBranch.localNamePrefix stringByAppendingString:branchName];
-	NSArray *branches = [repo branchesWithPrefix:reference error:NULL];
-	expect(branches).notTo(beNil());
-	expect(@(branches.count)).to(equal(@1));
-	expect(((GTBranch *)branches[0]).shortName).to(equal(branchName));
+	BOOL success = NO;
+	GTBranch *branch = [repo lookUpBranchWithName:branchName type:GTBranchTypeLocal success:&success error:NULL];
+	expect(branch).notTo(beNil());
+	expect(branch.shortName).to(equal(branchName));
 
-	return branches[0];
+	return branch;
 };