sx/src/ir/jni_descriptor.test.zig

// Tests for jni_descriptor.zig — Phase 2 step 2.8.
// Table-driven golden test for the primitive / array / *Self JNI
// signature alphabet. Cross-class references land in 2.9.

const std = @import("std");
const ast = @import("../ast.zig");
const desc = @import("jni_descriptor.zig");

const Node = ast.Node;

fn makeTypeExpr(allocator: std.mem.Allocator, name: []const u8) !*Node {
    const node = try allocator.create(Node);
    node.* = .{
        .span = .{ .start = 0, .end = 0 },
        .data = .{ .type_expr = .{ .name = name } },
    };
    return node;
}

fn makePointer(allocator: std.mem.Allocator, pointee: *Node) !*Node {
    const node = try allocator.create(Node);
    node.* = .{
        .span = .{ .start = 0, .end = 0 },
        .data = .{ .pointer_type_expr = .{ .pointee_type = pointee } },
    };
    return node;
}

fn makeSlice(allocator: std.mem.Allocator, element: *Node) !*Node {
    const node = try allocator.create(Node);
    node.* = .{
        .span = .{ .start = 0, .end = 0 },
        .data = .{ .slice_type_expr = .{ .element_type = element } },
    };
    return node;
}

fn expectType(name: []const u8, expected: []const u8) !void {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    const tn = try makeTypeExpr(aa, name);
    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{ .enclosing_path = "" }, tn);
    try std.testing.expectEqualStrings(expected, buf.items);
}

test "primitive descriptors" {
    try expectType("void", "V");
    try expectType("bool", "Z");
    try expectType("s8", "B");
    try expectType("u8", "B");
    try expectType("s16", "S");
    try expectType("u16", "C");
    try expectType("s32", "I");
    try expectType("s64", "J");
    try expectType("f32", "F");
    try expectType("f64", "D");
}

test "void return is V (null type_node)" {
    const a = std.testing.allocator;
    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{ .enclosing_path = "" }, null);
    try std.testing.expectEqualStrings("V", buf.items);
}

test "*void resolves to java/lang/Object (opaque jobject)" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    const void_te = try makeTypeExpr(aa, "void");
    const ptr = try makePointer(aa, void_te);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{ .enclosing_path = "anything" }, ptr);
    try std.testing.expectEqualStrings("Ljava/lang/Object;", buf.items);
}

test "*Self resolves to enclosing class L-form" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    const self_te = try makeTypeExpr(aa, "Self");
    const ptr = try makePointer(aa, self_te);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{ .enclosing_path = "android/view/View" }, ptr);
    try std.testing.expectEqualStrings("Landroid/view/View;", buf.items);
}

test "slice of primitive is array descriptor" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    const s32 = try makeTypeExpr(aa, "s32");
    const slice = try makeSlice(aa, s32);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{ .enclosing_path = "" }, slice);
    try std.testing.expectEqualStrings("[I", buf.items);
}

test "cross-class *Foo resolves via class registry" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    var registry = desc.ClassRegistry.init(a);
    defer registry.deinit();
    try registry.put("Window", "android/view/Window");
    try registry.put("View", "android/view/View");

    const foo = try makeTypeExpr(aa, "Window");
    const ptr = try makePointer(aa, foo);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    try desc.writeType(a, &buf, .{
        .enclosing_path = "android/view/View",
        .classes = &registry,
    }, ptr);
    try std.testing.expectEqualStrings("Landroid/view/Window;", buf.items);
}

test "cross-class *Foo without registry errors" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    const foo = try makeTypeExpr(aa, "Window");
    const ptr = try makePointer(aa, foo);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    const result = desc.writeType(a, &buf, .{
        .enclosing_path = "android/view/View",
    }, ptr);
    try std.testing.expectError(desc.DeriveError.UnknownClassAlias, result);
}

test "cross-class *Foo with empty registry errors" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    var registry = desc.ClassRegistry.init(a);
    defer registry.deinit();

    const foo = try makeTypeExpr(aa, "WindowInsets");
    const ptr = try makePointer(aa, foo);

    var buf: std.ArrayList(u8) = .empty;
    defer buf.deinit(a);
    const result = desc.writeType(a, &buf, .{
        .enclosing_path = "android/view/Window",
        .classes = &registry,
    }, ptr);
    try std.testing.expectError(desc.DeriveError.UnknownClassAlias, result);
}

