smali2java-python/parser/class_test.py

from _class import ClassParser
from java_file import JavaFile
import java_file

def test_class(input :str, expected_output :str) -> None:
    parser = ClassParser()
    file = JavaFile()
    line = java_file.Line()
    for word in input.split(): line.append(word)
    parser.parse(file, line)
    print(expected_output == file.first().string())

def main() -> None:
    test_class('.class Lcom/example/app/MainActivity$4', \
        'class com.example.app.MainActivity$4 {' )
    test_class('.class public Lcom/example/app/MainActivity$4', \
        'public class com.example.app.MainActivity$4 {' )
    test_class('.class public interface abstract Lcom/example/app/MainActivity$4', \
        'public abstract class com.example.app.MainActivity$4 {' )

if __name__ == '__main__':
    main()