1 files changed, 119 insertions, 0 deletions
diff --git a/tests/unit_tests/decompose_amount_into_digits.cpp b/tests/unit_tests/decompose_amount_into_digits.cpp
new file mode 100644
index 000000000..d7c66cc73
--- /dev/null
+++ b/tests/unit_tests/decompose_amount_into_digits.cpp
@@ -0,0 +1,119 @@
+// Copyright (c) 2012-2013 The Cryptonote developers
+// Distributed under the MIT/X11 software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#include "gtest/gtest.h"
+
+#include <cstdint>
+#include <vector>
+
+#include "cryptonote_core/cryptonote_format_utils.h"
+
+#define VEC_FROM_ARR(vec)                                               \
+  std::vector<uint64_t> vec;                                            \
+  for (size_t i = 0; i < sizeof(vec##_arr) / sizeof(vec##_arr[0]); ++i) \
+  {                                                                     \
+    vec.push_back(vec##_arr[i]);                                        \
+  }
+
+namespace
+{
+  struct chunk_handler_t
+  {
+    void operator()(uint64_t chunk) const
+    {
+      m_chunks.push_back(chunk);
+    }
+
+    mutable std::vector<uint64_t> m_chunks;
+  };
+
+  struct dust_handler_t
+  {
+    dust_handler_t()
+      : m_dust(0)
+      , m_has_dust(false)
+    {
+    }
+
+    void operator()(uint64_t dust) const
+    {
+      m_dust = dust;
+      m_has_dust = true;
+    }
+
+    mutable uint64_t m_dust;
+    mutable bool m_has_dust;
+  };
+
+  class decompose_amount_into_digits_test : public ::testing::Test
+  {
+  protected:
+    chunk_handler_t m_chunk_handler;
+    dust_handler_t m_dust_handler;
+  };
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_0)
+{
+  uint64_t expected_chunks_arr[] = {0};
+  VEC_FROM_ARR(expected_chunks);
+  cryptonote::decompose_amount_into_digits(0, 0, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_has_dust, false);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_1)
+{
+  uint64_t expected_chunks_arr[] = {0};
+  VEC_FROM_ARR(expected_chunks);
+  cryptonote::decompose_amount_into_digits(0, 10, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_has_dust, false);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_2)
+{
+  uint64_t expected_chunks_arr[] = {10};
+  VEC_FROM_ARR(expected_chunks);
+  cryptonote::decompose_amount_into_digits(10, 0, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_has_dust, false);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_3)
+{
+  std::vector<uint64_t> expected_chunks;
+  uint64_t expected_dust = 10;
+  cryptonote::decompose_amount_into_digits(10, 10, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_dust, expected_dust);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_4)
+{
+  uint64_t expected_dust = 8100;
+  std::vector<uint64_t> expected_chunks;
+  cryptonote::decompose_amount_into_digits(8100, 1000000, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_dust, expected_dust);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_5)
+{
+  uint64_t expected_chunks_arr[] = {100, 900000, 8000000};
+  VEC_FROM_ARR(expected_chunks);
+  cryptonote::decompose_amount_into_digits(8900100, 10, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_has_dust, false);
+}
+
+TEST_F(decompose_amount_into_digits_test, is_correct_6)
+{
+  uint64_t expected_chunks_arr[] = {900000, 8000000};
+  VEC_FROM_ARR(expected_chunks);
+  uint64_t expected_dust = 100;
+  cryptonote::decompose_amount_into_digits(8900100, 1000, m_chunk_handler, m_dust_handler);
+  ASSERT_EQ(m_chunk_handler.m_chunks, expected_chunks);
+  ASSERT_EQ(m_dust_handler.m_dust, expected_dust);
+}