Add CVV to credit card

Devskiller · Mar 16, 2018 · 8ecdff7 · 8ecdff7
1 parent 8688163
commit 8ecdff7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/main/java/com/devskiller/jfairy/producer/payment/CreditCard.java b/src/main/java/com/devskiller/jfairy/producer/payment/CreditCard.java
@@ -11,11 +11,13 @@ public class CreditCard {
 
 	private final String cardVendor;
 	private final String cardNumber;
+	private final String cvv;
 	private final LocalDateTime expiryDate;
 
-	public CreditCard(String cardVendor, String cardNumber, LocalDateTime expiryDate) {
+	public CreditCard(String cardVendor, String cardNumber, String cvv, LocalDateTime expiryDate) {
 		this.cardVendor = cardVendor;
 		this.cardNumber = cardNumber;
+		this.cvv = cvv;
 		this.expiryDate = expiryDate;
 	}
 
@@ -27,6 +29,10 @@ public String getCardNumber() {
 		return cardNumber;
 	}
 
+	public String getCvv() {
+		return cvv;
+	}
+
 	public LocalDateTime getExpiryDate() {
 		return expiryDate;
 	}

diff --git a/src/main/java/com/devskiller/jfairy/producer/payment/CreditCardProvider.java b/src/main/java/com/devskiller/jfairy/producer/payment/CreditCardProvider.java
@@ -32,7 +32,7 @@ public CreditCard get() {
 		String randomNumber = generateNumber();
 
 		LocalDateTime expiryDate = dateProducer.randomDateBetweenNowAndFuturePeriod(DEFAULT_VALIDITY);
-		return new CreditCard(CARD_VENDOR, randomNumber, expiryDate);
+		return new CreditCard(CARD_VENDOR, randomNumber, baseProducer.numerify("###"), expiryDate);
 	}
 
 	private String generateNumber() {