From 7cc9710b3b7de84b3db76c410ce2017f1a6552d4 Mon Sep 17 00:00:00 2001
From: Tim Shadel <tim@shadelsoftware.com>
Date: Thu, 11 Aug 2016 14:05:22 -0600
Subject: [PATCH] Explicitly check failure points during testing

This also catches and fixes the test failure on master involving a file
that isn't being created because the directory didn't exist.
---
 DVR/Session.swift                  |  2 +-
 DVR/Tests/SessionUploadTests.swift | 25 ++++++++++++++++++++-----
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/DVR/Session.swift b/DVR/Session.swift
index b63764b..5ec8029 100644
--- a/DVR/Session.swift
+++ b/DVR/Session.swift
@@ -66,7 +66,7 @@ public class Session: NSURLSession {
     }
 
     public override func uploadTaskWithRequest(request: NSURLRequest, fromFile fileURL: NSURL, completionHandler: (NSData?, NSURLResponse?, NSError?) -> Void) -> NSURLSessionUploadTask {
-        let data = NSData(contentsOfURL: fileURL)!
+        let data = NSData(contentsOfURL: fileURL)
         return addUploadTask(request, fromData: data, completionHandler: completionHandler)
     }
 
diff --git a/DVR/Tests/SessionUploadTests.swift b/DVR/Tests/SessionUploadTests.swift
index f3009e8..155a2b9 100644
--- a/DVR/Tests/SessionUploadTests.swift
+++ b/DVR/Tests/SessionUploadTests.swift
@@ -12,8 +12,8 @@ class SessionUploadTests: XCTestCase {
         return request
     }()
     let multipartBoundary = "---------------------------3klfenalksjflkjoi9auf89eshajsnl3kjnwal".UTF8Data()
-    lazy var testFile: NSURL = {
-        return NSBundle(forClass: self.dynamicType).URLForResource("testfile", withExtension: "txt")!
+    lazy var testFile: NSURL? = {
+        return NSBundle(forClass: self.dynamicType).URLForResource("testfile", withExtension: "txt")
     }()
 
     func testUploadFile() {
@@ -21,18 +21,26 @@ class SessionUploadTests: XCTestCase {
         session.recordingEnabled = false
         let expectation = expectationWithDescription("Network")
 
-        let data = encodeMultipartBody(NSData(contentsOfURL: testFile)!, parameters: [:])
+        guard let testFile = testFile else { XCTFail("Missing test file URL"); return }
+        guard let fileData = NSData(contentsOfURL: testFile) else { XCTFail("Missing body data"); return }
+        let data = encodeMultipartBody(fileData, parameters: [:])
         let file = writeDataToFile(data, fileName: "upload-file")
 
         session.uploadTaskWithRequest(request, fromFile: file) { data, response, error in
+            if let error = error {
+                XCTFail("Error uploading file: \(error)")
+                return
+            }
+            guard let data = data else { XCTFail("Missing request data"); return }
+
             do {
-                let JSON = try NSJSONSerialization.JSONObjectWithData(data!, options: []) as? [String: AnyObject]
+                let JSON = try NSJSONSerialization.JSONObjectWithData(data, options: []) as? [String: AnyObject]
                 XCTAssertEqual("test file\n", (JSON?["form"] as? [String: AnyObject])?["file"] as? String)
             } catch {
                 XCTFail("Failed to read JSON.")
             }
 
-            let HTTPResponse = response as! NSHTTPURLResponse
+            guard let HTTPResponse = response as? NSHTTPURLResponse else { XCTFail("Bad HTTP response"); return }
             XCTAssertEqual(200, HTTPResponse.statusCode)
 
             expectation.fulfill()
@@ -46,6 +54,7 @@ class SessionUploadTests: XCTestCase {
         session.recordingEnabled = false
         let expectation = expectationWithDescription("Network")
 
+        guard let testFile = testFile else { XCTFail("Missing testfile URL"); return }
         let data = encodeMultipartBody(NSData(contentsOfURL: testFile)!, parameters: [:])
 
         session.uploadTaskWithRequest(request, fromData: data) { data, response, error in
@@ -87,6 +96,12 @@ class SessionUploadTests: XCTestCase {
     func writeDataToFile(data: NSData, fileName: String) -> NSURL {
         let documentsPath = NSSearchPathForDirectoriesInDomains(.DocumentDirectory, .UserDomainMask, true)[0]
         let documentsURL = NSURL(fileURLWithPath: documentsPath, isDirectory: true)
+        
+        do {
+            try NSFileManager.defaultManager().createDirectoryAtURL(documentsURL, withIntermediateDirectories: true, attributes: nil)
+        } catch {
+            XCTFail("Failed to create documents directory \(documentsURL). Error \(error)")
+        }
 
         let url = documentsURL.URLByAppendingPathComponent(fileName + ".tmp")