jvmTest.plot.builder.interact.MappedDataAccessMock.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of plot-builder-portable Show documentation

The Let-Plot Kotlin API depends on this artifact.

The newest version!

/*
 * Copyright (c) 2019. JetBrains s.r.o.
 * Use of this source code is governed by the MIT license that can be found in the LICENSE file.
 */

package jetbrains.datalore.plot.builder.interact

import jetbrains.datalore.plot.base.Aes
import jetbrains.datalore.plot.base.interact.MappedDataAccess
import jetbrains.datalore.plot.builder.interact.mockito.eq
import org.mockito.Mockito.mock
import org.mockito.Mockito.`when`

class MappedDataAccessMock {

    private val mappedAes = HashSet>()
    val mappedDataAccess: MappedDataAccess = mock(MappedDataAccess::class.java)

    fun  add(mapping: Mapping): MappedDataAccessMock {
        return add(mapping, null)
    }

    fun  add(mapping: Mapping, index: Int?): MappedDataAccessMock {
        val aes = mapping.aes

//        if (index == null) {
//            `when`(mappedDataAccess.getMappedDataValue(eq(aes), anyInt(), eq(TestingPlotContext.DUMMY)))
//                .thenReturn(mapping.value)
//        } else {
//            `when`(mappedDataAccess.getMappedDataValue(eq(aes), eq(index), eq(TestingPlotContext.DUMMY)))
//                .thenReturn(mapping.value)
//        }

        `when`(mappedDataAccess.isMapped(eq(aes)))
            .thenReturn(true)
        `when`(mappedDataAccess.getMappedDataLabel(eq(aes)))
            .thenReturn(mapping.label)

        getMappedAes().add(aes)

        return this
    }

    fun getMappedAes(): MutableSet> {
        return mappedAes
    }

    class Mapping internal constructor(
        internal val aes: Aes,
        internal val label: String,
        internal val value: String
    ) {
        fun longTooltipText(): String {
            return "$label: $value"
        }

        fun shortTooltipText(): String {
            return value
        }
    }

    class Variable {
        private var name = ""
        private var value = ""
        private var isContinuous: Boolean = false

        fun name(v: String): Variable {
            this.name = v
            return this
        }

        fun value(v: String): Variable {
            this.value = v
            return this
        }

        fun isContinuous(v: Boolean): Variable {
            this.isContinuous = v
            return this
        }

        fun  mapping(aes: Aes): Mapping {
            return Mapping(aes, name, value)
        }

    }

    companion object {

        fun variable(): Variable {
            return Variable()
        }
    }
}