Add test for file URL transformer

Mantle · Mar 8, 2014 · 29fe1b8 · 29fe1b8
1 parent 4c084eb
commit 29fe1b8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Mantle/NSValueTransformer+MTLPredefinedTransformerAdditions.m b/Mantle/NSValueTransformer+MTLPredefinedTransformerAdditions.m
@@ -23,7 +23,7 @@ + (void)load {
 		MTLValueTransformer *URLValueTransformer = [MTLValueTransformer
 			reversibleTransformerWithForwardBlock:^ id (NSString *str) {
 				if (![str isKindOfClass:NSString.class]) return nil;
-				return ([NSURL URLWithString:str] ?: [NSURL fileURLWithPath:str]);
+				return [NSURL URLWithString:str] ?: [NSURL fileURLWithPath:str];
 			}
 			reverseBlock:^ id (NSURL *URL) {
 				if (![URL isKindOfClass:NSURL.class]) return nil;

diff --git a/MantleTests/MTLPredefinedTransformerAdditionsSpec.m b/MantleTests/MTLPredefinedTransformerAdditionsSpec.m
@@ -25,6 +25,11 @@
 	expect([transformer transformedValue:URLString]).to.equal([NSURL URLWithString:URLString]);
 	expect([transformer reverseTransformedValue:[NSURL URLWithString:URLString]]).to.equal(URLString);
 
+	// Write temporary file for use in testing file URLs
+    NSURL *fileURL = [NSURL fileURLWithPath:[NSTemporaryDirectory() stringByAppendingPathComponent:@"file.txt"]];
+	expect([transformer transformedValue:fileURL.absoluteString]).to.equal(fileURL);
+	expect([transformer reverseTransformedValue:fileURL]).to.equal(fileURL.absoluteString);
+
 	expect([transformer transformedValue:nil]).to.beNil();
 	expect([transformer reverseTransformedValue:nil]).to.beNil();
 });