| /* |
| * Copyright (C) 2019 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.protolog.tool |
| |
| import com.android.json.stream.JsonReader |
| import com.android.protolog.tool.ViewerConfigBuilder.LogCall |
| import org.junit.Assert.assertEquals |
| import org.junit.Test |
| import org.mockito.Mockito |
| import java.io.StringReader |
| |
| class ViewerConfigBuilderTest { |
| companion object { |
| private val TAG1 = "WM_TEST" |
| private val TAG2 = "WM_DEBUG" |
| private val TEST1 = ViewerConfigParser.ConfigEntry("test1", LogLevel.INFO.name, TAG1) |
| private val TEST2 = ViewerConfigParser.ConfigEntry("test2", LogLevel.DEBUG.name, TAG2) |
| private val TEST3 = ViewerConfigParser.ConfigEntry("test3", LogLevel.ERROR.name, TAG2) |
| private val GROUP1 = LogGroup("TEST_GROUP", true, true, TAG1) |
| private val GROUP2 = LogGroup("DEBUG_GROUP", true, true, TAG2) |
| private val GROUP3 = LogGroup("DEBUG_GROUP", true, true, TAG2) |
| private val GROUP_DISABLED = LogGroup("DEBUG_GROUP", false, true, TAG2) |
| private val GROUP_TEXT_DISABLED = LogGroup("DEBUG_GROUP", true, false, TAG2) |
| private const val PATH = "/tmp/test.java" |
| } |
| |
| private val configBuilder = ViewerConfigBuilder(Mockito.mock(ProtoLogCallProcessor::class.java)) |
| |
| private fun parseConfig(json: String): Map<Int, ViewerConfigParser.ConfigEntry> { |
| return ViewerConfigParser().parseConfig(JsonReader(StringReader(json))) |
| } |
| |
| @Test |
| fun processClass() { |
| configBuilder.addLogCalls(listOf( |
| LogCall(TEST1.messageString, LogLevel.INFO, GROUP1, PATH), |
| LogCall(TEST2.messageString, LogLevel.DEBUG, GROUP2, PATH), |
| LogCall(TEST3.messageString, LogLevel.ERROR, GROUP3, PATH)).withContext()) |
| |
| val parsedConfig = parseConfig(configBuilder.build()) |
| assertEquals(3, parsedConfig.size) |
| assertEquals(TEST1, parsedConfig[CodeUtils.hash(PATH, |
| TEST1.messageString, LogLevel.INFO, GROUP1)]) |
| assertEquals(TEST2, parsedConfig[CodeUtils.hash(PATH, TEST2.messageString, |
| LogLevel.DEBUG, GROUP2)]) |
| assertEquals(TEST3, parsedConfig[CodeUtils.hash(PATH, TEST3.messageString, |
| LogLevel.ERROR, GROUP3)]) |
| } |
| |
| @Test |
| fun processClass_nonUnique() { |
| configBuilder.addLogCalls(listOf( |
| LogCall(TEST1.messageString, LogLevel.INFO, GROUP1, PATH), |
| LogCall(TEST1.messageString, LogLevel.INFO, GROUP1, PATH), |
| LogCall(TEST1.messageString, LogLevel.INFO, GROUP1, PATH)).withContext()) |
| |
| val parsedConfig = parseConfig(configBuilder.build()) |
| assertEquals(1, parsedConfig.size) |
| assertEquals(TEST1, parsedConfig[CodeUtils.hash(PATH, TEST1.messageString, |
| LogLevel.INFO, GROUP1)]) |
| } |
| |
| @Test |
| fun processClass_disabled() { |
| configBuilder.addLogCalls(listOf( |
| LogCall(TEST1.messageString, LogLevel.INFO, GROUP1, PATH), |
| LogCall(TEST2.messageString, LogLevel.DEBUG, GROUP_DISABLED, PATH), |
| LogCall(TEST3.messageString, LogLevel.ERROR, GROUP_TEXT_DISABLED, PATH)) |
| .withContext()) |
| |
| val parsedConfig = parseConfig(configBuilder.build()) |
| assertEquals(2, parsedConfig.size) |
| assertEquals(TEST1, parsedConfig[CodeUtils.hash( |
| PATH, TEST1.messageString, LogLevel.INFO, GROUP1)]) |
| assertEquals(TEST3, parsedConfig[CodeUtils.hash( |
| PATH, TEST3.messageString, LogLevel.ERROR, GROUP_TEXT_DISABLED)]) |
| } |
| |
| private fun List<LogCall>.withContext() = map { it to ParsingContext() } |
| } |