chore: add TransactionModel and function for value generation

arieldossantos · Nov 14, 2020 · 596fad3 · 596fad3
1 parent e9ebaa5
commit 596fad3
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/src/main/kotlin/io/github/arieldossantos/cuddlytrain/models/TransactionModel.kt b/src/main/kotlin/io/github/arieldossantos/cuddlytrain/models/TransactionModel.kt
@@ -0,0 +1,12 @@
+package io.github.arieldossantos.cuddlytrain.models
+
+import java.sql.Timestamp
+
+/**
+ * Transaction model
+ *
+ * @param descricao Descrição da transação
+ * @param data Data da transação
+ * @param valor Valor da transação
+ */
+class TransactionModel(val descricao: String, val data: Timestamp, val valor: Int)
diff --git a/src/main/kotlin/io/github/arieldossantos/cuddlytrain/util/Generator.kt b/src/main/kotlin/io/github/arieldossantos/cuddlytrain/util/Generator.kt
@@ -14,4 +14,15 @@ object Generator {
     fun totalTransactionsInPeriod(id: Int, month: Int): Int {
         return General.getFirstDigit(id) * month
     }
+
+
+    /**
+     * Generate a random transaction value
+     *
+     * @param id user id
+     * @param year current year
+     */
+    fun generateTransactionValue(id: Int, year: Int, transactionIndex: Int): Int {
+        return General.getFirstDigit(id) * General.getFirstDigit(year) * transactionIndex;
+    }
 }
diff --git a/src/test/kotlin/io/github/arieldossantos/cuddlytrain/util/GeneratorTest.kt b/src/test/kotlin/io/github/arieldossantos/cuddlytrain/util/GeneratorTest.kt
@@ -13,4 +13,14 @@ class GeneratorTest {
         Assert.assertSame(Generator.totalTransactionsInPeriod(3423, 3), 9)
         Assert.assertSame(Generator.totalTransactionsInPeriod(723, 12), 84)
     }
+
+    /**
+     * Test generation of transaction value
+     */
+    @Test
+    fun testGenerateTransactionValue() {
+        Assert.assertSame(Generator.generateTransactionValue(12243, 2020, 1), 2)
+        Assert.assertSame(Generator.generateTransactionValue(3423, 2019, 12), 72)
+        Assert.assertSame(Generator.generateTransactionValue(723, 1997, 5), 35)
+    }
 }