test "deriveMethod respects #jni_method_descriptor override verbatim" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // The actual sx signature `(self: *Self) -> s32` would derive to
    // `()I`. The override should win regardless.
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const ret = try makeTypeExpr(aa, "s32");

    const method: ast.ForeignMethodDecl = .{
        .name = "weirdMethod",
        .params = &.{self_ptr},
        .param_names = &.{"self"},
        .return_type = ret,
        .is_static = false,
        .jni_descriptor_override = "(Ljava/lang/Object;)Ljava/util/List;",
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "com/example/Foo" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("(Ljava/lang/Object;)Ljava/util/List;", out);
}

test "deriveMethod override bypasses unresolvable cross-class refs" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // The signature references `*UnknownClass` that isn't in the
    // registry — would normally fail with `UnknownClassAlias`. The
    // override short-circuits derivation, so it succeeds.
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const unknown = try makeTypeExpr(aa, "UnknownClass");
    const unknown_ptr = try makePointer(aa, unknown);

    const method: ast.ForeignMethodDecl = .{
        .name = "weirdMethod",
        .params = &.{self_ptr},
        .param_names = &.{"self"},
        .return_type = unknown_ptr,
        .is_static = false,
        .jni_descriptor_override = "()V",
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "com/example/Foo" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("()V", out);
}

test "deriveMethod chains *Foo returns and params" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    var registry = desc.ClassRegistry.init(a);
    defer registry.deinit();
    try registry.put("Window", "android/view/Window");
    try registry.put("View", "android/view/View");
    try registry.put("WindowInsets", "android/view/WindowInsets");

    // getDecorView :: (self: *Self) -> *View  →  ()Landroid/view/View;
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const view_te = try makeTypeExpr(aa, "View");
    const view_ptr = try makePointer(aa, view_te);

    const method: ast.ForeignMethodDecl = .{
        .name = "getDecorView",
        .params = &.{self_ptr},
        .param_names = &.{"self"},
        .return_type = view_ptr,
        .is_static = false,
    };
    const out = try desc.deriveMethod(a, .{
        .enclosing_path = "android/view/Window",
        .classes = &registry,
    }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("()Landroid/view/View;", out);
}

test "deriveMethod skips implicit self for instance methods" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // method: getId :: (self: *Self) -> s32  →  ()I
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const ret = try makeTypeExpr(aa, "s32");

    const method: ast.ForeignMethodDecl = .{
        .name = "getId",
        .params = &.{self_ptr},
        .param_names = &.{"self"},
        .return_type = ret,
        .is_static = false,
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "android/view/View" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("()I", out);
}

test "deriveMethod for static method emits all params" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // static abs :: (n: s32) -> s32  →  (I)I
    const n_ty = try makeTypeExpr(aa, "s32");
    const ret = try makeTypeExpr(aa, "s32");

    const method: ast.ForeignMethodDecl = .{
        .name = "abs",
        .params = &.{n_ty},
        .param_names = &.{"n"},
        .return_type = ret,
        .is_static = true,
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "java/lang/Math" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("(I)I", out);
}

test "deriveMethod with multiple primitive params and void return" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // setBounds :: (self: *Self, x: s32, y: s32, w: s32, h: s32) -> void  →  (IIII)V
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const s = try makeTypeExpr(aa, "s32");

    const method: ast.ForeignMethodDecl = .{
        .name = "setBounds",
        .params = &.{ self_ptr, s, s, s, s },
        .param_names = &.{ "self", "x", "y", "w", "h" },
        .return_type = null,
        .is_static = false,
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "android/graphics/Rect" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("(IIII)V", out);
}

test "deriveMethod with slice param" {
    const a = std.testing.allocator;
    var arena = std.heap.ArenaAllocator.init(a);
    defer arena.deinit();
    const aa = arena.allocator();

    // copy :: (self: *Self, src: []s8) -> s32  →  ([B)I
    const self_te = try makeTypeExpr(aa, "Self");
    const self_ptr = try makePointer(aa, self_te);
    const s8 = try makeTypeExpr(aa, "s8");
    const src_slice = try makeSlice(aa, s8);
    const ret = try makeTypeExpr(aa, "s32");

    const method: ast.ForeignMethodDecl = .{
        .name = "copy",
        .params = &.{ self_ptr, src_slice },
        .param_names = &.{ "self", "src" },
        .return_type = ret,
        .is_static = false,
    };
    const out = try desc.deriveMethod(a, .{ .enclosing_path = "java/nio/ByteBuffer" }, method);
    defer a.free(out);
    try std.testing.expectEqualStrings("([B)I", out);
}