AndroidAPS/app/src/test/java/info/nightscout/androidaps/db/BgReadingTest.kt

package info.nightscout.androidaps.db

import dagger.android.AndroidInjector
import dagger.android.HasAndroidInjector
import info.AAPSMocker
import info.TestBase
import info.nightscout.androidaps.Constants
import info.nightscout.androidaps.MainApp
import info.nightscout.androidaps.logging.AAPSLogger
import info.nightscout.androidaps.logging.L
import info.nightscout.androidaps.plugins.configBuilder.ProfileFunction
import info.nightscout.androidaps.plugins.iob.iobCobCalculator.GlucoseStatus
import info.nightscout.androidaps.utils.DefaultValueHelper
import info.nightscout.androidaps.utils.SP
import info.nightscout.androidaps.utils.resources.ResourceHelper
import org.junit.Assert
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.ArgumentMatchers
import org.mockito.Mock
import org.mockito.Mockito
import org.mockito.Mockito.`when`
import org.powermock.core.classloader.annotations.PrepareForTest
import org.powermock.modules.junit4.PowerMockRunner
import java.util.*
import java.util.logging.Logger

@RunWith(PowerMockRunner::class)
@PrepareForTest(MainApp::class, Logger::class, L::class, SP::class, GlucoseStatus::class)
class BgReadingTest : TestBase() {

    @Mock lateinit var aapsLogger: AAPSLogger
    @Mock lateinit var defaultValueHelper: DefaultValueHelper
    @Mock lateinit var profileFunction: ProfileFunction
    @Mock lateinit var resourceHelper: ResourceHelper

    var injector: HasAndroidInjector = HasAndroidInjector {
        AndroidInjector {
            if (it is BgReading) {
                it.aapsLogger = aapsLogger
                it.resourceHelper = resourceHelper
                it.defaultValueHelper = defaultValueHelper
                it.profileFunction = profileFunction
            }
        }
    }

    @Test
    fun valueToUnits() {
        val bgReading = BgReading()
        bgReading.value = 18.0
        Assert.assertEquals(18.0, bgReading.valueToUnits(Constants.MGDL) * 1, 0.01)
        Assert.assertEquals(1.0, bgReading.valueToUnits(Constants.MMOL) * 1, 0.01)
    }

    @Test
    fun directionToSymbol() {
        val bgReading = BgReading()
        bgReading.direction = "DoubleDown"
        Assert.assertEquals("\u21ca", bgReading.directionToSymbol())
        bgReading.direction = "SingleDown"
        Assert.assertEquals("\u2193", bgReading.directionToSymbol())
        bgReading.direction = "FortyFiveDown"
        Assert.assertEquals("\u2198", bgReading.directionToSymbol())
        bgReading.direction = "Flat"
        Assert.assertEquals("\u2192", bgReading.directionToSymbol())
        bgReading.direction = "FortyFiveUp"
        Assert.assertEquals("\u2197", bgReading.directionToSymbol())
        bgReading.direction = "SingleUp"
        Assert.assertEquals("\u2191", bgReading.directionToSymbol())
        bgReading.direction = "DoubleUp"
        Assert.assertEquals("\u21c8", bgReading.directionToSymbol())
        bgReading.direction = "OUT OF RANGE"
        Assert.assertEquals("??", bgReading.directionToSymbol())
    }

    @Test fun dateTest() {
        val bgReading = BgReading()
        val now = System.currentTimeMillis()
        bgReading.date = now
        val nowDate = Date(now)
        Assert.assertEquals(now, bgReading.date(now).date)
        Assert.assertEquals(now, bgReading.date(nowDate).date)
    }

    @Test fun valueTest() {
        val bgReading = BgReading()
        val valueToSet = 81.0 // 4.5 mmol
        Assert.assertEquals(81.0, bgReading.value(valueToSet).value, 0.01)
    }

    @Test fun copyFromTest() {
        val bgReading = BgReading()
        val copy = BgReading()
        bgReading.value = 81.0
        val now = System.currentTimeMillis()
        bgReading.date = now
        copy.date = now
        copy.copyFrom(bgReading)
        Assert.assertEquals(81.0, copy.value, 0.1)
        Assert.assertEquals(now, copy.date)
        Assert.assertEquals(bgReading.directionToSymbol(), copy.directionToSymbol())
    }

    @Test
    fun isEqualTest() {
        val bgReading = BgReading()
        val copy = BgReading()
        bgReading.value = 81.0
        val now = System.currentTimeMillis()
        bgReading.date = now
        copy.date = now
        copy.copyFrom(bgReading)
        Assert.assertTrue(copy.isEqual(bgReading))
        Assert.assertFalse(copy.isEqual(BgReading()))
    }

    @Test fun calculateDirection() {
        val bgReading = BgReading()
        val bgReadingsList: List<BgReading>? = null
        AAPSMocker.mockDatabaseHelper()
        `when`(MainApp.getDbHelper().getAllBgreadingsDataFromTime(ArgumentMatchers.anyLong(), ArgumentMatchers.anyBoolean())).thenReturn(bgReadingsList)
        Assert.assertEquals("NONE", bgReading.calculateDirection())
        setReadings(72, 0)
        Assert.assertEquals("DoubleUp", bgReading.calculateDirection())
        setReadings(76, 60)
        Assert.assertEquals("SingleUp", bgReading.calculateDirection())
        setReadings(74, 65)
        Assert.assertEquals("FortyFiveUp", bgReading.calculateDirection())
        setReadings(72, 72)
        Assert.assertEquals("Flat", bgReading.calculateDirection())
        setReadings(0, 72)
        Assert.assertEquals("DoubleDown", bgReading.calculateDirection())
        setReadings(60, 76)
        Assert.assertEquals("SingleDown", bgReading.calculateDirection())
        setReadings(65, 74)
        Assert.assertEquals("FortyFiveDown", bgReading.calculateDirection())
    }

    @Before
    fun prepareMock() {
        val mainApp = AAPSMocker.mockMainApp()
        AAPSMocker.mockApplicationContext()
        AAPSMocker.mockSP()
        AAPSMocker.mockL()
        AAPSMocker.mockDatabaseHelper()
        `when`(mainApp.androidInjector()).thenReturn(injector.androidInjector())
    }

    fun setReadings(current_value: Int, previous_value: Int) {
        val now = BgReading()
        now.value = current_value.toDouble()
        now.date = System.currentTimeMillis()
        val previous = BgReading()
        previous.value = previous_value.toDouble()
        previous.date = System.currentTimeMillis() - 6 * 60 * 1000L
        val bgReadings: MutableList<BgReading> = mutableListOf()
        bgReadings.add(now)
        bgReadings.add(previous)
        `when`(MainApp.getDbHelper().getAllBgreadingsDataFromTime(ArgumentMatchers.anyLong(), ArgumentMatchers.anyBoolean())).thenReturn(bgReadings)
    }
}