|
1 | 1 | use solana_sbpf::{ |
2 | 2 | elf::Executable, |
3 | | - program::BuiltinProgram, |
4 | | - vm::{RuntimeEnvironmentSlot, TestContextObject}, |
| 3 | + program::{BuiltinProgram, BuiltinFunction, FunctionRegistry}, |
| 4 | + syscalls, |
| 5 | + vm::{RuntimeEnvironmentSlot, TestContextObject, Config}, |
5 | 6 | }; |
6 | 7 | use std::{fs::File, io::Read, sync::Arc}; |
7 | 8 | use test_utils::create_vm; |
@@ -49,3 +50,30 @@ fn test_runtime_environment_slots() { |
49 | 50 | check_slot!(env, program_result, ProgramResult); |
50 | 51 | check_slot!(env, memory_mapping, MemoryMapping); |
51 | 52 | } |
| 53 | + |
| 54 | +#[test] |
| 55 | +fn test_builtin_program_eq() { |
| 56 | + let mut function_registry_a = FunctionRegistry::<BuiltinFunction<TestContextObject>>::default(); |
| 57 | + function_registry_a |
| 58 | + .register_function_hashed(*b"log", syscalls::SyscallString::vm) |
| 59 | + .unwrap(); |
| 60 | + function_registry_a |
| 61 | + .register_function_hashed(*b"log_64", syscalls::SyscallU64::vm) |
| 62 | + .unwrap(); |
| 63 | + let mut function_registry_b = FunctionRegistry::<BuiltinFunction<TestContextObject>>::default(); |
| 64 | + function_registry_b |
| 65 | + .register_function_hashed(*b"log_64", syscalls::SyscallU64::vm) |
| 66 | + .unwrap(); |
| 67 | + function_registry_b |
| 68 | + .register_function_hashed(*b"log", syscalls::SyscallString::vm) |
| 69 | + .unwrap(); |
| 70 | + let mut function_registry_c = FunctionRegistry::<BuiltinFunction<TestContextObject>>::default(); |
| 71 | + function_registry_c |
| 72 | + .register_function_hashed(*b"log_64", syscalls::SyscallU64::vm) |
| 73 | + .unwrap(); |
| 74 | + let builtin_program_a = BuiltinProgram::new_loader(Config::default(), function_registry_a); |
| 75 | + let builtin_program_b = BuiltinProgram::new_loader(Config::default(), function_registry_b); |
| 76 | + assert_eq!(builtin_program_a, builtin_program_b); |
| 77 | + let builtin_program_c = BuiltinProgram::new_loader(Config::default(), function_registry_c); |
| 78 | + assert_ne!(builtin_program_a, builtin_program_c); |
| 79 | +} |
0 commit comments