import org.jetbrains.kotlin.test.services.assertions

abstract class AbstractFileScopeTest : AbstractScopeTestBase() {
    override fun KaSession.getScope(mainFile: KtFile, testServices: TestServices): KaScope = mainFile.getFileSymbol().getFileScope()

    override fun doTestByMainFile(mainFile: KtFile, mainModule: KtTestModule, testServices: TestServices) {
        super.doTestByMainFile(mainFile, mainModule, testServices)

        analyseForTest(mainFile) {

        withValidityAssertion { analysisSession.scopeProvider.getDelegatedMemberScope(this) }

    public fun KaFileSymbol.getFileScope(): KaScope =
        withValidityAssertion { analysisSession.scopeProvider.getFileScope(this) }

    public fun KaPackageSymbol.getPackageScope(): KaScope =
        withValidityAssertion { analysisSession.scopeProvider.getPackageScope(this) }

            else -> containingDeclaration == owner
        }
    }

    override fun getEmptyScope(): KaScope {
        return KaEmptyScope(token)
    }

    override fun getFileScope(fileSymbol: KaFileSymbol): KaScope {
        require(fileSymbol is KaFe10FileSymbol)
        return KaFe10FileScope(fileSymbol.psi, analysisContext, token)
    }

                            is KaDeclarationSymbol -> symbol.render(prettyRenderer)
                            is KaFileSymbol -> prettyPrint {
                                printCollection(symbol.getFileScope().getAllSymbols().asIterable(), separator = "\n\n") {
                                    append(it.render(prettyRenderer))
                                }
                            }

            getScopeSession(),
            fir,
            declaredScope,
            delegateFields
        )
        return KaFirDelegatedMemberScope(firScope, builder)
    }

    override fun getFileScope(fileSymbol: KaFileSymbol): KaScope {
        check(fileSymbol is KaFirFileSymbol) { "KtFirScopeProvider can only work with KtFirFileSymbol, but ${fileSymbol::class} was provided" }
        return KaFirFileScope(fileSymbol, builder)