RustNotMC/tools/test_inventory_pickup_parity.py

#!/usr/bin/env python3
"""Compare native InventoryPickup transfers, shared stack identity and change callbacks."""
import argparse
import json
import os
from pathlib import Path
import platform
import subprocess
import random
import struct
import math
try:
    from . import lab, test_rust_parity as base
except ImportError:
    import lab
    import test_rust_parity as base


def corpus():
    edges=[-2147483648,-1,0,1,2,64,2147483647]
    def op(kind,**kw):
        v=dict(kind=kind,inventory=0,other=1,a=0,b=3,slot=-1,amount=1,all=False,equipment=0,key=0,value=64,max=99,infinite=False,fail_at=0,mutate_at=0,mutate_ref=0,after_count=2,after_max=8,mutate_slot=1,replace_ref=2,after_selected=-1);v.update(kw);return v
    def stack(item,count,patch=None):return dict(item=item,count=count,pop=7,patch=patch or [])
    def row(ops,count=5,slots=None,damaged=False):return dict(prototypes=[[],[[0,64]],[[0,1],[1,0],[2,100]]],stacks=[stack(2 if damaged else 1,count,[[1,3]] if damaged else []),stack(1,60),stack(2,1),stack(-1,0)],equipment_order=list(range(7)),shared_equipment=False,inventories=[dict(selected=0,changed=2147483647,mainhand=0,slots=slots if slots is not None else [[0,1],[35,2],[40,1]]),dict(selected=8,changed=0,mainhand=-1,slots=[[2,1],[39,0]])],ops=ops)
    for damaged in [False,True]:
        for slot in [-2,-1,0,8,35,36,40,42,43,2147483647]:
            for count in edges:
                for infinite in [False,True]:
                    # Unknown positive slots discard newly created copies on
                    # every iteration; bound counts to avoid billions of steps.
                    if slot>=43 and count>3:continue
                    yield row([op('space'),op('add',slot=slot,infinite=infinite),op('space')],count=count,damaged=damaged)
    for damaged in [False,True]:
        for slots in [[],[[0,0]],[[i,1] for i in range(43)],[[0,1],[40,1]],[[i,2] for i in range(43)]]:
            for fail in range(5):
                for mutate in range(4):
                    for after in [-1,0,4]:
                        yield row([op('add',fail_at=fail,mutate_at=mutate,after_selected=after)],slots=slots,damaged=damaged)
    for limit in edges:
        for count in edges:
            for size in [0,1,64,2147483647]:
                v=row([op('space',max=limit),op('add',max=limit)],count=count)
                v['prototypes'][1]=[[0,size]]
                yield v
    rng=random.Random(0x5049434b5550)
    for _ in range(1000):
        damaged=bool(rng.randrange(2))
        v=row([],count=rng.choice(edges),slots=[[i,rng.randrange(4)] for i in rng.sample(range(43),rng.randrange(44))],damaged=damaged)
        v['shared_equipment']=bool(rng.randrange(2))
        # Restrict insertion to actual slots; counts can cover the entire int.
        for __ in range(10):
            kind=rng.choice(['add','add','space','set','select','clear','count','alias','copy','remove'])
            v['ops'].append(op(kind,inventory=rng.randrange(2),a=rng.randrange(4),b=rng.randrange(4),slot=rng.choice([-2,-1,0,8,35,36,40,42]),amount=rng.choice(edges),max=rng.choice(edges+[99]),infinite=bool(rng.randrange(2)),fail_at=rng.randrange(5),mutate_at=rng.randrange(4),mutate_ref=rng.randrange(4),after_count=rng.choice(edges),after_max=rng.choice(edges),mutate_slot=rng.choice([-1,0,8,40]),replace_ref=rng.randrange(4),after_selected=rng.choice([-1,0,8])))
        yield v


