mirror of
https://github.com/irmen/prog8.git
synced 2024-07-19 04:28:56 +00:00
101 lines
4.1 KiB
Kotlin
101 lines
4.1 KiB
Kotlin
package prog8tests
|
|
|
|
import io.kotest.assertions.fail
|
|
import io.kotest.core.spec.style.FunSpec
|
|
import io.kotest.matchers.shouldBe
|
|
import io.kotest.matchers.shouldNotBe
|
|
import prog8.code.*
|
|
import prog8.code.core.DataType
|
|
import prog8.code.core.Position
|
|
import prog8.code.core.ZeropageWish
|
|
|
|
class TestSymbolTable: FunSpec({
|
|
test("empty symboltable") {
|
|
val st = SymbolTable()
|
|
st.scopedName shouldBe ""
|
|
st.name shouldBe ""
|
|
st.type shouldBe StNodeType.GLOBAL
|
|
st.children shouldBe mutableMapOf()
|
|
st.position shouldBe Position.DUMMY
|
|
}
|
|
|
|
test("symboltable flatten") {
|
|
val st = makeSt()
|
|
st.flat["zzzzz"] shouldBe null
|
|
st.flat.getValue("msb").type shouldBe StNodeType.BUILTINFUNC
|
|
st.flat.getValue("block2").type shouldBe StNodeType.BLOCK
|
|
st.flat.getValue("block2.sub2.subsub.label").type shouldBe StNodeType.LABEL
|
|
st.flat["block2.sub2.subsub.label.zzzz"] shouldBe null
|
|
}
|
|
|
|
test("symboltable global lookups") {
|
|
val st = makeSt()
|
|
st.lookupUnscoped("undefined") shouldBe null
|
|
st.lookup("undefined") shouldBe null
|
|
st.lookup("undefined.undefined") shouldBe null
|
|
var default = st.lookupUnscopedOrElse("undefined") { StNode("default", StNodeType.LABEL, Position.DUMMY) }
|
|
default.name shouldBe "default"
|
|
default = st.lookupUnscopedOrElse("undefined") { StNode("default", StNodeType.LABEL, Position.DUMMY) }
|
|
default.name shouldBe "default"
|
|
|
|
val msbFunc = st.lookupUnscopedOrElse("msb") { fail("msb must be found") }
|
|
msbFunc.type shouldBe StNodeType.BUILTINFUNC
|
|
|
|
val variable = st.lookupOrElse("block1.sub2.v2") { fail("v2 must be found") }
|
|
variable.type shouldBe StNodeType.STATICVAR
|
|
}
|
|
|
|
test("symboltable nested lookups") {
|
|
val st = makeSt()
|
|
|
|
val sub1 = st.lookupOrElse("block1.sub1") { fail("should find sub1") }
|
|
sub1.name shouldBe "sub1"
|
|
sub1.scopedName shouldBe "block1.sub1"
|
|
sub1.type shouldBe StNodeType.SUBROUTINE
|
|
sub1.children.size shouldBe 2
|
|
|
|
val v1 = sub1.lookupUnscopedOrElse("v1") { fail("v1 must be found") } as StStaticVariable
|
|
v1.type shouldBe StNodeType.STATICVAR
|
|
v1.name shouldBe "v1"
|
|
v1.dt shouldBe DataType.BYTE
|
|
|
|
val blockc = sub1.lookupUnscopedOrElse("blockc") { fail("blockc") } as StConstant
|
|
blockc.type shouldBe StNodeType.CONSTANT
|
|
blockc.value shouldBe 999.0
|
|
|
|
val subsub = st.lookupOrElse("block2.sub2.subsub") { fail("should find subsub") }
|
|
subsub.lookupUnscoped("blockc") shouldBe null
|
|
subsub.lookupUnscoped("label") shouldNotBe null
|
|
}
|
|
})
|
|
|
|
|
|
private fun makeSt(): SymbolTable {
|
|
val st = SymbolTable()
|
|
val block1 = StNode("block1", StNodeType.BLOCK, Position.DUMMY)
|
|
val sub11 = StNode("sub1", StNodeType.SUBROUTINE, Position.DUMMY)
|
|
val sub12 = StNode("sub2", StNodeType.SUBROUTINE, Position.DUMMY)
|
|
block1.add(sub11)
|
|
block1.add(sub12)
|
|
block1.add(StConstant("c1", DataType.UWORD, 12345.0, Position.DUMMY))
|
|
block1.add(StConstant("blockc", DataType.UWORD, 999.0, Position.DUMMY))
|
|
sub11.add(StStaticVariable("v1", DataType.BYTE, true, null, null, null, null, ZeropageWish.DONTCARE, Position.DUMMY))
|
|
sub11.add(StStaticVariable("v2", DataType.BYTE, true, null, null, null, null, ZeropageWish.DONTCARE, Position.DUMMY))
|
|
sub12.add(StStaticVariable("v1", DataType.BYTE, true, null, null, null, null, ZeropageWish.DONTCARE, Position.DUMMY))
|
|
sub12.add(StStaticVariable("v2", DataType.BYTE, true, null, null, null, null, ZeropageWish.DONTCARE, Position.DUMMY))
|
|
|
|
val block2 = StNode("block2", StNodeType.BLOCK, Position.DUMMY)
|
|
val sub21 = StNode("sub1", StNodeType.SUBROUTINE, Position.DUMMY)
|
|
val sub22 = StNode("sub2", StNodeType.SUBROUTINE, Position.DUMMY)
|
|
block2.add(sub21)
|
|
block2.add(sub22)
|
|
val sub221 = StNode("subsub", StNodeType.SUBROUTINE, Position.DUMMY)
|
|
sub221.add(StNode("label", StNodeType.LABEL, Position.DUMMY))
|
|
sub22.add(sub221)
|
|
|
|
val builtinfunc = StNode("msb", StNodeType.BUILTINFUNC, Position.DUMMY)
|
|
st.add(block1)
|
|
st.add(block2)
|
|
st.add(builtinfunc)
|
|
return st
|
|
} |