#include <gtest/gtest.h>

#include "jianzhi_offer.h"

TEST(NodeListTest, BasicAssertions) {
    yaha::ListNode *node = new yaha::ListNode(1);
    node->next = new yaha::ListNode(3);
    node->next->next = new yaha::ListNode(2);
    auto res = node->reverse_print(node);
    ASSERT_EQ(2, res[0]);
    ASSERT_EQ(3, res[1]);
    ASSERT_EQ(1, res[2]);
}

TEST(NodeListStackTest, BasicAssertions) {
    yaha::ListNode *node = new yaha::ListNode(1);
    node->next = new yaha::ListNode(3);
    node->next->next = new yaha::ListNode(2);
    std::vector<int> res;
    node->reverse_stack(node, res);
    ASSERT_EQ(2, res[0]);
    ASSERT_EQ(3, res[1]);
    ASSERT_EQ(1, res[2]);
}