appium-espresso-driver 8.0.0 → 8.2.0

package/espresso-server/library/src/composeOff/java/io/appium/espressoserver/lib/compose/ComposeHandlerBridge.kt

@@ -0,0 +1,73 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.compose
+
+import io.appium.espressoserver.lib.handlers.exceptions.ComposeNotSupportedException
+import io.appium.espressoserver.lib.model.AppiumParams
+import io.appium.espressoserver.lib.model.BaseElement
+import io.appium.espressoserver.lib.model.Locator
+import io.appium.espressoserver.lib.model.Location
+import io.appium.espressoserver.lib.model.MobileSwipeParams
+import io.appium.espressoserver.lib.model.Rect
+import io.appium.espressoserver.lib.model.Size
+import io.appium.espressoserver.lib.model.TextValueParams
+
+internal object ComposeHandlerBridge {
+    private fun unsupported(): Nothing = throw ComposeNotSupportedException()
+
+    fun findElement(params: Locator): BaseElement = unsupported()
+
+    fun findElements(params: Locator): List<BaseElement> = unsupported()
+
+    fun click(params: AppiumParams) {
+        unsupported()
+    }
+
+    fun clear(params: AppiumParams) {
+        unsupported()
+    }
+
+    fun elementValue(params: TextValueParams, isReplacing: Boolean) {
+        unsupported()
+    }
+
+    fun keys(params: TextValueParams) {
+        unsupported()
+    }
+
+    fun mobileSwipe(params: MobileSwipeParams): Void? = unsupported()
+
+    fun getDisplayed(params: AppiumParams): Boolean = unsupported()
+
+    fun elementScreenshot(elementId: String): String = unsupported()
+
+    fun getAttribute(elementId: String, attributeName: String): String? = unsupported()
+
+    fun getEnabled(params: AppiumParams): Boolean = unsupported()
+
+    fun getLocation(params: AppiumParams): Location = unsupported()
+
+    fun getName(params: AppiumParams): String? = unsupported()
+
+    fun getRect(params: AppiumParams): Rect = unsupported()
+
+    fun getSelected(params: AppiumParams): Boolean = unsupported()
+
+    fun getSize(params: AppiumParams): Size = unsupported()
+
+    fun text(params: AppiumParams): String = unsupported()
+}

package/espresso-server/library/src/composeOff/java/io/appium/espressoserver/lib/helpers/ComposeViewCache.kt

@@ -0,0 +1,29 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.helpers
+
+/** No-op when Jetpack Compose is not included in this server build. */
+object ComposeViewCache {
+    @Suppress("UNUSED_PARAMETER")
+    fun put(id: String, composeNode: Any) {
+    }
+
+    fun get(id: String?): Any? = null
+
+    fun reset() {
+    }
+}

package/espresso-server/library/src/composeOff/java/io/appium/espressoserver/lib/http/Server.kt

@@ -0,0 +1,21 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.http
+
+class Server : ServerBase() {
+    override fun advanceComposeMainClockIfNeeded() = Unit
+}

package/espresso-server/library/src/composeOff/java/io/appium/espressoserver/lib/model/SourceDocumentComposeHooks.kt

