Add colors function for vehicle

chadobado · May 9, 2019 · 6caa058 · 6caa058
1 parent 4219618
commit 6caa058
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/Sources/Fakery/Generators/Vehicle.swift b/Sources/Fakery/Generators/Vehicle.swift
@@ -2,8 +2,12 @@ public final class Vehicle: Generator {
   public func manufacture() -> String {
     return generate("vehicle.manufacture")
   }
-  
+
   public func make() -> String {
     return generate("vehicle.makes")
   }
+
+  public func colors() -> String {
+    return generate("vehicle.colors")
+  }
 }
diff --git a/Tests/Fakery/Generators/VehicleSpec.swift b/Tests/Fakery/Generators/VehicleSpec.swift
@@ -25,6 +25,13 @@ final class VehicleSpec: QuickSpec {
           expect(make).to(equal("BMW"))
         }
       }
+
+      describe("#colors") {
+        it("returns the correct color") {
+          let color = vehicle.colors()
+          expect(color).to(equal("Red"))
+        }
+      }
     }
   }
 }