feat: add ooni test descriptors.

composeApp/src/commonMain/kotlin/org/ooni/probe/data/models/Descriptor.kt

@@ -20,30 +20,32 @@ data class Descriptor(
     val expirationDate: LocalDateTime?,
     val netTests: List<NetTest>,
     val longRunningTests: List<NetTest> = emptyList(),
-    val source: Source,
+    val source: InstalledTestDescriptorModel,
     val updateStatus: UpdateStatus,
     val enabled: Boolean = true,
 ) {
-    sealed interface Source {
-        data class Default(val value: DefaultTestDescriptor) : Source
-
-        data class Installed(val value: InstalledTestDescriptorModel) : Source
-    }
-
     val isExpired get() = expirationDate != null && expirationDate < LocalDateTime.now()
 
     val updatable get() = updateStatus is UpdateStatus.UpdateRejected
 
     val key: String
-        get() = when (source) {
-            is Source.Default -> name
-            is Source.Installed -> source.value.id.value.toString()
-        }
+        get() = source.key
 
     val allTests get() = netTests + longRunningTests
 
     val estimatedDuration
         get() = allTests
             .sumOf { it.test.runtime(it.inputs).inWholeSeconds }
             .seconds
+
+    val settingsPrefix: String
+        get() = source.id.value.toString()
+
+    fun isDefaultDescriptor(): Boolean {
+        return source.isDefaultTestDescriptor
+    }
+
+    fun isInstalledNonDefaultDescriptor(): Boolean {
+        return !source.isDefaultTestDescriptor
+    }
 }

composeApp/src/commonMain/kotlin/org/ooni/probe/data/models/InstalledTestDescriptorModel.kt

@@ -12,11 +12,16 @@ import kotlinx.serialization.json.Json
 import ooniprobe.composeapp.generated.resources.Dashboard_Runv2_Overview_Description
 import ooniprobe.composeapp.generated.resources.Dashboard_Runv2_Overview_LastUpdated
 import ooniprobe.composeapp.generated.resources.Res
+import ooniprobe.composeapp.generated.resources.TestResults_NotAvailable
+import ooniprobe.composeapp.generated.resources.performance_datausage
+import ooniprobe.composeapp.generated.resources.small_datausage
 import ooniprobe.composeapp.generated.resources.test_circumvention
 import ooniprobe.composeapp.generated.resources.test_experimental
 import ooniprobe.composeapp.generated.resources.test_instant_messaging
 import ooniprobe.composeapp.generated.resources.test_performance
 import ooniprobe.composeapp.generated.resources.test_websites
+import ooniprobe.composeapp.generated.resources.websites_datausage
+import org.jetbrains.compose.resources.StringResource
 import org.jetbrains.compose.resources.stringResource
 import org.ooni.probe.data.TestDescriptor
 import org.ooni.probe.shared.InstalledDescriptorIcons
@@ -25,6 +30,21 @@ import org.ooni.probe.shared.now
 import org.ooni.probe.shared.stringMonthArrayResource
 import org.ooni.probe.shared.toEpoch
 
