testapps/toolbox/src/com/android/healthconnect/testapps/toolbox/utils/EnumFieldsWithValues.kt

/**
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * ```
 *      http://www.apache.org/licenses/LICENSE-2.0
 * ```
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package com.android.healthconnect.testapps.toolbox.utils

class EnumFieldsWithValues(private val fieldNameToValue: Map<String, Any>) {

    fun getAllFieldNames(): List<String> {
        return fieldNameToValue.keys.toList()
    }

    fun getFieldValue(fieldName: String): Any {
        return fieldNameToValue.getOrElse(fieldName) {
            throw java.lang.IllegalArgumentException("Field name not found!")
        }
    }
}