test fix

buildasaurs · Apr 12, 2015 · f0cb908 · f0cb908
1 parent ff67237
commit f0cb908
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/BuildaGitServerTests/GitHubServerTests.swift b/BuildaGitServerTests/GitHubServerTests.swift
@@ -56,9 +56,13 @@ class GitHubSourceTests: XCTestCase {
         self.tryEndpoint(.GET, endpoint: .PullRequests, params: params) { (body, error) -> () in
 
             XCTAssertNotNil(body, "Body must be non-nil")
-            let prs: [PullRequest] = GitHubArray(body as! NSArray)
-            XCTAssertGreaterThan(prs.count, 0, "We need > 0 items to test parsing")
-            Log.verbose("Parsed PRs: \(prs)")
+            if let body = body as? NSArray {
+                let prs: [PullRequest] = GitHubArray(body as! NSArray)
+                XCTAssertGreaterThan(prs.count, 0, "We need > 0 items to test parsing")
+                Log.verbose("Parsed PRs: \(prs)")
+            } else {
+                XCTFail("Body nil")
+            }
         }
     }
 
@@ -71,9 +75,13 @@ class GitHubSourceTests: XCTestCase {
         self.tryEndpoint(.GET, endpoint: .Branches, params: params) { (body, error) -> () in
 
             XCTAssertNotNil(body, "Body must be non-nil")
-            let branches: [Branch] = GitHubArray(body as! NSArray)
-            XCTAssertGreaterThan(branches.count, 0, "We need > 0 items to test parsing")
-            Log.verbose("Parsed branches: \(branches)")
+            if let body = body as? NSArray {
+                let branches: [Branch] = GitHubArray(body)
+                XCTAssertGreaterThan(branches.count, 0, "We need > 0 items to test parsing")
+                Log.verbose("Parsed branches: \(branches)")
+            } else {
+                XCTFail("Body nil")
+            }
         }
     }