X-Git-Url: http://plrg.eecs.uci.edu/git/?p=folly.git;a=blobdiff_plain;f=folly%2Fcompression%2Ftest%2FCompressionTest.cpp;h=9483979a7b99275b4939f97f51649ef8acb44209;hp=a1725765d067ccce0ac7d674206a3ede7ba186f5;hb=59cb2ab00419bffb627e89cb096e1288c2f37eeb;hpb=0c20289b724cf462078cf70d0ea1487f67b45fa8

diff --git a/folly/compression/test/CompressionTest.cpp b/folly/compression/test/CompressionTest.cpp
index a1725765..9483979a 100644
--- a/folly/compression/test/CompressionTest.cpp
+++ b/folly/compression/test/CompressionTest.cpp
@@ -26,6 +26,7 @@
 #include <boost/noncopyable.hpp>
 #include <glog/logging.h>
 
+#include <folly/Benchmark.h>
 #include <folly/Memory.h>
 #include <folly/Random.h>
 #include <folly/Varint.h>
@@ -1488,3 +1489,14 @@ INSTANTIATE_TEST_CASE_P(
 } // namespace test
 } // namespace io
 } // namespace folly
+
+int main(int argc, char *argv[]) {
+  testing::InitGoogleTest(&argc, argv);
+  gflags::ParseCommandLineFlags(&argc, &argv, true);
+
+  auto ret = RUN_ALL_TESTS();
+  if (!ret) {
+    folly::runBenchmarksOnFlag();
+  }
+  return ret;
+}