Add make function for vehicle

chadobado · May 9, 2019 · 4219618 · 4219618
1 parent 376f52d
commit 4219618
Show file tree

Hide file tree

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