Merge pull request #227 from octokit/set-nested-repo-server

Set PR's nested repo server
octokit · Jan 7, 2015 · 209b090 · 209b090
2 parents f9f2155 + 086aace
commit 209b090
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/OctoKit/OCTPullRequest.m b/OctoKit/OCTPullRequest.m
@@ -10,6 +10,7 @@
 #import "OCTUser.h"
 #import "OCTRepository.h"
 #import "NSValueTransformer+OCTPredefinedTransformerAdditions.h"
+#import "OCTObject+Private.h"
 
 @implementation OCTPullRequest
 
@@ -101,4 +102,11 @@ + (NSValueTransformer *)baseRepositoryJSONTransformer {
 	return [MTLValueTransformer mtl_JSONDictionaryTransformerWithModelClass:OCTRepository.class];
 }
 
+- (void)setBaseURL:(NSURL *)baseURL {
+	super.baseURL = baseURL;
+
+	self.headRepository.baseURL = baseURL;
+	self.baseRepository.baseURL = baseURL;
+}
+
 @end
diff --git a/OctoKitTests/OCTPullRequestSpec.m b/OctoKitTests/OCTPullRequestSpec.m
@@ -12,6 +12,7 @@
 #import <Quick/Quick.h>
 
 #import "OCTObjectSpec.h"
+#import "OCTObject+Private.h"
 
 QuickSpecBegin(OCTPullRequestSpec)
 
@@ -177,4 +178,11 @@
 	expect(pullRequest.mergedDate).to(equal([[[ISO8601DateFormatter alloc] init] dateFromString:@"2011-01-26T19:04:12Z"]));
 });
 
+it(@"should set its nested repository's servers", ^{
+	NSURL *URL = [NSURL URLWithString:@"https://myserver.com"];
+	pullRequest.baseURL = URL;
+	expect(pullRequest.baseRepository.baseURL).to(equal(URL));
+	expect(pullRequest.headRepository.baseURL).to(equal(URL));
+});
+
 QuickSpecEnd