mirror of
https://github.com/irmen/prog8.git
synced 2024-12-02 07:49:27 +00:00
116 lines
4.3 KiB
Kotlin
116 lines
4.3 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.expressions.IdentifierReference
|
|
import prog8.ast.expressions.NumericLiteral
|
|
import prog8.ast.statements.VarDeclType
|
|
import prog8.code.core.BaseDataType
|
|
import prog8.code.core.DataTypeFull
|
|
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({
|
|
|
|
test("testCharLitAsRomsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
romsub ${"$"}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 BaseDataType.UBYTE
|
|
arg.number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0].toDouble()
|
|
}
|
|
|
|
test("testCharVarAsRomsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
romsub ${"$"}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 BaseDataType.UBYTE
|
|
|
|
withClue("initializer value should have been moved to separate assignment"){
|
|
decl.value shouldBe null
|
|
}
|
|
val assignInitialValue = decl.findInitializer(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 BaseDataType.UBYTE
|
|
initializerValue.number shouldBe platform.encodeString("\n", Encoding.PETSCII)[0].toDouble()
|
|
}
|
|
|
|
test("testCharConstAsRomsubArg") {
|
|
val platform = Cx16Target()
|
|
val result = compileText(platform, false, """
|
|
main {
|
|
romsub ${"$"}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 DataTypeFull.forDt(BaseDataType.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
|
|
}
|
|
}
|
|
|
|
})
|