gasm-sdk/cmd/gasm/audit_test.go

// Copyright (c) 2026 Petr Balvín <opensource@petrbalvin.org> (https://petrbalvin.org)
// SPDX-License-Identifier: BSD-3-Clause

package main

import (
	"testing"

	"sourcedock.dev/petrbalvin/gasm-devkit/arch"
)

func TestDerivedFamily(t *testing.T) {
	for _, n := range []string{"JEQ", "JLT", "JCC", "CMOVLGT", "SETNE", "SETA"} {
		if !derivedFamily(n) {
			t.Errorf("derivedFamily(%q) = false, want true", n)
		}
	}
	for _, n := range []string{"JMP", "ADDQ", "VPGATHERDD", "MOVBE", "PSHUFB"} {
		if derivedFamily(n) {
			t.Errorf("derivedFamily(%q) = true, want false", n)
		}
	}
}

func TestSanitize(t *testing.T) {
	if got := sanitize("VPCMP.UB"); got != "VPCMP_UB" {
		t.Errorf("sanitize: got %q", got)
	}
}

func TestAuditArch(t *testing.T) {
	for in, want := range map[string]arch.Arch{
		"amd64": arch.AMD64, "arm64": arch.ARM64,
		"riscv64": arch.RISCV, "riscv": arch.RISCV,
		"loong64": arch.LOONG64, "LOONG": arch.LOONG64,
	} {
		got, err := auditArch(in)
		if err != nil || got != want {
			t.Errorf("auditArch(%q) = %v, %v; want %v", in, got, err, want)
		}
	}
	if _, err := auditArch("mips"); err == nil {
		t.Error("auditArch(mips) must fail")
	}
}

func TestGasmEncodable(t *testing.T) {
	cases := []struct {
		a   arch.Arch
		yes string
		no  string
	}{
		{arch.AMD64, "ADDQ", "NOSUCHMNEMONIC"},
		{arch.ARM64, "ADD", "NOSUCHMNEMONIC"},
		{arch.RISCV, "ADD", "NOSUCHMNEMONIC"},
		{arch.LOONG64, "ADDV", "NOSUCHMNEMONIC"},
	}
	for _, c := range cases {
		if !gasmEncodable(c.a, c.yes) {
			t.Errorf("%s: %s should be encodable", c.a, c.yes)
		}
		if gasmEncodable(c.a, c.no) {
			t.Errorf("%s: %s should not be encodable", c.a, c.no)
		}
	}
}