Successfully created and wrote to new file: /home/Messier82/projects/service-book/app/src/test/java/com/servicebook/ui/components/UpgradeDialogScreenshotTest.kt. Here is the updated code:
package com.servicebook.ui.components

import androidx.compose.ui.test.junit4.createComposeRule
import com.servicebook.ui.theme.ServiceBookTheme
import com.servicebook.ui.util.BaseScreenshotTest
import org.junit.Rule
import org.junit.Test

class UpgradeDialogScreenshotTest : BaseScreenshotTest() {
    @get:Rule
    val composeTestRule = createComposeRule()

    @Test
    fun dialog_light() {
        composeTestRule.setContent {
            ServiceBookTheme(dynamicColor = false, darkTheme = false) {
                UpgradeDialog(onUpgradeClick = {}, onDismissRequest = {})
            }
        }
        captureFull("UpgradeDialog_light")
    }

    @Test
    fun dialog_dark() {
        composeTestRule.setContent {
            ServiceBookTheme(dynamicColor = false, darkTheme = true) {
                UpgradeDialog(onUpgradeClick = {}, onDismissRequest = {})
            }
        }
        captureFull("UpgradeDialog_dark")
    }
}