package io.delphiplatform.api.v3.model.mapper

import com.sonymusic.delphi.etl.apps.proto.Apple
import com.sonymusic.delphi.etl.apps.proto.Spotify
import com.sonymusic.delphi.etl.apps.proto.Youtube
import fr.xebia.extras.selma.Selma
import org.junit.jupiter.api.Assertions
import org.junit.jupiter.api.Test

internal class AgeBandsMapperTest {
    private val mapper = Selma.builder(AgeBandsMapper::class.java).build()

    @Test
    fun test_toStreamsAgeBands_appleAgeBands() {
        val inputObject = Apple.AppleMusicAggregation.AppleAgeBands.newBuilder()
            .setAll017(10)
            .setMale017(11)
            .setFemale017(12)
            .setUnknown017(13)
            .setAll1824(14)
            .setAll2534(15)
            .setAll3544(16)
            .setAll4554(17)
            .setAll5564(18)
            .setAll65Plus(19)
            .setAllUnknown(20)
            .setMale1824(21)
            .setMale2534(22)
            .setMale3544(23)
            .setMale4554(24)
            .setMale5564(25)
            .setMale65Plus(26)
            .setMaleUnknown(27)
            .setFemale1824(28)
            .setFemale2534(29)
            .setFemale3544(30)
            .setFemale4554(31)
            .setFemale5564(32)
            .setFemale65Plus(33)
            .setFemaleUnknown(34)
            .setUnknown1824(35)
            .setUnknown2534(36)
            .setUnknown3544(37)
            .setUnknown4554(38)
            .setUnknown5564(39)
            .setUnknown65Plus(40)
            .setUnknownUnknown(41)
            .build()
        val result = mapper.toStreamsAgeBands(inputObject)
        Assertions.assertEquals(inputObject.all017, result.all017)
        Assertions.assertEquals(inputObject.male017, result.male017)
        Assertions.assertEquals(inputObject.female017, result.female017)
        Assertions.assertEquals(inputObject.unknown017, result.unknown017)
        Assertions.assertEquals(inputObject.all1824, result.all1824)
        Assertions.assertEquals(inputObject.all2534, result.all2534)
        Assertions.assertEquals(inputObject.all3544, result.all3544)
        Assertions.assertEquals(inputObject.all4554, result.all4554)
        Assertions.assertEquals(inputObject.all5564, result.all5564)
        Assertions.assertEquals(inputObject.all65Plus, result.all65Plus)
        Assertions.assertEquals(inputObject.allUnknown, result.allUnknown)
        Assertions.assertEquals(inputObject.male1824, result.male1824)
        Assertions.assertEquals(inputObject.male2534, result.male2534)
        Assertions.assertEquals(inputObject.male3544, result.male3544)
        Assertions.assertEquals(inputObject.male4554, result.male4554)
        Assertions.assertEquals(inputObject.male5564, result.male5564)
        Assertions.assertEquals(inputObject.male65Plus, result.male65Plus)
        Assertions.assertEquals(inputObject.maleUnknown, result.maleUnknown)
        Assertions.assertEquals(inputObject.female1824, result.female1824)
        Assertions.assertEquals(inputObject.female2534, result.female2534)
        Assertions.assertEquals(inputObject.female3544, result.female3544)
        Assertions.assertEquals(inputObject.female4554, result.female4554)
        Assertions.assertEquals(inputObject.female5564, result.female5564)
        Assertions.assertEquals(inputObject.female65Plus, result.female65Plus)
        Assertions.assertEquals(inputObject.femaleUnknown, result.femaleUnknown)
        Assertions.assertEquals(inputObject.unknown1824, result.unknown1824)
        Assertions.assertEquals(inputObject.unknown2534, result.unknown2534)
        Assertions.assertEquals(inputObject.unknown3544, result.unknown3544)
        Assertions.assertEquals(inputObject.unknown4554, result.unknown4554)
        Assertions.assertEquals(inputObject.unknown5564, result.unknown5564)
        Assertions.assertEquals(inputObject.unknown65Plus, result.unknown65Plus)
        Assertions.assertEquals(inputObject.unknownUnknown, result.unknownUnknown)
    }