@@ -0,0 +1,26 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.model
+
+import io.appium.espressoserver.lib.handlers.exceptions.ComposeNotSupportedException
+
+internal object SourceDocumentComposeHooks {
+    @Suppress("UNUSED_PARAMETER")
+    fun appendComposeStrategy(doc: SourceDocument, root: Any?) {
+        throw ComposeNotSupportedException()
+    }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/compose/ComposeHandlerBridge.kt

@@ -0,0 +1,180 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.compose
+
+import androidx.compose.ui.test.assertIsDisplayed
+import androidx.compose.ui.test.performClick
+import androidx.compose.ui.test.performGesture
+import androidx.compose.ui.test.performTextClearance
+import androidx.compose.ui.test.performTextInput
+import androidx.compose.ui.test.swipeDown
+import androidx.compose.ui.test.swipeLeft
+import androidx.compose.ui.test.swipeRight
+import androidx.compose.ui.test.swipeUp
+import io.appium.espressoserver.lib.handlers.exceptions.InvalidArgumentException
+import io.appium.espressoserver.lib.handlers.exceptions.InvalidElementStateException
+import io.appium.espressoserver.lib.handlers.exceptions.InvalidSelectorException
+import io.appium.espressoserver.lib.handlers.exceptions.NoSuchElementException
+import io.appium.espressoserver.lib.handlers.exceptions.StaleElementException
+import io.appium.espressoserver.lib.helpers.AndroidLogger
+import io.appium.espressoserver.lib.helpers.getNodeInteractionById
+import io.appium.espressoserver.lib.helpers.getSemanticsNode
+import io.appium.espressoserver.lib.helpers.toNodeInteractionsCollection
+import io.appium.espressoserver.lib.model.AppiumParams
+import io.appium.espressoserver.lib.model.BaseElement
+import io.appium.espressoserver.lib.model.ComposeElement
+import io.appium.espressoserver.lib.model.ComposeNodeElement
+import io.appium.espressoserver.lib.model.Locator
+import io.appium.espressoserver.lib.model.Location
+import io.appium.espressoserver.lib.model.MobileSwipeParams
+import io.appium.espressoserver.lib.model.Rect
+import io.appium.espressoserver.lib.model.Size
+import io.appium.espressoserver.lib.model.TextValueParams
+
+internal object ComposeHandlerBridge {
+    fun findElement(params: Locator): BaseElement {
+        val nodeInteractions = toNodeInteractionsCollection(params)
+        if (nodeInteractions.fetchSemanticsNodes(false).isEmpty()) {
+            throw NoSuchElementException(
+                String.format(
+                    "Could not find a compose element with strategy '%s' and selector '%s'",
+                    params.using,
+                    params.value,
+                ),
+            )
+        }
+        return ComposeElement(nodeInteractions[0])
+    }
+
+    fun findElements(params: Locator): List<BaseElement> {
+        val nodeInteractions = toNodeInteractionsCollection(params)
+        return List(nodeInteractions.fetchSemanticsNodes(false).size) { index ->
+            ComposeElement(nodeInteractions[index])
+        }
+    }
+
+    fun click(params: AppiumParams) {
+        try {
+            getNodeInteractionById(params.elementId).performClick()
+        } catch (e: AssertionError) {
+            throw StaleElementException(params.elementId!!)
+        } catch (e: IllegalArgumentException) {
+            throw InvalidElementStateException("Click", params.elementId!!, e)
+        }
+    }
+
+    fun clear(params: AppiumParams) {
+        try {
+            getNodeInteractionById(params.elementId).performTextClearance()
+        } catch (e: AssertionError) {
+            throw StaleElementException(params.elementId!!)
+        } catch (e: IllegalArgumentException) {
+            throw InvalidElementStateException("Clear", params.elementId!!, e)
+        }
+    }
+
+    fun elementValue(params: TextValueParams, isReplacing: Boolean) {
+        val value: String = extractTextToEnter(params)
+        try {
+            if (isReplacing) {
+                getNodeInteractionById(params.elementId).performTextClearance()
+            }
+            getNodeInteractionById(params.elementId).performTextInput(value)
+        } catch (e: IllegalArgumentException) {
+            throw InvalidElementStateException("Clear", params.elementId!!, e)
+        }
+    }
+
+    fun keys(params: TextValueParams) {
+        try {
+            val keys = params.value ?: return
+            keys.forEach {
+                getNodeInteractionById(params.elementId).performTextInput(it)
+            }
+        } catch (e: AssertionError) {
+            throw StaleElementException(params.elementId!!)
+        } catch (e: IllegalArgumentException) {
+            throw InvalidElementStateException("Keys", params.elementId!!, e)
+        }
+    }
+
+    fun mobileSwipe(params: MobileSwipeParams): Void? {
+        val nodeInteractions = getNodeInteractionById(params.elementId)
+        AndroidLogger.info("Performing swipe action with direction '${params.direction}'")
+        when (params.direction) {
+            MobileSwipeParams.Direction.UP -> nodeInteractions.performGesture { swipeUp() }
+            MobileSwipeParams.Direction.DOWN -> nodeInteractions.performGesture { swipeDown() }
+            MobileSwipeParams.Direction.LEFT -> nodeInteractions.performGesture { swipeLeft() }
+            MobileSwipeParams.Direction.RIGHT -> nodeInteractions.performGesture { swipeRight() }
+            else -> throw InvalidArgumentException(
+                "Unknown swipe direction '${params.direction}'. " +
+                    "Only the following values are supported: " +
+                    MobileSwipeParams.Direction.values().joinToString(",") { x -> x.name.lowercase() },
+            )
+        }
+        return null
+    }
+
+    fun getDisplayed(params: AppiumParams): Boolean =
+        try {
+            getNodeInteractionById(params.elementId).assertIsDisplayed()
+            true
+        } catch (e: AssertionError) {
+            false
+        }
+
+    fun elementScreenshot(elementId: String): String =
+        ComposeScreenshot.takeComposeNodeScreenshot(getNodeInteractionById(elementId))
+
+    fun getAttribute(elementId: String, attributeName: String): String? =
+        ComposeNodeElement(getSemanticsNode(elementId)).getAttribute(attributeName)
+
+    fun getEnabled(params: AppiumParams): Boolean =
+        ComposeNodeElement(getSemanticsNode(params.elementId!!)).isEnabled
+
+    fun getLocation(params: AppiumParams): Location {
+        val composeNodeElement = ComposeNodeElement(getSemanticsNode(params.elementId!!))
+        return Location(composeNodeElement.bounds.left, composeNodeElement.bounds.top)
+    }
+
+    fun getName(params: AppiumParams): String? {
+        val composeNodeElement = ComposeNodeElement(getSemanticsNode(params.elementId!!))
+        return composeNodeElement.contentDescription?.toString()
+    }
+
+    fun getRect(params: AppiumParams): Rect =
+        ComposeNodeElement(getSemanticsNode(params.elementId!!)).rect
+
+    fun getSelected(params: AppiumParams): Boolean =
+        ComposeNodeElement(getSemanticsNode(params.elementId!!)).isSelected
+
+    fun getSize(params: AppiumParams): Size {
+        val bounds = ComposeNodeElement(getSemanticsNode(params.elementId!!)).bounds
+        return Size(bounds.width(), bounds.height())
+    }
+
+    fun text(params: AppiumParams): String =
+        ComposeNodeElement(getSemanticsNode(params.elementId!!)).text.toString()
+
+    private fun extractTextToEnter(params: TextValueParams) =
+        when (Pair(params.value == null, params.text == null)) {
+            Pair(first = true, second = true) ->
+                throw InvalidArgumentException("Must provide 'value' or 'text' property")
+            Pair(first = false, second = true) -> params.value!!.joinToString(separator = "")
+            else -> params.text!!
+        }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/compose/ComposeScreenshot.kt

@@ -0,0 +1,49 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.compose
+
+import android.graphics.Bitmap
+import android.util.Base64
+import androidx.compose.ui.graphics.asAndroidBitmap
+import androidx.compose.ui.test.SemanticsNodeInteraction
+import androidx.compose.ui.test.captureToImage
+import io.appium.espressoserver.lib.handlers.exceptions.ScreenCaptureException
+import java.io.ByteArrayOutputStream
+
+internal object ComposeScreenshot {
+    fun takeComposeNodeScreenshot(nodeInteraction: SemanticsNodeInteraction): String =
+        try {
+            encodeBitmap(
+                nodeInteraction.captureToImage().asAndroidBitmap(),
+            ) { nodeInteraction.fetchSemanticsNode().toString() }
+        } catch (e: RuntimeException) {
+            throw ScreenCaptureException("Cannot take a screenshot of a node", e)
+        }
+
+    private fun encodeBitmap(bitmap: Bitmap?, targetNameSupplier: () -> String): String {
+        if (bitmap == null || bitmap.height == 0 || bitmap.width == 0) {
+            throw ScreenCaptureException(
+                "Cannot capture a shot of the ${targetNameSupplier()}. " +
+                    "Make sure none of the currently visible " +
+                    "views have FLAG_SECURE set and that it is possible to take a screenshot manually",
+            )
+        }
+        val outputStream = ByteArrayOutputStream()
+        bitmap.compress(Bitmap.CompressFormat.PNG, 100, outputStream)
+        return Base64.encodeToString(outputStream.toByteArray(), Base64.NO_WRAP)
+    }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/drivers/DriverContextComposeRule.kt

@@ -0,0 +1,29 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.drivers
+
+import androidx.compose.ui.test.junit4.ComposeTestRule
+import androidx.compose.ui.test.junit4.createEmptyComposeRule
+
+private val emptyComposeRule: ComposeTestRule = createEmptyComposeRule()
+
+/**
+ * Compose UI test rule for this build. Import [composeTestRule] when using from other packages
+ * (`import io.appium.espressoserver.lib.drivers.composeTestRule`).
+ */
+val DriverContext.composeTestRule: ComposeTestRule
+    get() = emptyComposeRule

package/espresso-server/library/src/{main → composeOn}/java/io/appium/espressoserver/lib/helpers/ComposeNodeFinder.kt

@@ -19,6 +19,7 @@ package io.appium.espressoserver.lib.helpers
 import androidx.compose.ui.semantics.SemanticsNode
 import androidx.compose.ui.test.*
 import io.appium.espressoserver.lib.drivers.DriverContext
+import io.appium.espressoserver.lib.drivers.composeTestRule
 import io.appium.espressoserver.lib.handlers.exceptions.InvalidSelectorException
 import io.appium.espressoserver.lib.model.Strategy
 import io.appium.espressoserver.lib.handlers.exceptions.InvalidArgumentException

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/helpers/ComposeViewCache.kt

@@ -0,0 +1,37 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.helpers
+
+import android.util.LruCache
+import androidx.compose.ui.test.SemanticsNodeInteraction
+
+const val COMPOSE_VIEW_CACHE_MAX_SIZE = 500
+
+/** Compose does not have stable instances, use locator again. */
+object ComposeViewCache {
+    private val cache = LruCache<String, SemanticsNodeInteraction>(COMPOSE_VIEW_CACHE_MAX_SIZE)
+
+    fun put(id: String, composeNode: SemanticsNodeInteraction) {
+        cache.put(id, composeNode)
+    }
+
+    fun get(id: String?): SemanticsNodeInteraction? = cache.get(id)
+
+    fun reset() {
+        cache.evictAll()
+    }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/http/Server.kt

@@ -0,0 +1,30 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.http
+
+import io.appium.espressoserver.lib.drivers.DriverContext
+import io.appium.espressoserver.lib.drivers.composeTestRule
+import org.junit.rules.TestRule
+
+class Server : ServerBase(), TestRule by DriverContext.composeTestRule {
+
+    override fun advanceComposeMainClockIfNeeded() {
+        if (DriverContext.currentStrategyType == DriverContext.StrategyType.COMPOSE) {
+            DriverContext.composeTestRule.mainClock.advanceTimeByFrame()
+        }
+    }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/model/ComposeElement.kt

@@ -0,0 +1,33 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.model
+
+import androidx.compose.ui.test.SemanticsNodeInteraction
+import com.google.gson.annotations.SerializedName
+import io.appium.espressoserver.lib.helpers.ComposeViewCache
+import java.util.UUID
+
+class ComposeElement(node: SemanticsNodeInteraction) : BaseElement {
+    @Suppress("JoinDeclarationAndAssignment")
+    @SerializedName(JSONWP_ELEMENT_KEY, alternate = [W3C_ELEMENT_KEY])
+    override val element: String
+
+    init {
+        element = UUID.randomUUID().toString()
+        ComposeViewCache.put(element, node)
+    }
+}

package/espresso-server/library/src/composeOn/java/io/appium/espressoserver/lib/model/SourceDocumentComposeHooks.kt

@@ -0,0 +1,103 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * See the NOTICE file distributed with this work for additional
+ * information regarding copyright ownership.
+ * 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 io.appium.espressoserver.lib.model
+
+import androidx.compose.ui.semantics.SemanticsNode
+import androidx.compose.ui.test.SelectionResult
+import androidx.compose.ui.test.SemanticsNodeInteraction
+import androidx.compose.ui.test.onRoot
+import io.appium.espressoserver.lib.drivers.DriverContext
+import io.appium.espressoserver.lib.drivers.composeTestRule
+import io.appium.espressoserver.lib.helpers.AndroidLogger
+import kotlin.reflect.full.declaredMemberFunctions
+import kotlin.reflect.jvm.isAccessible
+
+internal object SourceDocumentComposeHooks {
+    fun appendComposeStrategy(doc: SourceDocument, root: Any?) {
+        if (root != null) {
+            serializeComposeNode(doc, root as SemanticsNode, 0)
+        } else {
+            val rootNodes = rootSemanticNodes()
+            if (rootNodes.size == 1) {
+                serializeComposeNode(doc, rootNodes.first(), 0)
+            } else {
+                doc.serializer?.startTag(NAMESPACE, DEFAULT_TAG_NAME)
+                rootNodes.forEach { semanticsNode -> serializeComposeNode(doc, semanticsNode, 0) }
+                doc.serializer?.endTag(NAMESPACE, DEFAULT_TAG_NAME)
+            }
+        }
+    }
+
+    private fun serializeComposeNode(doc: SourceDocument, semanticsNode: SemanticsNode?, depth: Int) {
+        if (semanticsNode == null) {
+            return
+        }
+        val nodeElement = ComposeNodeElement(semanticsNode)
+        val className = nodeElement.className
+        val tagName = toXmlNodeName(className)
+        doc.serializer?.startTag(NAMESPACE, tagName)
+
+        linkedMapOf(
+            AttributesEnum.CLASS to { className },
+            AttributesEnum.INDEX to { nodeElement.index },
+            AttributesEnum.CLICKABLE to { nodeElement.isClickable },
+            AttributesEnum.ENABLED to { nodeElement.isEnabled },
+            AttributesEnum.FOCUSED to { nodeElement.isFocused },
+            AttributesEnum.SCROLLABLE to { nodeElement.isScrollable },
+            AttributesEnum.SELECTED to { nodeElement.isSelected },
+            AttributesEnum.CHECKED to { nodeElement.isChecked },
+            AttributesEnum.VIEW_TAG to { nodeElement.viewTag },
+            AttributesEnum.CONTENT_DESC to { nodeElement.contentDescription },
+            AttributesEnum.BOUNDS to { nodeElement.bounds.toShortString() },
+            AttributesEnum.TEXT to { nodeElement.text },
+            AttributesEnum.PASSWORD to { nodeElement.isPassword },
+            AttributesEnum.RESOURCE_ID to { nodeElement.resourceId },
+        ).forEach {
+            doc.setAttribute(it.key, it.value())
+        }
+
+        if (depth < MAX_TRAVERSAL_DEPTH) {
+            val children = semanticsNode.children
+            for (index in 0 until children.count()) {
+                serializeComposeNode(doc, children[index], depth + 1)
+            }
+        } else {
+            AndroidLogger.warn(
+                "Skipping traversal of ${semanticsNode.javaClass.name}'s children, since " +
+                    "the current depth has reached its maximum allowed value of $depth",
+            )
+        }
+
+        doc.serializer?.endTag(NAMESPACE, tagName)
+    }
+
+    private fun rootSemanticNodes(): List<SemanticsNode> =
+        try {
+            listOf(DriverContext.composeTestRule.onRoot(useUnmergedTree = true).fetchSemanticsNode())
+        } catch (e: AssertionError) {
+            val result: SelectionResult =
+                SemanticsNodeInteraction::class.declaredMemberFunctions.find { it.name == "fetchSemanticsNodes" }?.let {
+                    it.isAccessible = true
+                    if (it.parameters.size == 4) {
+                        it.call(DriverContext.composeTestRule.onRoot(useUnmergedTree = true), true, null, true)
+                    } else {
+                        it.call(DriverContext.composeTestRule.onRoot(useUnmergedTree = true), true, null)
+                    }
+                } as SelectionResult
+            result.selectedNodes
+        }
+}

package/espresso-server/library/src/main/java/io/appium/espressoserver/lib/drivers/DriverContext.kt

@@ -16,19 +16,16 @@
 
 package io.appium.espressoserver.lib.drivers
 
-import androidx.compose.ui.test.junit4.createEmptyComposeRule
-
 object DriverContext {
     var currentStrategyType: StrategyType = StrategyType.ESPRESSO
         private set
 
-    val composeTestRule = createEmptyComposeRule()
-
     fun setDriverStrategy(strategyType: StrategyType) {
         currentStrategyType = strategyType
     }
 
     enum class StrategyType {
-        ESPRESSO, COMPOSE
+        ESPRESSO,
+        COMPOSE,
     }
-}
+}