| /* |
| * Copyright 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 androidx.ui.material |
| |
| import androidx.compose.Composable |
| import androidx.ui.core.Modifier |
| import androidx.ui.geometry.Size |
| import androidx.ui.layout.DpConstraints |
| import androidx.ui.layout.Stack |
| import androidx.ui.layout.preferredSizeIn |
| import androidx.ui.test.BigTestConstraints |
| import androidx.ui.test.CollectedSizes |
| import androidx.ui.test.ComposeTestRule |
| import androidx.ui.test.SemanticsNodeInteraction |
| import androidx.ui.test.getAlignmentLinePosition |
| import androidx.ui.test.runOnIdleCompose |
| import androidx.ui.test.setContentAndGetPixelSize |
| import androidx.ui.text.FirstBaseline |
| import androidx.ui.text.LastBaseline |
| import androidx.ui.unit.Density |
| |
| fun ComposeTestRule.setMaterialContent( |
| modifier: Modifier = Modifier, |
| composable: @Composable () -> Unit |
| ) { |
| setContent { |
| MaterialTheme { |
| Surface(modifier = modifier, content = composable) |
| } |
| } |
| } |
| fun ComposeTestRule.setMaterialContentWithConstraints( |
| parentConstraints: DpConstraints, |
| composable: @Composable () -> Unit |
| ) { |
| setContent { |
| MaterialTheme { |
| Stack { |
| Stack(Modifier.preferredSizeIn(parentConstraints)) { |
| composable() |
| } |
| } |
| } |
| } |
| } |
| fun <T> ComposeTestRule.runOnIdleComposeWithDensity(action: Density.() -> T): T { |
| return runOnIdleCompose { |
| density.action() |
| } |
| } |
| fun ComposeTestRule.setMaterialContentAndCollectSizes( |
| modifier: Modifier = Modifier, |
| parentConstraints: DpConstraints = BigTestConstraints, |
| children: @Composable () -> Unit |
| ): CollectedSizes { |
| val sizes = setMaterialContentAndGetPixelSize(modifier, parentConstraints, children) |
| return CollectedSizes(sizes, density) |
| } |
| |
| private fun ComposeTestRule.setMaterialContentAndGetPixelSize( |
| modifier: Modifier = Modifier, |
| parentConstraints: DpConstraints = BigTestConstraints, |
| children: @Composable () -> Unit |
| ): Size = setContentAndGetPixelSize( |
| parentConstraints, |
| { setMaterialContent(composable = it) } |
| ) { |
| MaterialTheme { |
| Surface(modifier = modifier, content = children) |
| } |
| } |
| |
| fun SemanticsNodeInteraction.getFirstBaselinePosition() = getAlignmentLinePosition(FirstBaseline) |
| |
| fun SemanticsNodeInteraction.getLastBaselinePosition() = getAlignmentLinePosition(LastBaseline) |