PVer4: RICE decode bytes to list of uint32 or 4-byte hash prefixes

components/safe_browsing_db/v4_rice_unittest.cc

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/safe_browsing_db/v4_rice.h"
+#include "testing/platform_test.h"
+
+namespace safe_browsing {
+
+class V4RiceTest : public PlatformTest {
+ public:
+  V4RiceTest() {}
+};
+
+TEST_F(V4RiceTest, TestDecoderGetNextWordWithNoData) {
+  uint32_t word;
+  V4RiceDecoder decoder(5, 1, "");
+  EXPECT_EQ(DECODE_RAN_OUT_OF_BITS_FAILURE, decoder.GetNextWord(&word));
+}
+
+TEST_F(V4RiceTest, TestDecoderGetNextBitsWithNoData) {
+  uint32_t word;
+  V4RiceDecoder decoder(5, 1, "");
+  EXPECT_EQ(DECODE_RAN_OUT_OF_BITS_FAILURE, decoder.GetNextBits(1, &word));
+}
+
+TEST_F(V4RiceTest, TestDecoderGetNextValueWithNoData) {
+  uint32_t word;
+  V4RiceDecoder decoder(5, 1, "");
+  EXPECT_EQ(DECODE_RAN_OUT_OF_BITS_FAILURE, decoder.GetNextValue(&word));
+}
+
+TEST_F(V4RiceTest, TestDecoderGetNextValueWithNoEntries) {
+  uint32_t word;
+  V4RiceDecoder decoder(28, 0, "\xbf\xa8");
+  EXPECT_FALSE(decoder.HasAnotherValue());
+  EXPECT_EQ(DECODE_NO_MORE_ENTRIES_FAILURE, decoder.GetNextValue(&word));
+}
+
+TEST_F(V4RiceTest, TestDecoderGetNextValueWithSmallValues) {
+  uint32_t word;
+  V4RiceDecoder decoder(2, 2, "\xf7\x2");
+  EXPECT_EQ(DECODE_SUCCESS, decoder.GetNextValue(&word));
+  EXPECT_EQ(15u, word);
+  EXPECT_EQ(DECODE_SUCCESS, decoder.GetNextValue(&word));
+  EXPECT_EQ(9u, word);
+  EXPECT_FALSE(decoder.HasAnotherValue());
+}
+
+TEST_F(V4RiceTest, TestDecoderGetNextValueWithLargeValues) {
+  uint32_t word;
+  V4RiceDecoder decoder(28, 3,
+                        "\xbf\xa8\x3f\xfb\xfc\xfb\x5e\x27\xe6\xc3\x1d\xc6\x38");
+  EXPECT_EQ(DECODE_SUCCESS, decoder.GetNextValue(&word));
+  EXPECT_EQ(1777762129u, word);
+  EXPECT_EQ(DECODE_SUCCESS, decoder.GetNextValue(&word));
+  EXPECT_EQ(2093280223u, word);
+  EXPECT_EQ(DECODE_SUCCESS, decoder.GetNextValue(&word));
+  EXPECT_EQ(924369848u, word);
+  EXPECT_FALSE(decoder.HasAnotherValue());
+}
+
+TEST_F(V4RiceTest, TestDecoderIntegersWithNoData) {
+  std::vector<uint32_t> out;
+  EXPECT_EQ(DECODE_RAN_OUT_OF_BITS_FAILURE,
+            V4RiceDecoder::DecodeIntegers(3, 5, 1, "", &out));
+}
+
+TEST_F(V4RiceTest, TestDecoderIntegersWithNullOut) {
+  EXPECT_EQ(DECODE_OUTPUT_IS_NULL_FAILURE,
+            V4RiceDecoder::DecodeIntegers(3, 5, 1, "", nullptr));
+}
+
+TEST_F(V4RiceTest, TestDecoderIntegersWithOneValue) {
+  std::vector<uint32_t> out;
+  EXPECT_EQ(DECODE_SUCCESS, V4RiceDecoder::DecodeIntegers(3, 2, 0, "", &out));
+  EXPECT_EQ(1u, out.size());
+  EXPECT_EQ(3u, out[0]);
+}
+
+TEST_F(V4RiceTest, TestDecoderIntegersWithMultipleValues) {
+  std::vector<uint32_t> out;
+  EXPECT_EQ(DECODE_SUCCESS,
+            V4RiceDecoder::DecodeIntegers(5, 2, 2, "\xf7\x2", &out));
+  EXPECT_EQ(3u, out.size());
+  EXPECT_EQ(5u, out[0]);
+  EXPECT_EQ(20u, out[1]);
+  EXPECT_EQ(29u, out[2]);
+}
+
+TEST_F(V4RiceTest, TestDecoderBytesWithNoData) {
+  std::string out;
+  EXPECT_EQ(DECODE_RAN_OUT_OF_BITS_FAILURE,
+            V4RiceDecoder::DecodeBytes(3, 5, 1, "", &out));
+}
+
+TEST_F(V4RiceTest, TestDecoderBytesWithNullOut) {
+  EXPECT_EQ(DECODE_OUTPUT_IS_NULL_FAILURE,
+            V4RiceDecoder::DecodeBytes(3, 5, 1, "", nullptr));
+}
+
+TEST_F(V4RiceTest, TestDecoderBytesWithOneValue) {
+  std::string out;
+  EXPECT_EQ(DECODE_SUCCESS,
+            V4RiceDecoder::DecodeBytes(0x69F67F51u, 2, 0, "", &out));
+  EXPECT_EQ(4u, out.size());
+  EXPECT_EQ("Q\x7F\xF6i", out);
+}
+
+TEST_F(V4RiceTest, TestDecoderBytesWithOneSmallValue) {
+  std::string out;
+  EXPECT_EQ(DECODE_SUCCESS, V4RiceDecoder::DecodeBytes(17u, 2, 0, "", &out));
+  EXPECT_EQ(1u, out.size());
+  EXPECT_EQ("\x11", out);
+}
+
+TEST_F(V4RiceTest, TestDecoderBytesWithMultipleValues) {
+  std::string out;
+  EXPECT_EQ(DECODE_SUCCESS,
+            V4RiceDecoder::DecodeBytes(
+                5, 28, 3,
+                "\xbf\xa8\x3f\xfb\xfc\xfb\x5e\x27\xe6\xc3\x1d\xc6\x38", &out));
+  EXPECT_EQ(13u, out.size());
+  EXPECT_EQ("\x5V\x7F\xF6i5k\xBB\xE6\xED.\xD4\x1D", out);
+}
+
+}  // namespace safe_browsing