    @Test
    fun test_toStreamsAgeBands_spotifyAgeBands() {
        val inputObject = Spotify.SpotifyAgeBand.newBuilder()
            .setAll017(10L)
            .setAll1822(11L)
            .setAll2327(12L)
            .setAll2834(13L)
            .setAll3544(14L)
            .setAll4559(15L)
            .setAll60150(16L)
            .setAllUnknown(17L)
            .setFemale017(18L)
            .setFemale1822(19L)
            .setFemale2327(20L)
            .setFemale2834(21L)
            .setFemale3544(22L)
            .setFemale4559(23L)
            .setFemale60150(24L)
            .setFemaleUnknown(25L)
            .setMale017(26L)
            .setMale1822(27L)
            .setMale2327(28L)
            .setMale2834(29L)
            .setMale3544(30L)
            .setMale4559(31L)
            .setMale60150(32L)
            .setMaleUnknown(33L)
            .setNeutral017(34L)
            .setNeutral1822(35L)
            .setNeutral2327(36L)
            .setNeutral2834(37L)
            .setNeutral3544(38L)
            .setNeutral4559(39L)
            .setNeutral60150(40L)
            .setNeutralUnknown(41L)
            .setUnknown017(42L)
            .setUnknown1822(43L)
            .setUnknown2327(44L)
            .setUnknown2834(45L)
            .setUnknown3544(46L)
            .setUnknown4559(47L)
            .setUnknown60150(48L)
            .setUnknownUnknown(49L)
            .build()
        val result = mapper.toStreamsAgeBands(
            inputObject
        )
        Assertions.assertEquals(inputObject.all017, result.all017)
        Assertions.assertEquals(inputObject.all1822, result.all1822)
        Assertions.assertEquals(inputObject.all2327, result.all2327)
        Assertions.assertEquals(inputObject.all2834, result.all2834)
        Assertions.assertEquals(inputObject.all3544, result.all3544)
        Assertions.assertEquals(inputObject.all4559, result.all4559)
        Assertions.assertEquals(inputObject.all60150, result.all60150)
        Assertions.assertEquals(inputObject.allUnknown, result.allUnknown)
        Assertions.assertEquals(inputObject.female017, result.female017)
        Assertions.assertEquals(inputObject.female1822, result.female1822)
        Assertions.assertEquals(inputObject.female2327, result.female2327)
        Assertions.assertEquals(inputObject.female2834, result.female2834)
        Assertions.assertEquals(inputObject.female3544, result.female3544)
        Assertions.assertEquals(inputObject.female4559, result.female4559)
        Assertions.assertEquals(inputObject.female60150, result.female60150)
        Assertions.assertEquals(inputObject.femaleUnknown, result.femaleUnknown)
        Assertions.assertEquals(inputObject.male017, result.male017)
        Assertions.assertEquals(inputObject.male1822, result.male1822)
        Assertions.assertEquals(inputObject.male2327, result.male2327)
        Assertions.assertEquals(inputObject.male2834, result.male2834)
        Assertions.assertEquals(inputObject.male3544, result.male3544)
        Assertions.assertEquals(inputObject.male4559, result.male4559)
        Assertions.assertEquals(inputObject.male60150, result.male60150)
        Assertions.assertEquals(inputObject.maleUnknown, result.maleUnknown)
        Assertions.assertEquals(inputObject.neutral017, result.neutral017)
        Assertions.assertEquals(inputObject.neutral1822, result.neutral1822)
        Assertions.assertEquals(inputObject.neutral2327, result.neutral2327)
        Assertions.assertEquals(inputObject.neutral2834, result.neutral2834)
        Assertions.assertEquals(inputObject.neutral3544, result.neutral3544)
        Assertions.assertEquals(inputObject.neutral4559, result.neutral4559)
        Assertions.assertEquals(inputObject.neutral60150, result.neutral60150)
        Assertions.assertEquals(inputObject.neutralUnknown, result.neutralUnknown)
        Assertions.assertEquals(inputObject.unknown017, result.unknown017)
        Assertions.assertEquals(inputObject.unknown1822, result.unknown1822)
        Assertions.assertEquals(inputObject.unknown2327, result.unknown2327)
        Assertions.assertEquals(inputObject.unknown2834, result.unknown2834)
        Assertions.assertEquals(inputObject.unknown3544, result.unknown3544)
        Assertions.assertEquals(inputObject.unknown4559, result.unknown4559)
        Assertions.assertEquals(inputObject.unknown60150, result.unknown60150)
        Assertions.assertEquals(inputObject.unknownUnknown, result.unknownUnknown)
    }