def main():
    parser=argparse.ArgumentParser(description=__doc__);parser.add_argument('--prepare',action='store_true');args=parser.parse_args()
    build=lab.LOCAL/'rust-inventory-pickup-parity';build.mkdir(parents=True,exist_ok=True)
    report_path=build/'report.json';report_path.unlink(missing_ok=True)
    if args.prepare:lab.prepare()
    lock,jars,libraries=base.verify_artifacts()
    oracle=lab.ROOT/'mods/horizon/src/test/java/dev/horizonlab/parity/VanillaInventoryPickupOracle.java'
    sources=[oracle,Path(__file__),lab.ROOT/'tools/test_rust_parity.py',lab.ROOT/'tools/lab.py',lab.ROOT/'Cargo.toml',lab.ROOT/'Cargo.lock',lab.ROOT/'crates/horizon-core/Cargo.toml',lab.ROOT/'crates/horizon-world/Cargo.toml',*sorted((lab.ROOT/'crates/horizon-core/src').rglob('*.rs')),*sorted((lab.ROOT/'crates/horizon-world/src').rglob('*.rs')),lab.ROOT/'crates/horizon-world/examples/probe/inventory_pickup_fixture.rs']
    tested_sources=base.source_hashes(sources)
    classes=build/'classes';classes.mkdir(exist_ok=True)
    def run(command,name):
        with (build/(name+'.log')).open('w',encoding='utf-8') as log:base.run(list(map(str,command)),stdout=log,stderr=subprocess.STDOUT)
    dependency_path=os.pathsep.join(map(str,libraries))
    run(['javac','--release',lock['minecraft']['java_major'],'-proc:none','-sourcepath',str(lab.ROOT/'mods/horizon/src/test/java'),'-cp',str(jars['server'])+os.pathsep+dependency_path,'-d',classes,oracle],'compile-java')
    def java(side,mode,path):
        run(['java','-Xmx512M','-cp',os.pathsep.join([str(classes),str(jars[side]),dependency_path]),'dev.horizonlab.parity.VanillaInventoryPickupOracle',mode,path],side+'-'+mode)
    fixtures=list(corpus());inputs=build/'inputs.txt';inputs.write_text(''.join(json.dumps(value,separators=(',',':'))+'\n' for value in fixtures),encoding='utf-8')
    fixture_paths=[inputs];tested_fixtures=base.source_hashes(fixture_paths)
    run(['cargo','build','--release','-p','horizon-world','--example','world_probe'],'compile-rust')
    metadata=json.loads(subprocess.check_output(['cargo','metadata','--no-deps','--format-version','1'],cwd=lab.ROOT,text=True))
    binary=Path(metadata['target_directory'])/'release/examples'/('world_probe.exe' if os.name=='nt' else 'world_probe')
    run([binary,'inventory_pickup_fixture',build],'rust');references=[]
    for side,jar in jars.items():
        java(side,side,build);reference=build/f'{side}.txt';matched=base.compare(reference,build/'rust.txt')
        references.append({'side':side,'jar_sha256':lab.digest(jar,'sha256'),'output_sha256':lab.digest(reference,'sha256'),'matched_output_lines':matched})
        print(f'{side}: {matched:,} inventory pickup output lines match Rust',flush=True)
    base.ensure_sources_unchanged(sources,tested_sources);base.ensure_sources_unchanged(fixture_paths,tested_fixtures)
    report={'schema':1,'minecraft':lock['minecraft']['version'],'manifest_sha1':lock['minecraft']['manifest_sha1'],'platform':platform.platform(),'java':subprocess.check_output(['java','--version'],text=True).splitlines()[0],'rustc':subprocess.check_output(['rustc','--version'],text=True).strip(),'fixtures':len(fixtures),'sources_sha256':tested_sources,'fixtures_sha256':tested_fixtures,'rust_output_sha256':lab.digest(build/'rust.txt','sha256'),'references':references,'scope':'Actual Inventory.add and getSlotWithRemainingSpace over stored ItemStack and shared equipment. Traces cover selected/offhand/ordinary slot search, component-sensitive stacking, damaged copy-and-clear, source aliases, pop time, creative fallback and virtual size/material callbacks with mutations/failures. Native crash-report wrapping is represented as an error without rendering its diagnostic text. Item entity pickup events, networking, menus, arbitrary method overrides, full item components and complete gameplay integration remain pending.'}
    lab.write_json(report_path,report);print(f'Scoped inventory pickup parity passed: {report_path}')


if __name__=='__main__':main()