File: module_ids_unittest.cc

package info (click to toggle)
chromium-browser 41.0.2272.118-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie-kfreebsd
  • size: 2,189,132 kB
  • sloc: cpp: 9,691,462; ansic: 3,341,451; python: 712,689; asm: 518,779; xml: 208,926; java: 169,820; sh: 119,353; perl: 68,907; makefile: 28,311; yacc: 13,305; objc: 11,385; tcl: 3,186; cs: 2,225; sql: 2,217; lex: 2,215; lisp: 1,349; pascal: 1,256; awk: 407; ruby: 155; sed: 53; php: 14; exp: 11
file content (75 lines) | stat: -rw-r--r-- 3,129 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// Copyright 2013 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/browser/install_verification/win/module_ids.h"

#include "base/strings/string_piece.h"
#include "chrome/browser/install_verification/win/module_verification_common.h"
#include "testing/gtest/include/gtest/gtest.h"

TEST(ModuleIDsTest, LoadModuleIDs) {
  ModuleIDs module_ids;
  LoadModuleIDs(&module_ids);
  ASSERT_EQ(1u, module_ids[CalculateModuleNameDigest(L"chrome.dll")]);
}

TEST(ModuleIDsTest, ParseAdditionalModuleIDs) {
  ModuleIDs module_ids;

  ParseAdditionalModuleIDs(base::StringPiece(), &module_ids);
  ASSERT_EQ(0u, module_ids.size());

  // Invalid input.
  ParseAdditionalModuleIDs("hello", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello world", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello world\nfoo bar", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("123 world\nfoo bar", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello 0123456789abcdef0123456789abcdef\nfoo bar",
                           &module_ids);
  ASSERT_EQ(0u, module_ids.size());

  // A single valid line followed by an invalid line.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\nfoo bar",
                         &module_ids);
  ASSERT_EQ(1u, module_ids.size());
  ASSERT_EQ(123u, module_ids.begin()->second);
  ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
  module_ids.clear();

  // The same, but with \r\n.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\nfoo bar",
                           &module_ids);
  ASSERT_EQ(1u, module_ids.size());
  ASSERT_EQ(123u, module_ids.begin()->second);
  ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
  module_ids.clear();

  // Several valid and invalid lines, with varying line terminations etc.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\n"
                           "456 DEADBEEFDEADBEEF\r"
                           "789 DEADBEEFDEADBEEFDEADBEEFDEADBEEF\n"
                           "\n"
                           "\n"
                           "321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE\n",
                           &module_ids);
  ASSERT_EQ(3u, module_ids.size());
  ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
  ASSERT_EQ(789u, module_ids["DEADBEEFDEADBEEFDEADBEEFDEADBEEF"]);
  ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
  module_ids.clear();

  // Leading empty lines, no termination on final line.
  ParseAdditionalModuleIDs("\n"
                           "123 0123456789abcdef0123456789abcdef\r\n"
                           "321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE",
                           &module_ids);
  ASSERT_EQ(2u, module_ids.size());
  ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
  ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
  module_ids.clear();
}