sample8_unittest.cc 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. // Copyright 2008 Google Inc.
  2. // All Rights Reserved.
  3. //
  4. // Redistribution and use in source and binary forms, with or without
  5. // modification, are permitted provided that the following conditions are
  6. // met:
  7. //
  8. // * Redistributions of source code must retain the above copyright
  9. // notice, this list of conditions and the following disclaimer.
  10. // * Redistributions in binary form must reproduce the above
  11. // copyright notice, this list of conditions and the following disclaimer
  12. // in the documentation and/or other materials provided with the
  13. // distribution.
  14. // * Neither the name of Google Inc. nor the names of its
  15. // contributors may be used to endorse or promote products derived from
  16. // this software without specific prior written permission.
  17. //
  18. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29. // This sample shows how to test code relying on some global flag variables.
  30. // Combine() helps with generating all possible combinations of such flags,
  31. // and each test is given one combination as a parameter.
  32. // Use class definitions to test from this header.
  33. #include <tuple>
  34. #include "prime_tables.h"
  35. #include "gtest/gtest.h"
  36. namespace {
  37. // Suppose we want to introduce a new, improved implementation of PrimeTable
  38. // which combines speed of PrecalcPrimeTable and versatility of
  39. // OnTheFlyPrimeTable (see prime_tables.h). Inside it instantiates both
  40. // PrecalcPrimeTable and OnTheFlyPrimeTable and uses the one that is more
  41. // appropriate under the circumstances. But in low memory conditions, it can be
  42. // told to instantiate without PrecalcPrimeTable instance at all and use only
  43. // OnTheFlyPrimeTable.
  44. class HybridPrimeTable : public PrimeTable {
  45. public:
  46. HybridPrimeTable(bool force_on_the_fly, int max_precalculated)
  47. : on_the_fly_impl_(new OnTheFlyPrimeTable),
  48. precalc_impl_(force_on_the_fly
  49. ? nullptr
  50. : new PreCalculatedPrimeTable(max_precalculated)),
  51. max_precalculated_(max_precalculated) {}
  52. ~HybridPrimeTable() override {
  53. delete on_the_fly_impl_;
  54. delete precalc_impl_;
  55. }
  56. bool IsPrime(int n) const override {
  57. if (precalc_impl_ != nullptr && n < max_precalculated_)
  58. return precalc_impl_->IsPrime(n);
  59. else
  60. return on_the_fly_impl_->IsPrime(n);
  61. }
  62. int GetNextPrime(int p) const override {
  63. int next_prime = -1;
  64. if (precalc_impl_ != nullptr && p < max_precalculated_)
  65. next_prime = precalc_impl_->GetNextPrime(p);
  66. return next_prime != -1 ? next_prime : on_the_fly_impl_->GetNextPrime(p);
  67. }
  68. private:
  69. OnTheFlyPrimeTable* on_the_fly_impl_;
  70. PreCalculatedPrimeTable* precalc_impl_;
  71. int max_precalculated_;
  72. };
  73. using ::testing::Bool;
  74. using ::testing::Combine;
  75. using ::testing::TestWithParam;
  76. using ::testing::Values;
  77. // To test all code paths for HybridPrimeTable we must test it with numbers
  78. // both within and outside PreCalculatedPrimeTable's capacity and also with
  79. // PreCalculatedPrimeTable disabled. We do this by defining fixture which will
  80. // accept different combinations of parameters for instantiating a
  81. // HybridPrimeTable instance.
  82. class PrimeTableTest : public TestWithParam< ::std::tuple<bool, int> > {
  83. protected:
  84. void SetUp() override {
  85. bool force_on_the_fly;
  86. int max_precalculated;
  87. std::tie(force_on_the_fly, max_precalculated) = GetParam();
  88. table_ = new HybridPrimeTable(force_on_the_fly, max_precalculated);
  89. }
  90. void TearDown() override {
  91. delete table_;
  92. table_ = nullptr;
  93. }
  94. HybridPrimeTable* table_;
  95. };
  96. TEST_P(PrimeTableTest, ReturnsFalseForNonPrimes) {
  97. // Inside the test body, you can refer to the test parameter by GetParam().
  98. // In this case, the test parameter is a PrimeTable interface pointer which
  99. // we can use directly.
  100. // Please note that you can also save it in the fixture's SetUp() method
  101. // or constructor and use saved copy in the tests.
  102. EXPECT_FALSE(table_->IsPrime(-5));
  103. EXPECT_FALSE(table_->IsPrime(0));
  104. EXPECT_FALSE(table_->IsPrime(1));
  105. EXPECT_FALSE(table_->IsPrime(4));
  106. EXPECT_FALSE(table_->IsPrime(6));
  107. EXPECT_FALSE(table_->IsPrime(100));
  108. }
  109. TEST_P(PrimeTableTest, ReturnsTrueForPrimes) {
  110. EXPECT_TRUE(table_->IsPrime(2));
  111. EXPECT_TRUE(table_->IsPrime(3));
  112. EXPECT_TRUE(table_->IsPrime(5));
  113. EXPECT_TRUE(table_->IsPrime(7));
  114. EXPECT_TRUE(table_->IsPrime(11));
  115. EXPECT_TRUE(table_->IsPrime(131));
  116. }
  117. TEST_P(PrimeTableTest, CanGetNextPrime) {
  118. EXPECT_EQ(2, table_->GetNextPrime(0));
  119. EXPECT_EQ(3, table_->GetNextPrime(2));
  120. EXPECT_EQ(5, table_->GetNextPrime(3));
  121. EXPECT_EQ(7, table_->GetNextPrime(5));
  122. EXPECT_EQ(11, table_->GetNextPrime(7));
  123. EXPECT_EQ(131, table_->GetNextPrime(128));
  124. }
  125. // In order to run value-parameterized tests, you need to instantiate them,
  126. // or bind them to a list of values which will be used as test parameters.
  127. // You can instantiate them in a different translation module, or even
  128. // instantiate them several times.
  129. //
  130. // Here, we instantiate our tests with a list of parameters. We must combine
  131. // all variations of the boolean flag suppressing PrecalcPrimeTable and some
  132. // meaningful values for tests. We choose a small value (1), and a value that
  133. // will put some of the tested numbers beyond the capability of the
  134. // PrecalcPrimeTable instance and some inside it (10). Combine will produce all
  135. // possible combinations.
  136. INSTANTIATE_TEST_SUITE_P(MeaningfulTestParameters, PrimeTableTest,
  137. Combine(Bool(), Values(1, 10)));
  138. } // namespace