folly/compression/test/CountersTest.cpp

   1 /*
   2  * Copyright 2018-present Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <folly/compression/Counters.h>
  18 #include <folly/compression/Compression.h>
  19 #include <folly/portability/GTest.h>
  20
  21 using ::folly::CompressionCounterType;
  22 using ::folly::detail::CompressionCounter;
  23
  24 namespace {
  25 constexpr auto kCodecType = folly::io::CodecType::USER_DEFINED;
  26 constexpr folly::StringPiece kCodecName = "test";
  27 constexpr auto kKey = folly::CompressionCounterKey::BYTES_AFTER_COMPRESSION;
  28 } // namespace
  29
  30 TEST(FollyCountersTest, HasImplementation) {
  31   CompressionCounter counter(
  32       kCodecType, kCodecName, folly::none, kKey, CompressionCounterType::SUM);
  33   EXPECT_FALSE(counter.hasImplementation());
  34 }
  35
  36 TEST(FollyCountersTest, SumWorks) {
  37   CompressionCounter counter(
  38       kCodecType, kCodecName, folly::none, kKey, CompressionCounterType::SUM);
  39   for (int i = 0; i < 100; ++i) {
  40     ++counter;
  41     counter++;
  42   }
  43 }
  44
  45 TEST(FollyCountersTest, AvgWorks) {
  46   CompressionCounter counter(
  47       kCodecType,
  48       kCodecName,
  49       folly::none,
  50       kKey,
  51       CompressionCounterType::AVG);
  52   for (int i = 0; i < 100; ++i) {
  53     counter += 5;
  54   }
  55 }