From 92b4455c588fb722887e8fdbefde269d6e6e7cb0 Mon Sep 17 00:00:00 2001
From: Dimitrios Siganos <dimitris@siganos.org>
Date: Sat, 13 Apr 2024 02:09:27 +0900
Subject: [PATCH] Test case for pending table query at different epochs

---
 nano/core_test/CMakeLists.txt |  1 +
 nano/core_test/receivable.cpp | 99 +++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 nano/core_test/receivable.cpp

diff --git a/nano/core_test/CMakeLists.txt b/nano/core_test/CMakeLists.txt
index 793c7c26..d78b56c2 100644
--- a/nano/core_test/CMakeLists.txt
+++ b/nano/core_test/CMakeLists.txt
@@ -39,6 +39,7 @@ add_executable(
   processing_queue.cpp
   processor_service.cpp
   rep_crawler.cpp
+  receivable.cpp
   peer_container.cpp
   rep_weight_store.cpp
   scheduler_buckets.cpp
diff --git a/nano/core_test/receivable.cpp b/nano/core_test/receivable.cpp
new file mode 100644
index 00000000..aef5534f
--- /dev/null
+++ b/nano/core_test/receivable.cpp
@@ -0,0 +1,99 @@
+#include <nano/test_common/system.hpp>
+#include <nano/test_common/testutil.hpp>
+
+#include <gtest/gtest.h>
+
+using namespace std::chrono_literals;
+
+// this test sends 3 send blocks in 3 different epochs and checks that
+// the pending table records the epochs correctly for each send
+TEST (receivable, pending_table_query_epochs)
+{
+	nano::test::system system{ 1 };
+	auto & node = *system.nodes[0];
+	nano::keypair key2;
+	nano::block_builder builder;
+
+	// epoch 0 send
+	auto send0 = builder
+				 .send ()
+				 .previous (nano::dev::genesis->hash ())
+				 .destination (key2.pub)
+				 .balance (nano::dev::constants.genesis_amount - 1)
+				 .sign (nano::dev::genesis_key.prv, nano::dev::genesis_key.pub)
+				 .work (*system.work.generate (nano::dev::genesis->hash ()))
+				 .build ();
+	nano::test::process (node, { send0 });
+	ASSERT_TIMELY (5s, nano::test::exists (node, { send0 }));
+
+	auto epoch1 = system.upgrade_genesis_epoch (node, nano::epoch::epoch_1);
+	ASSERT_TRUE (epoch1);
+	ASSERT_TIMELY (5s, nano::test::exists (node, { epoch1 }));
+
+	// epoch 1 send
+	auto send1 = builder
+				 .state ()
+				 .account (nano::dev::genesis_key.pub)
+				 .representative (nano::dev::genesis_key.pub)
+				 .previous (epoch1->hash ())
+				 .link (key2.pub)
+				 .balance (nano::dev::constants.genesis_amount - 11)
+				 .sign (nano::dev::genesis_key.prv, nano::dev::genesis_key.pub)
+				 .work (*system.work.generate (epoch1->hash ()))
+				 .build ();
+	ASSERT_TRUE (nano::test::process (node, { send1 }));
+	ASSERT_TIMELY (5s, nano::test::exists (node, { send1 }));
+
+	auto epoch2 = system.upgrade_genesis_epoch (node, nano::epoch::epoch_2);
+	ASSERT_TRUE (epoch2);
+	ASSERT_TIMELY (5s, nano::test::exists (node, { epoch2 }));
+
+	// epoch 2 send
+	auto send2 = builder
+				 .state ()
+				 .account (nano::dev::genesis_key.pub)
+				 .representative (nano::dev::genesis_key.pub)
+				 .previous (epoch2->hash ())
+				 .link (key2.pub)
+				 .balance (nano::dev::constants.genesis_amount - 111)
+				 .sign (nano::dev::genesis_key.prv, nano::dev::genesis_key.pub)
+				 .work (*system.work.generate (epoch2->hash ()))
+				 .build ();
+	nano::test::process (node, { send2 });
+	ASSERT_TIMELY (5s, nano::test::exists (node, { send2 }));
+
+	auto tx = node.store.tx_begin_read ();
+
+	// check epoch 0 send
+	{
+		nano::pending_key key{ key2.pub, send0->hash () };
+		auto opt_info = node.store.pending.get (tx, key);
+		ASSERT_TRUE (opt_info.has_value ());
+		auto info = opt_info.value ();
+		ASSERT_EQ (info.source, nano::dev::genesis_key.pub);
+		ASSERT_EQ (info.amount, 1);
+		ASSERT_EQ (info.epoch, nano::epoch::epoch_0);
+	}
+
+	// check epoch 1 send
+	{
+		nano::pending_key key{ key2.pub, send1->hash () };
+		auto opt_info = node.store.pending.get (tx, key);
+		ASSERT_TRUE (opt_info.has_value ());
+		auto info = opt_info.value ();
+		ASSERT_EQ (info.source, nano::dev::genesis_key.pub);
+		ASSERT_EQ (info.amount, 10);
+		ASSERT_EQ (info.epoch, nano::epoch::epoch_1);
+	}
+
+	// check epoch 2 send
+	{
+		nano::pending_key key{ key2.pub, send2->hash () };
+		auto opt_info = node.store.pending.get (tx, key);
+		ASSERT_TRUE (opt_info.has_value ());
+		auto info = opt_info.value ();
+		ASSERT_EQ (info.source, nano::dev::genesis_key.pub);
+		ASSERT_EQ (info.amount, 100);
+		ASSERT_EQ (info.epoch, nano::epoch::epoch_2);
+	}
+}
\ No newline at end of file