Switch to standard integer types in chrome/.

chrome/renderer/safe_browsing/phishing_term_feature_extractor_unittest.cc

 // Copyright (c) 2012 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 "chrome/renderer/safe_browsing/phishing_term_feature_extractor.h"
 
+#include <stddef.h>
+#include <stdint.h>
+
 #include <string>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/containers/hash_tables.h"
 #include "base/location.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string16.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
+#include "build/build_config.h"
 #include "chrome/renderer/safe_browsing/features.h"
 #include "chrome/renderer/safe_browsing/mock_feature_extractor_clock.h"
 #include "chrome/renderer/safe_browsing/murmurhash3_util.h"
 #include "chrome/renderer/safe_browsing/test_utils.h"
 #include "crypto/sha2.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::ASCIIToUTF16;
 using ::testing::Return;
 
-
-static const uint32 kMurmurHash3Seed = 2777808611U;
+static const uint32_t kMurmurHash3Seed = 2777808611U;
 
 namespace safe_browsing {
 
 class PhishingTermFeatureExtractorTest : public ::testing::Test {
  protected:
   void SetUp() override {
     base::hash_set<std::string> terms;
     terms.insert("one");
     terms.insert("one one");
     terms.insert("two");
@@ skipping 41 matching lines @@
         kMurmurHash3Seed,
         max_shingles_per_page,
         4 /* shingle_size */,
         &clock_));
   }
 
   // Runs the TermFeatureExtractor on |page_text|, waiting for the
   // completion callback.  Returns the success boolean from the callback.
   bool ExtractFeatures(const base::string16* page_text,
                        FeatureMap* features,
-                       std::set<uint32>* shingle_hashes) {
+                       std::set<uint32_t>* shingle_hashes) {
     success_ = false;
     extractor_->ExtractFeatures(
         page_text,
         features,
         shingle_hashes,
         base::Bind(&PhishingTermFeatureExtractorTest::ExtractionDone,
                    base::Unretained(this)));
     msg_loop_.Run();
     return success_;
   }
 
   void PartialExtractFeatures(const base::string16* page_text,
                               FeatureMap* features,
-                              std::set<uint32>* shingle_hashes) {
+                              std::set<uint32_t>* shingle_hashes) {
     extractor_->ExtractFeatures(
         page_text,
         features,
         shingle_hashes,
         base::Bind(&PhishingTermFeatureExtractorTest::ExtractionDone,
                    base::Unretained(this)));
     msg_loop_.task_runner()->PostTask(
         FROM_HERE, base::Bind(&PhishingTermFeatureExtractorTest::QuitExtraction,
                               base::Unretained(this)));
     msg_loop_.RunUntilIdle();
   }
 
   // Completion callback for feature extraction.
   void ExtractionDone(bool success) {
     success_ = success;
     msg_loop_.QuitWhenIdle();
   }
 
   void QuitExtraction() {
     extractor_->CancelPendingExtraction();
     msg_loop_.QuitWhenIdle();
   }
 
   base::MessageLoop msg_loop_;
   MockFeatureExtractorClock clock_;
   scoped_ptr<PhishingTermFeatureExtractor> extractor_;
   base::hash_set<std::string> term_hashes_;