    @Test
    fun test_toStreamsAgeBands_youtubeAgeBands() {
        val inputObject = Youtube.YoutubeAgeBands.newBuilder()
            .setAll1317(10f)
            .setAll1824(11f)
            .setAll2534(12f)
            .setAll3544(13f)
            .setAll4554(14f)
            .setAll5564(15f)
            .setAll65Plus(16f)
            .setFemale1317(17f)
            .setFemale1824(18f)
            .setFemale2534(19f)
            .setFemale3544(20f)
            .setFemale4554(21f)
            .setFemale5564(22f)
            .setFemale65Plus(23f)
            .setMale1317(24f)
            .setMale1824(25f)
            .setMale2534(26f)
            .setMale3544(27f)
            .setMale4554(28f)
            .setMale5564(29f)
            .setMale65Plus(30f)
            .setOther1317(31f)
            .setOther1824(32f)
            .setOther2534(33f)
            .setOther3544(34f)
            .setOther4554(35f)
            .setOther5564(36f)
            .setOther65Plus(37f)
            .build()
        val result = mapper.toAgeBands(inputObject)
        Assertions.assertEquals(inputObject.all1317, result.all1317.get())
        Assertions.assertEquals(inputObject.all1824, result.all1824.get())
        Assertions.assertEquals(inputObject.all2534, result.all2534.get())
        Assertions.assertEquals(inputObject.all3544, result.all3544.get())
        Assertions.assertEquals(inputObject.all4554, result.all4554.get())
        Assertions.assertEquals(inputObject.all5564, result.all5564.get())
        Assertions.assertEquals(inputObject.all65Plus, result.all65Plus.get())
        Assertions.assertEquals(inputObject.female1317, result.female1317.get())
        Assertions.assertEquals(inputObject.female1824, result.female1824.get())
        Assertions.assertEquals(inputObject.female2534, result.female2534.get())
        Assertions.assertEquals(inputObject.female3544, result.female3544.get())
        Assertions.assertEquals(inputObject.female4554, result.female4554.get())
        Assertions.assertEquals(inputObject.female5564, result.female5564.get())
        Assertions.assertEquals(inputObject.female65Plus, result.female65Plus.get())
        Assertions.assertEquals(inputObject.male1317, result.male1317.get())
        Assertions.assertEquals(inputObject.male1824, result.male1824.get())
        Assertions.assertEquals(inputObject.male2534, result.male2534.get())
        Assertions.assertEquals(inputObject.male3544, result.male3544.get())
        Assertions.assertEquals(inputObject.male4554, result.male4554.get())
        Assertions.assertEquals(inputObject.male5564, result.male5564.get())
        Assertions.assertEquals(inputObject.male65Plus, result.male65Plus.get())
        Assertions.assertEquals(inputObject.other1317, result.unknown1317.get())
        Assertions.assertEquals(inputObject.other1824, result.unknown1824.get())
        Assertions.assertEquals(inputObject.other2534, result.unknown2534.get())
        Assertions.assertEquals(inputObject.other3544, result.unknown3544.get())
        Assertions.assertEquals(inputObject.other4554, result.unknown4554.get())
        Assertions.assertEquals(inputObject.other5564, result.unknown5564.get())
        Assertions.assertEquals(inputObject.other65Plus, result.unknown65Plus.get())
    }
}