mirror of
https://github.com/irmen/prog8.git
synced 2024-11-22 00:31:56 +00:00
127 lines
4.7 KiB
Kotlin
127 lines
4.7 KiB
Kotlin
package prog8tests.compiler
|
|
|
|
import io.kotest.assertions.fail
|
|
import io.kotest.assertions.withClue
|
|
import io.kotest.core.spec.style.FunSpec
|
|
import io.kotest.matchers.shouldBe
|
|
import io.kotest.matchers.types.instanceOf
|
|
import prog8.ast.IFunctionCall
|
|
import prog8.ast.IStatementContainer
|
|
import prog8.ast.Program
|
|
import prog8.ast.expressions.IdentifierReference
|
|
import prog8.ast.expressions.NumericLiteral
|
|
import prog8.ast.statements.Assignment
|
|
import prog8.ast.statements.AssignmentOrigin
|
|
import prog8.ast.statements.VarDecl
|
|
import prog8.ast.statements.VarDeclType
|
|
import prog8.code.core.DataType
|
|
import prog8.code.core.Encoding
|
|
import prog8.code.target.Cx16Target
|
|
import prog8tests.helpers.compileText
|
|
|
|
|
|
/**
|
|
* ATTENTION: this is just kludge!
|
|
* They are not really unit tests, but rather tests of the whole process,
|
|
* from source file loading all the way through to running 64tass.
|
|
*/
|
|
class TestCompilerOnCharLit: FunSpec({
|
|
|
|
fun findInitializer(vardecl: VarDecl, program: Program): Assignment? =
|
|
(vardecl.parent as IStatementContainer).statements
|
|
.asSequence()
|
|
.filterIsInstance<Assignment>()
|
|
.singleOrNull { it.origin== AssignmentOrigin.VARINIT && it.target.identifier?.targetVarDecl(program) === vardecl }
|
|
|
|
|
|
test("testCharLitAsExtsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
extsub ${"$"}FFD2 = chrout(ubyte ch @ A)
|
|
sub start() {
|
|
chrout('\n')
|
|
}
|
|
}
|
|
""")!!
|
|
|
|
val program = result.compilerAst
|
|
val startSub = program.entrypoint
|
|
val funCall = startSub.statements.filterIsInstance<IFunctionCall>()[0]
|
|
|
|
withClue("char literal should have been replaced by ubyte literal") {
|
|
funCall.args[0] shouldBe instanceOf<NumericLiteral>()
|
|
}
|
|
val arg = funCall.args[0] as NumericLiteral
|
|
arg.type shouldBe DataType.UBYTE
|
|
arg.number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0].toDouble()
|
|
}
|
|
|
|
test("testCharVarAsExtsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
extsub ${"$"}FFD2 = chrout(ubyte ch @ A)
|
|
sub start() {
|
|
ubyte ch = '\n'
|
|
chrout(ch)
|
|
}
|
|
}
|
|
""")!!
|
|
|
|
val program = result.compilerAst
|
|
val startSub = program.entrypoint
|
|
val funCall = startSub.statements.filterIsInstance<IFunctionCall>()[0]
|
|
|
|
funCall.args[0] shouldBe instanceOf<IdentifierReference>()
|
|
val arg = funCall.args[0] as IdentifierReference
|
|
val decl = arg.targetVarDecl(program)!!
|
|
decl.type shouldBe VarDeclType.VAR
|
|
decl.datatype shouldBe DataType.UBYTE
|
|
|
|
withClue("initializer value should have been moved to separate assignment"){
|
|
decl.value shouldBe null
|
|
}
|
|
val assignInitialValue = findInitializer(decl, program)!!
|
|
assignInitialValue.target.identifier!!.nameInSource shouldBe listOf("ch")
|
|
withClue("char literal should have been replaced by ubyte literal") {
|
|
assignInitialValue.value shouldBe instanceOf<NumericLiteral>()
|
|
}
|
|
val initializerValue = assignInitialValue.value as NumericLiteral
|
|
initializerValue.type shouldBe DataType.UBYTE
|
|
initializerValue.number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0].toDouble()
|
|
}
|
|
|
|
test("testCharConstAsExtsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
extsub ${"$"}FFD2 = chrout(ubyte ch @ A)
|
|
sub start() {
|
|
const ubyte ch = '\n'
|
|
chrout(ch)
|
|
}
|
|
}
|
|
""")!!
|
|
|
|
val program = result.compilerAst
|
|
val startSub = program.entrypoint
|
|
val funCall = startSub.statements.filterIsInstance<IFunctionCall>()[0]
|
|
|
|
// Now, both is ok for the arg: a) still the IdRef or b) replaced by numeric literal
|
|
when (val arg = funCall.args[0]) {
|
|
is IdentifierReference -> {
|
|
val decl = arg.targetVarDecl(program)!!
|
|
decl.type shouldBe VarDeclType.CONST
|
|
decl.datatype shouldBe DataType.UBYTE
|
|
(decl.value as NumericLiteral).number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0]
|
|
}
|
|
is NumericLiteral -> {
|
|
arg.number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0].toDouble()
|
|
}
|
|
else -> fail("invalid arg type") // funCall.args[0] shouldBe instanceOf<IdentifierReference>() // make test fail
|
|
}
|
|
}
|
|
|
|
})
|