|
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