Fix stock download. #3

slomkowski · Oct 9, 2020 · bd44413 · bd44413
1 parent c858f1b
commit bd44413
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/...toglowd/src/main/kotlin/eu/slomkowski/octoglow/octoglowd/daemon/frontdisplay/StockView.kt b/...toglowd/src/main/kotlin/eu/slomkowski/octoglow/octoglowd/daemon/frontdisplay/StockView.kt
@@ -34,8 +34,8 @@ class StockView(
             val high: BigDecimal,
             val low: BigDecimal,
             val close: BigDecimal,
-            val volume: Int,
-            val openInt: Int) {
+            val volume: Long,
+            val openInt: Long) {
         init {
             require(ticker.isNotBlank())
             require(interval > Duration.ZERO)
@@ -70,8 +70,8 @@ class StockView(
                         record[5].toBigDecimal(),
                         record[6].toBigDecimal(),
                         record[7].toBigDecimal(),
-                        record[8].toInt(),
-                        record[9].toInt())
+                        record[8].toLong(),
+                        record[9].toLong())
             })
         }
     }

diff --git a/...owd/src/test/kotlin/eu/slomkowski/octoglow/octoglowd/daemon/frontdisplay/StockViewTest.kt b/...owd/src/test/kotlin/eu/slomkowski/octoglow/octoglowd/daemon/frontdisplay/StockViewTest.kt
@@ -6,6 +6,7 @@ import org.junit.jupiter.api.Assertions.assertTrue
 import org.junit.jupiter.api.Test
 import java.time.DayOfWeek
 import java.time.LocalDate
+import kotlin.test.assertNotNull
 
 internal class StockViewTest {
 
@@ -27,5 +28,12 @@ internal class StockViewTest {
         val stockData = runBlocking { StockView.downloadStockData(lastWeekDay) }
         assertTrue(stockData.isNotEmpty())
         logger.info("Downloaded {} stocks.", stockData.size)
+
+        val stockDataByTicker = stockData.groupBy { it.ticker }
+        logger.info("Found {} tickers.", stockDataByTicker.size)
+
+        assertNotNull(stockDataByTicker["CDR"])
+        assertNotNull(stockDataByTicker["TPE"])
+        assertNotNull(stockDataByTicker["PKO"])
     }
 }