context_stack_trace.test.cpp

Go to the documentation of this file.

#include <gmock/gmock.h>
#include <gtest/gtest.h>
 
#include "barretenberg/vm2/common/field.hpp"
#include "barretenberg/vm2/testing/macros.hpp"
#include "barretenberg/vm2/tracegen/context_stack_trace.hpp"
#include "barretenberg/vm2/tracegen/test_trace_container.hpp"
 
namespace bb::avm2::tracegen {
namespace {
 
using testing::ElementsAre;
using testing::Field;
 
using R = TestTraceContainer::Row;
 
TEST(ContextStackTraceGenTest, TraceGenerationSnapshot)
{
    TestTraceContainer trace;
    ContextStackTraceBuilder builder;
 
    TreeStates tree_states = TreeStates{
        .note_hash_tree = {
            .tree = {
                .root = 10,
                .next_available_leaf_index = 9,
            },
            .counter = 8,
        },
        .nullifier_tree = {
            .tree = {
                .root = 7,
                .next_available_leaf_index = 6,
            },
            .counter = 5,
        },
        .l1_to_l2_message_tree = {
            .tree = {
                .root = 4,
                .next_available_leaf_index = 3,
            },
            .counter = 0,
        },
        .public_data_tree = {
            .tree = {
                .root = 2,
                .next_available_leaf_index = 1,
            },
            .counter = 1,
        }
    };
 
    AppendOnlyTreeSnapshot written_public_data_slots_tree_snapshot = AppendOnlyTreeSnapshot{
        .root = 0x12345678,
        .next_available_leaf_index = 10,
    };
 
    builder.process({ {
                        .id = 1,
                        .parent_id = 0,
                        .next_pc = 20,
                        .msg_sender = 30,
                        .contract_addr = 40,
                        .is_static = false,
                        .tree_states = tree_states,
                        .written_public_data_slots_tree_snapshot = written_public_data_slots_tree_snapshot,
                        .numUnencryptedLogFields = 1,
                        .numL2ToL1Messages = 2,
                    } },
                    trace);
 
    EXPECT_THAT(trace.as_rows(),
                ElementsAre(
                    // Only one row.
                    AllOf(ROW_FIELD_EQ(context_stack_context_id, 1),
                          ROW_FIELD_EQ(context_stack_parent_id, 0),
                          ROW_FIELD_EQ(context_stack_next_pc, 20),
                          ROW_FIELD_EQ(context_stack_msg_sender, 30),
                          ROW_FIELD_EQ(context_stack_contract_address, 40),
                          ROW_FIELD_EQ(context_stack_is_static, false),
                          ROW_FIELD_EQ(context_stack_note_hash_tree_root, tree_states.note_hash_tree.tree.root),
                          ROW_FIELD_EQ(context_stack_note_hash_tree_size,
                                       tree_states.note_hash_tree.tree.next_available_leaf_index),
                          ROW_FIELD_EQ(context_stack_num_note_hashes_emitted, tree_states.note_hash_tree.counter),
                          ROW_FIELD_EQ(context_stack_nullifier_tree_root, tree_states.nullifier_tree.tree.root),
                          ROW_FIELD_EQ(context_stack_nullifier_tree_size,
                                       tree_states.nullifier_tree.tree.next_available_leaf_index),
                          ROW_FIELD_EQ(context_stack_num_nullifiers_emitted, tree_states.nullifier_tree.counter),
                          ROW_FIELD_EQ(context_stack_public_data_tree_root, tree_states.public_data_tree.tree.root),
                          ROW_FIELD_EQ(context_stack_public_data_tree_size,
                                       tree_states.public_data_tree.tree.next_available_leaf_index),
                          ROW_FIELD_EQ(context_stack_written_public_data_slots_tree_root,
                                       written_public_data_slots_tree_snapshot.root),
                          ROW_FIELD_EQ(context_stack_written_public_data_slots_tree_size,
                                       written_public_data_slots_tree_snapshot.next_available_leaf_index),
                          ROW_FIELD_EQ(context_stack_num_unencrypted_log_fields, 1),
                          ROW_FIELD_EQ(context_stack_num_l2_to_l1_messages, 2))));
}
 
} // namespace
} // namespace bb::avm2::tracegen