+enum class OoniTest(val id: Long, val key: String) {
+    WEBSITES(10470L, "websites"),
+    INSTANT_MESSAGING(10471L, "instant_messaging"),
+    CIRCUMVENTION(10472L, "circumvention"),
+    PERFORMANCE(10473L, "performance"),
+    EXPERIMENTAL(10474L, "experimental"),
+    ;
+
+    companion object {
+        private val map = entries.associateBy(OoniTest::id)
+
+        fun fromId(id: Long) = map[id]
+    }
+}
+
 @Serializable
 data class InstalledTestDescriptorModel(
     val id: Id,
@@ -52,6 +72,10 @@ data class InstalledTestDescriptorModel(
 
     val isExpired get() = expirationDate != null && expirationDate < LocalDateTime.now()
 
+    val isDefaultTestDescriptor get() = id.value in 10470..10474 // TODO(aanorbel): switch to OONI reserved namespace
+
+    val key get() = OoniTest.fromId(id.value)?.key ?: id.value.toString()
+
     fun shouldUpdate(other: InstalledTestDescriptorModel): Boolean {
         return dateUpdated != null && other.dateUpdated != null && other.dateUpdated > dateUpdated
     }
@@ -79,13 +103,24 @@ fun InstalledTestDescriptorModel.toDescriptor(updateStatus: UpdateStatus = Updat
         icon = icon?.let(InstalledDescriptorIcons::getIconFromValue),
         color = color?.hexToColor(),
         animation = icon?.let { determineAnimation(it) } ?: animation?.let(Animation::fromFileName),
-        dataUsage = { null },
+        dataUsage = { if (isDefaultTestDescriptor) stringResource(getDataUsage()) else null },
         expirationDate = expirationDate,
         netTests = netTests.orEmpty(),
-        source = Descriptor.Source.Installed(this),
+        source = this,
         updateStatus = updateStatus,
     )
 
+fun InstalledTestDescriptorModel.getDataUsage(): StringResource {
+    return when (this.key) {
+        OoniTest.WEBSITES.key -> Res.string.websites_datausage
+        OoniTest.INSTANT_MESSAGING.key -> Res.string.small_datausage
+        OoniTest.CIRCUMVENTION.key -> Res.string.small_datausage
+        OoniTest.PERFORMANCE.key -> Res.string.performance_datausage
+        OoniTest.EXPERIMENTAL.key -> Res.string.TestResults_NotAvailable
+        else -> Res.string.TestResults_NotAvailable
+    }
+}
+
 private val iconAnimationMap = mapOf(
     Res.drawable.test_websites to Animation.Websites,
     Res.drawable.test_instant_messaging to Animation.InstantMessaging,

composeApp/src/commonMain/kotlin/org/ooni/probe/data/models/RunSpecification.kt

@@ -17,24 +17,7 @@ sealed interface RunSpecification {
 
     @Serializable
     data class Test(
-        val source: Source,
+        val sourceId: InstalledTestDescriptorModel.Id,
         val netTests: List<NetTest>,
-    ) {
-        @Serializable
-        sealed interface Source {
-            @Serializable
-            data class Default(val name: String) : Source
-
-            @Serializable
-            data class Installed(val id: InstalledTestDescriptorModel.Id) : Source
-
-            companion object {
-                fun fromDescriptor(descriptor: Descriptor) =
-                    when (descriptor.source) {
-                        is Descriptor.Source.Default -> Default(descriptor.name)
-                        is Descriptor.Source.Installed -> Installed(descriptor.source.value.id)
-                    }
-            }
-        }
-    }
+    )
 }

composeApp/src/commonMain/kotlin/org/ooni/probe/data/repositories/PreferenceRepository.kt

@@ -132,8 +132,7 @@ class PreferenceRepository(
     ): Flow<Boolean> {
         val key = getPreferenceKey(
             name = descriptor.name,
-            prefix = (descriptor.source as? Descriptor.Source.Installed)
-                ?.value?.id?.value?.toString(),
+            prefix = descriptor.source.id.value.toString(),
             autoRun = isAutoRun,
         )
         return dataStore.data.map {
@@ -180,8 +179,7 @@ class PreferenceRepository(
         isAutoRun: Boolean,
     ) = getPreferenceKey(
         name = netTest.test.name,
-        prefix = (descriptor.source as? Descriptor.Source.Installed)
-            ?.value?.id?.value?.toString(),
+        prefix = descriptor.settingsPrefix,
         autoRun = isAutoRun,
     )
 

composeApp/src/commonMain/kotlin/org/ooni/probe/di/Dependencies.kt

@@ -53,7 +53,6 @@ import org.ooni.probe.domain.FinishInProgressData
 import org.ooni.probe.domain.GetAutoRunSettings
 import org.ooni.probe.domain.GetAutoRunSpecification
 import org.ooni.probe.domain.GetBootstrapTestDescriptors
-import org.ooni.probe.domain.GetDefaultTestDescriptors
 import org.ooni.probe.domain.GetEnginePreferences
 import org.ooni.probe.domain.GetFirstRun
 import org.ooni.probe.domain.GetProxySettings
@@ -244,7 +243,6 @@ class Dependencies(
     private val getBootstrapTestDescriptors by lazy {
         GetBootstrapTestDescriptors(readAssetFile, json, backgroundContext)
     }
-    private val getDefaultTestDescriptors by lazy { GetDefaultTestDescriptors() }
     private val getProxySettings by lazy { GetProxySettings(preferenceRepository) }
     private val getEnginePreferences by lazy {
         GetEnginePreferences(
@@ -288,7 +286,6 @@ class Dependencies(
     @VisibleForTesting
     val getTestDescriptors by lazy {
         GetTestDescriptors(
-            getDefaultTestDescriptors = getDefaultTestDescriptors::invoke,
             listInstalledTestDescriptors = testDescriptorRepository::list,
             descriptorUpdates = getDescriptorUpdate::observeAvailableUpdatesState,
             getPreferenceValues = preferenceRepository::allSettings,

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/GetAutoRunSpecification.kt

@@ -17,7 +17,7 @@ class GetAutoRunSpecification(
         return RunSpecification.Full(
             tests = descriptors.map { descriptor ->
                 RunSpecification.Test(
-                    source = RunSpecification.Test.Source.fromDescriptor(descriptor),
+                    sourceId = descriptor.source.id,
                     netTests = descriptor.netTests,
                 )
             },

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/GetResults.kt

@@ -34,7 +34,7 @@ fun List<Descriptor>.forResult(result: ResultModel): Descriptor? =
     result.testDescriptorId
         ?.let { descriptorId ->
             firstOrNull {
-                it.source is Descriptor.Source.Installed && it.source.value.id == descriptorId
+                it.source.id == descriptorId
             }
         }
         ?: firstOrNull { it.name == result.testGroupName }

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/GetTestDescriptors.kt

@@ -3,22 +3,18 @@ package org.ooni.probe.domain
 import androidx.compose.runtime.Composable
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.combine
-import kotlinx.coroutines.flow.flowOf
 import kotlinx.coroutines.flow.map
 import ooniprobe.composeapp.generated.resources.Res
 import ooniprobe.composeapp.generated.resources.Settings_TestOptions_LongRunningTest
 import org.jetbrains.compose.resources.stringResource
 import org.ooni.engine.models.WebConnectivityCategory
-import org.ooni.probe.data.models.DefaultTestDescriptor
 import org.ooni.probe.data.models.Descriptor
 import org.ooni.probe.data.models.DescriptorUpdatesStatus
 import org.ooni.probe.data.models.InstalledTestDescriptorModel
 import org.ooni.probe.data.models.SettingsKey
-import org.ooni.probe.data.models.UpdateStatus
 import org.ooni.probe.data.models.toDescriptor
 
 class GetTestDescriptors(
-    private val getDefaultTestDescriptors: () -> List<DefaultTestDescriptor>,
     private val listInstalledTestDescriptors: () -> Flow<List<InstalledTestDescriptorModel>>,
     private val descriptorUpdates: () -> Flow<DescriptorUpdatesStatus>,
     private val getPreferenceValues: (List<SettingsKey>) -> Flow<Map<SettingsKey, Any?>>,
@@ -27,14 +23,12 @@ class GetTestDescriptors(
         return combine(
             listInstalledTestDescriptors(),
             descriptorUpdates(),
-            flowOf(getDefaultTestDescriptors()),
             isWebsitesDescriptorEnabled(),
-        ) { installedDescriptors, descriptorUpdates, defaultDescriptors, isWebsitesEnabled ->
+        ) { installedDescriptors, descriptorUpdates, isWebsitesEnabled ->
             val updatedDescriptors = installedDescriptors.map { item ->
                 item.toDescriptor(updateStatus = descriptorUpdates.getStatusOf(item.id))
             }
-            val allDescriptors = defaultDescriptors.map { it.toDescriptor() } + updatedDescriptors
-            return@combine allDescriptors.map {
+            return@combine updatedDescriptors.map {
                 it.copy(enabled = it.name != "websites" || isWebsitesEnabled)
             }
         }
@@ -44,29 +38,6 @@ class GetTestDescriptors(
         getPreferenceValues(WebConnectivityCategory.entries.mapNotNull { it.settingsKey })
             .map { preferences -> preferences.any { it.value == true } }
 
-    private fun DefaultTestDescriptor.toDescriptor() =
-        Descriptor(
-            name = label,
-            title = { stringResource(title) },
-            shortDescription = { stringResource(shortDescription) },
-            description = {
-                if (label == "experimental") {
-                    stringResource(description, experimentalLinks())
-                } else {
-                    stringResource(description)
-                }
-            },
-            icon = icon,
-            color = color,
-            animation = animation,
-            dataUsage = { stringResource(dataUsage) },
-            expirationDate = null,
-            netTests = netTests,
-            longRunningTests = longRunningTests,
-            source = Descriptor.Source.Default(this),
-            updateStatus = UpdateStatus.NotApplicable,
-        )
-
     @Composable
     private fun experimentalLinks() =
         """

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/GetTestDescriptorsBySpec.kt

@@ -32,16 +32,6 @@ class GetTestDescriptorsBySpec(
     // Is this descriptor contained in the RunSpecification's list of tests
     private fun RunSpecification.Full.forDescriptor(descriptor: Descriptor) =
         tests.firstOrNull { specTest ->
-            when (descriptor.source) {
-                is Descriptor.Source.Default -> {
-                    specTest.source is RunSpecification.Test.Source.Default &&
-                        specTest.source.name == descriptor.name
-                }
-
-                is Descriptor.Source.Installed -> {
-                    specTest.source is RunSpecification.Test.Source.Installed &&
-                        specTest.source.id == descriptor.source.value.id
-                }
-            }
+            specTest.sourceId == descriptor.source.id
         }
 }

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/RunDescriptors.kt

@@ -129,7 +129,7 @@ class RunDescriptors(
     ) {
         val result = ResultModel(
             testGroupName = descriptor.name,
-            testDescriptorId = (descriptor.source as? Descriptor.Source.Installed)?.value?.id,
+            testDescriptorId = descriptor.source.id,
             taskOrigin = taskOrigin,
         )
         val resultId = storeResult(result)

composeApp/src/commonMain/kotlin/org/ooni/probe/domain/RunNetTest.kt

@@ -60,8 +60,7 @@ class RunNetTest(
                 testTotal = spec.testTotal,
             )
         }
-        val installedDescriptorId =
-            (spec.descriptor.source as? Descriptor.Source.Installed)?.value?.id
+        val installedDescriptorId = spec.descriptor.source.id
 
         startTest(
             spec.netTest,

composeApp/src/commonMain/kotlin/org/ooni/probe/ui/choosewebsites/ChooseWebsitesViewModel.kt

@@ -11,7 +11,9 @@ import kotlinx.coroutines.flow.onEach
 import kotlinx.coroutines.flow.update
 import org.ooni.engine.models.TaskOrigin
 import org.ooni.engine.models.TestType
+import org.ooni.probe.data.models.InstalledTestDescriptorModel
 import org.ooni.probe.data.models.NetTest
+import org.ooni.probe.data.models.OoniTest
 import org.ooni.probe.data.models.RunSpecification
 import org.ooni.probe.ui.shared.isValidUrl
 
@@ -104,7 +106,7 @@ class ChooseWebsitesViewModel(
                     RunSpecification.Full(
                         tests = listOf(
                             RunSpecification.Test(
-                                source = RunSpecification.Test.Source.Default("websites"),
+                                sourceId = InstalledTestDescriptorModel.Id(OoniTest.WEBSITES.id),
                                 netTests = listOf(
                                     NetTest(
                                         test = TestType.WebConnectivity,

composeApp/src/commonMain/kotlin/org/ooni/probe/ui/dashboard/DashboardViewModel.kt

@@ -113,9 +113,8 @@ class DashboardViewModel(
         events
             .filterIsInstance<Event.FetchUpdatedDescriptors>()
             .onEach {
-                state.value.descriptors[DescriptorType.Installed]
-                    ?.map { (it.source as Descriptor.Source.Installed).value }
-                    ?.let { descriptors ->
+                state.value.descriptors.flatMap { it.value }
+                    .map { it.source }.let { descriptors ->
                         fetchDescriptorUpdate(descriptors)
                     }
             }.launchIn(viewModelScope)
@@ -152,8 +151,8 @@ class DashboardViewModel(
 
     private fun List<Descriptor>.groupByType() =
         mapOf(
-            DescriptorType.Default to filter { it.source is Descriptor.Source.Default },
-            DescriptorType.Installed to filter { it.source is Descriptor.Source.Installed },
+            DescriptorType.Default to filter { it.isDefaultDescriptor() },
+            DescriptorType.Installed to filter { it.isInstalledNonDefaultDescriptor() },
         )
 
     data class State(

composeApp/src/commonMain/kotlin/org/ooni/probe/ui/descriptor/DescriptorScreen.kt

@@ -76,7 +76,7 @@ fun DescriptorScreen(
                 isRefreshing = state.isRefreshing,
                 onRefresh = { onEvent(DescriptorViewModel.Event.FetchUpdatedDescriptor) },
                 state = pullRefreshState,
-                enabled = state.descriptor?.source is Descriptor.Source.Installed,
+                enabled = true,
             )
             .background(MaterialTheme.colorScheme.background),
     ) {
@@ -127,9 +127,7 @@ fun DescriptorScreen(
                     modifier = Modifier.padding(horizontal = 16.dp, vertical = 16.dp),
                 )
 
-                if (descriptor.source is Descriptor.Source.Installed) {
-                    ConfigureUpdates(onEvent, descriptor.source.value.autoUpdate)
-                }
+                ConfigureUpdates(onEvent, descriptor.source.autoUpdate)
                 Text(
                     stringResource(Res.string.AddDescriptor_Settings),
                     style = MaterialTheme.typography.titleMedium,
@@ -162,13 +160,11 @@ fun DescriptorScreen(
                     TestDisplayMode.WebsitesOnly -> WebsiteItems(state.tests)
                 }
 
-                if (descriptor.source is Descriptor.Source.Installed) {
-                    InstalledDescriptorActionsView(
-                        descriptor = descriptor.source.value,
-                        onEvent = onEvent,
-                        modifier = Modifier.padding(horizontal = 16.dp),
-                    )
-                }
+                InstalledDescriptorActionsView(
+                    descriptor = descriptor.source,
+                    onEvent = onEvent,
+                    modifier = Modifier.padding(horizontal = 16.dp),
+                )
             }
         }
         if (state.isRefreshing) {