From 4382ceeb01982b0f05c7ce2f7dcb3ee2cba3198b Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Tue, 5 Mar 2019 23:27:21 +0300
Subject: [PATCH] Add get_vote_percentage test.

GitOrigin-RevId: 87534944b1d1d49623dd5693bd78c295f519dbe7
---
 test/poll.cpp | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 test/poll.cpp

diff --git a/test/poll.cpp b/test/poll.cpp
new file mode 100644
index 000000000..fcbd8ed09
--- /dev/null
+++ b/test/poll.cpp
@@ -0,0 +1,57 @@
+//
+// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+#include "td/telegram/PollManager.h"
+
+#include "td/utils/logging.h"
+#include "td/utils/tests.h"
+
+REGISTER_TESTS(poll);
+
+static void check_vote_percentage(std::vector<td::int32> voter_counts, td::int32 total_count,
+                                  std::vector<td::int32> expected) {
+  auto result = td::PollManager::get_vote_percentage(voter_counts, total_count);
+  if (result != expected) {
+    LOG(FATAL) << "Have " << voter_counts << " and " << total_count << ", but received " << result << " instead of "
+               << expected;
+  }
+}
+
+TEST(Poll, get_vote_percentage) {
+  check_vote_percentage({1}, 1, {100});
+  check_vote_percentage({999}, 999, {100});
+  check_vote_percentage({0}, 0, {0});
+  check_vote_percentage({100, 100}, 200, {50, 50});
+  check_vote_percentage({101, 99}, 200, {50, 50});
+  check_vote_percentage({102, 98}, 200, {51, 49});
+  check_vote_percentage({198, 2}, 200, {99, 1});
+  check_vote_percentage({199, 1}, 200, {99, 1});
+  check_vote_percentage({200}, 200, {100});
+  check_vote_percentage({0, 999}, 999, {0, 100});
+  check_vote_percentage({999, 999}, 999, {100, 100});
+  check_vote_percentage({499, 599}, 999, {50, 60});
+  check_vote_percentage({1, 1}, 2, {50, 50});
+  check_vote_percentage({1, 1, 1}, 3, {33, 33, 33});
+  check_vote_percentage({1, 1, 1, 1}, 4, {25, 25, 25, 25});
+  check_vote_percentage({1, 1, 1, 1, 1}, 5, {20, 20, 20, 20, 20});
+  check_vote_percentage({1, 1, 1, 1, 1, 1}, 6, {16, 16, 16, 16, 16, 16});
+  check_vote_percentage({1, 1, 1, 1, 1, 1, 1}, 7, {14, 14, 14, 14, 14, 14, 14});
+  check_vote_percentage({1, 1, 1, 1, 1, 1, 1, 1}, 8, {12, 12, 12, 12, 12, 12, 12, 12});
+  check_vote_percentage({1, 1, 1, 1, 1, 1, 1, 1, 1}, 9, {11, 11, 11, 11, 11, 11, 11, 11, 11});
+  check_vote_percentage({1, 1, 1, 1, 1, 1, 2}, 8, {12, 12, 12, 12, 12, 12, 25});
+  check_vote_percentage({1, 1, 1, 2, 2, 2, 3}, 12, {8, 8, 8, 17, 17, 17, 25});
+  check_vote_percentage({0, 1, 1, 1, 2, 2, 2, 3}, 12, {0, 8, 8, 8, 17, 17, 17, 25});
+  check_vote_percentage({9949, 9950, 9999}, 10000, {99, 100, 100});
+  check_vote_percentage({1234, 2345, 3456, 2841}, 9876,
+                        {12 /* 12.49 */, 24 /* 23.74 */, 35 /* 34.99 */, 29 /* 28.76 */});
+  check_vote_percentage({1234, 2301, 3500, 2841}, 9876,
+                        {12 /* 12.49 */, 23 /* 23.29 */, 35 /* 35.43 */, 29 /* 28.76 */});
+  check_vote_percentage({200, 200, 200, 270, 270, 60}, 1200, {17, 17, 17, 22, 22, 5});
+  check_vote_percentage({200, 200, 200, 270, 270, 60}, 1200, {17, 17, 17, 22, 22, 5});
+  check_vote_percentage({200, 200, 200, 300, 240, 60}, 1200, {16, 16, 16, 25, 20, 5});
+  check_vote_percentage({200, 200, 200, 250, 250, 20}, 1120, {18, 18, 18, 22, 22, 2});
+  check_vote_percentage({200, 200, 200, 250, 250, 40}, 1140, {17, 17, 17, 22, 22, 4});
+};