-  base::hash_set<uint32> word_hashes_;
+  base::hash_set<uint32_t> word_hashes_;
   bool success_;  // holds the success value from ExtractFeatures
 };
 
 TEST_F(PhishingTermFeatureExtractorTest, ExtractFeatures) {
   // This test doesn't exercise the extraction timing.
   EXPECT_CALL(clock_, Now()).WillRepeatedly(Return(base::TimeTicks::Now()));
 
   base::string16 page_text = ASCIIToUTF16("blah");
   FeatureMap expected_features;  // initially empty
-  std::set<uint32> expected_shingle_hashes;
+  std::set<uint32_t> expected_shingle_hashes;
 
   FeatureMap features;
-  std::set<uint32> shingle_hashes;
+  std::set<uint32_t> shingle_hashes;
   ASSERT_TRUE(ExtractFeatures(&page_text, &features, &shingle_hashes));
   ExpectFeatureMapsAreEqual(features, expected_features);
   EXPECT_THAT(expected_shingle_hashes, testing::ContainerEq(shingle_hashes));
 
   page_text = ASCIIToUTF16("one one");
   expected_features.Clear();
   expected_features.AddBooleanFeature(features::kPageTerm +
                                       std::string("one"));
   expected_features.AddBooleanFeature(features::kPageTerm +
                                       std::string("one one"));
@@ skipping 72 matching lines @@
   expected_features.Clear();
   expected_shingle_hashes.clear();
   expected_shingle_hashes.insert(MurmurHash3String("this page has way ",
                                                    kMurmurHash3Seed));
   expected_shingle_hashes.insert(MurmurHash3String("page has way too ",
                                                    kMurmurHash3Seed));
   expected_shingle_hashes.insert(MurmurHash3String("has way too many ",
                                                    kMurmurHash3Seed));
   expected_shingle_hashes.insert(MurmurHash3String("way too many words ",
                                                    kMurmurHash3Seed));
-  std::set<uint32>::iterator it = expected_shingle_hashes.end();
+  std::set<uint32_t>::iterator it = expected_shingle_hashes.end();
   expected_shingle_hashes.erase(--it);
 
   features.Clear();
   shingle_hashes.clear();
   ASSERT_TRUE(ExtractFeatures(&page_text, &features, &shingle_hashes));
   ExpectFeatureMapsAreEqual(features, expected_features);
   EXPECT_THAT(expected_shingle_hashes, testing::ContainerEq(shingle_hashes));
 
   // Test with empty page text.
   page_text = base::string16();
@@ skipping 71 matching lines @@
       // Time check after the next 5 words.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(28)))
       // A final check for the histograms.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(30)));
 
   FeatureMap expected_features;
   expected_features.AddBooleanFeature(features::kPageTerm +
                                       std::string("one"));
   expected_features.AddBooleanFeature(features::kPageTerm +
                                       std::string("two"));
-  std::set<uint32> expected_shingle_hashes;
+  std::set<uint32_t> expected_shingle_hashes;
   expected_shingle_hashes.insert(
       MurmurHash3String("one 0 1 2 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("0 1 2 3 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("1 2 3 4 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("2 3 4 5 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("3 4 5 6 ", kMurmurHash3Seed));
@@ skipping 36 matching lines @@
   expected_shingle_hashes.insert(
       MurmurHash3String("22 23 24 25 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("23 24 25 26 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("24 25 26 27 ", kMurmurHash3Seed));
   expected_shingle_hashes.insert(
       MurmurHash3String("25 26 27 two ", kMurmurHash3Seed));
 
   FeatureMap features;
-  std::set<uint32> shingle_hashes;
+  std::set<uint32_t> shingle_hashes;
   ASSERT_TRUE(ExtractFeatures(&page_text, &features, &shingle_hashes));
   ExpectFeatureMapsAreEqual(features, expected_features);
   EXPECT_THAT(expected_shingle_hashes, testing::ContainerEq(shingle_hashes));
   // Make sure none of the mock expectations carry over to the next test.
   ::testing::Mock::VerifyAndClearExpectations(&clock_);
 
   // Now repeat the test with the same text, but advance the clock faster so
   // that the extraction time exceeds the maximum total time for the feature
   // extractor.  Extraction should fail.  Note that this assumes
   // kMaxTotalTimeMs = 500.
@@ skipping 29 matching lines @@
       .WillOnce(Return(now))
       // Time check at the start of the first chunk of work.
       .WillOnce(Return(now))
       // Time check after the first 5 words.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(7)))
       // Time check after the next 5 words. This should be greater than
       // kMaxTimePerChunkMs so that we stop and schedule extraction for later.
       .WillOnce(Return(now + base::TimeDelta::FromMilliseconds(14)));
 
   FeatureMap features;
-  std::set<uint32> shingle_hashes;
+  std::set<uint32_t> shingle_hashes;
   // Extract first 10 words then stop.
   PartialExtractFeatures(page_text.get(), &features, &shingle_hashes);
 
   page_text.reset(new base::string16());
   for (int i = 30; i < 58; ++i) {
     page_text->append(ASCIIToUTF16(base::StringPrintf("%d ", i)));
   }
   page_text->append(ASCIIToUTF16("multi word test "));
   features.Clear();
   shingle_hashes.clear();
 
   // This part doesn't exercise the extraction timing.
   EXPECT_CALL(clock_, Now()).WillRepeatedly(Return(base::TimeTicks::Now()));
 
   // Now extract normally and make sure nothing breaks.
   EXPECT_TRUE(ExtractFeatures(page_text.get(), &features, &shingle_hashes));
 
   FeatureMap expected_features;
   expected_features.AddBooleanFeature(features::kPageTerm +
                                       std::string("multi word test"));
   ExpectFeatureMapsAreEqual(features, expected_features);
 }
 
 }  // namespace safe_browsing