summaryrefslogtreecommitdiffstats
path: root/dom/quota/test/gtest/TestScopedLogExtraInfo.cpp
blob: 395246c1fb01d8db47f614b229901beb2394ae22 (plain)
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
76
77
78
79
80
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "mozilla/dom/quota/ScopedLogExtraInfo.h"

#include "gtest/gtest.h"

using namespace mozilla::dom::quota;

TEST(DOM_Quota_ScopedLogExtraInfo, AddAndRemove)
{
  static constexpr auto text = "foo"_ns;

  {
    const auto extraInfo =
        ScopedLogExtraInfo{ScopedLogExtraInfo::kTagQueryTainted, text};

#ifdef QM_SCOPED_LOG_EXTRA_INFO_ENABLED
    const auto& extraInfoMap = ScopedLogExtraInfo::GetExtraInfoMap();

    const auto& queryValueTainted =
        *extraInfoMap.at(ScopedLogExtraInfo::kTagQueryTainted);

    EXPECT_EQ(text, MOZ_NO_VALIDATE(queryValueTainted,
                                    "It's ok to use query value in tests."));
#endif
  }

#ifdef QM_SCOPED_LOG_EXTRA_INFO_ENABLED
  const auto& extraInfoMap = ScopedLogExtraInfo::GetExtraInfoMap();

  EXPECT_EQ(0u, extraInfoMap.count(ScopedLogExtraInfo::kTagQueryTainted));
#endif
}

TEST(DOM_Quota_ScopedLogExtraInfo, Nested)
{
  static constexpr auto text = "foo"_ns;
  static constexpr auto nestedText = "bar"_ns;

  {
    const auto extraInfo =
        ScopedLogExtraInfo{ScopedLogExtraInfo::kTagQueryTainted, text};

    {
      const auto extraInfo =
          ScopedLogExtraInfo{ScopedLogExtraInfo::kTagQueryTainted, nestedText};

#ifdef QM_SCOPED_LOG_EXTRA_INFO_ENABLED
      const auto& extraInfoMap = ScopedLogExtraInfo::GetExtraInfoMap();

      const auto& queryValueTainted =
          *extraInfoMap.at(ScopedLogExtraInfo::kTagQueryTainted);

      EXPECT_EQ(nestedText,
                MOZ_NO_VALIDATE(queryValueTainted,
                                "It's ok to use query value in tests."));
#endif
    }

#ifdef QM_SCOPED_LOG_EXTRA_INFO_ENABLED
    const auto& extraInfoMap = ScopedLogExtraInfo::GetExtraInfoMap();

    const auto& queryValueTainted =
        *extraInfoMap.at(ScopedLogExtraInfo::kTagQueryTainted);

    EXPECT_EQ(text, MOZ_NO_VALIDATE(queryValueTainted,
                                    "It's ok to use query value in tests."));
#endif
  }

#ifdef QM_SCOPED_LOG_EXTRA_INFO_ENABLED
  const auto& extraInfoMap = ScopedLogExtraInfo::GetExtraInfoMap();

  EXPECT_EQ(0u, extraInfoMap.count(ScopedLogExtraInfo::kTagQueryTainted));
#endif
}