permission_request_queue_unittest.cc 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. // Copyright 2022 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <stddef.h>
  5. #include <memory>
  6. #include <string>
  7. #include "base/bind.h"
  8. #include "base/command_line.h"
  9. #include "base/test/scoped_feature_list.h"
  10. #include "build/build_config.h"
  11. #include "components/permissions/features.h"
  12. #include "components/permissions/permission_request.h"
  13. #include "components/permissions/permission_request_queue.h"
  14. #include "components/permissions/request_type.h"
  15. #include "components/permissions/test/mock_permission_prompt_factory.h"
  16. #include "components/permissions/test/mock_permission_request.h"
  17. #include "testing/gtest/include/gtest/gtest.h"
  18. namespace permissions {
  19. enum UiFeatureConfig { NoChip, QuietChipOnly, ChipOnly, QuietChipAndChip };
  20. class PermissionRequestQueueTest
  21. : public ::testing::Test,
  22. public ::testing::WithParamInterface<UiFeatureConfig> {
  23. public:
  24. PermissionRequestQueueTest()
  25. : request1_(RequestType::kGeolocation,
  26. PermissionRequestGestureType::GESTURE),
  27. request2_(RequestType::kMultipleDownloads,
  28. PermissionRequestGestureType::NO_GESTURE) {
  29. switch (GetParam()) {
  30. case NoChip:
  31. feature_list_.InitWithFeatures(
  32. {}, {permissions::features::kPermissionChip,
  33. permissions::features::kPermissionQuietChip});
  34. break;
  35. case QuietChipOnly:
  36. feature_list_.InitWithFeatures(
  37. {permissions::features::kPermissionQuietChip},
  38. {permissions::features::kPermissionChip});
  39. break;
  40. case ChipOnly:
  41. feature_list_.InitWithFeatures(
  42. {permissions::features::kPermissionChip},
  43. {permissions::features::kPermissionQuietChip});
  44. break;
  45. case QuietChipAndChip:
  46. feature_list_.InitWithFeatures(
  47. {permissions::features::kPermissionChip,
  48. permissions::features::kPermissionQuietChip},
  49. {});
  50. break;
  51. }
  52. }
  53. protected:
  54. PermissionRequestQueue permission_request_queue_;
  55. MockPermissionRequest request1_;
  56. MockPermissionRequest request2_;
  57. base::test::ScopedFeatureList feature_list_;
  58. };
  59. TEST_P(PermissionRequestQueueTest, CountNumberOfRequestsInQueue) {
  60. EXPECT_EQ(0ul, permission_request_queue_.Count());
  61. permission_request_queue_.Push(&request1_);
  62. permission_request_queue_.Push(&request2_);
  63. EXPECT_EQ(2ul, permission_request_queue_.Count());
  64. permission_request_queue_.Pop();
  65. EXPECT_EQ(1ul, permission_request_queue_.Count());
  66. }
  67. TEST_P(PermissionRequestQueueTest, CountDuplicateRequests) {
  68. EXPECT_EQ(0ul, permission_request_queue_.Count());
  69. permission_request_queue_.Push(&request1_);
  70. permission_request_queue_.Push(&request1_);
  71. EXPECT_EQ(2ul, permission_request_queue_.Count());
  72. }
  73. TEST_P(PermissionRequestQueueTest, CountNumberOfRequestOccurencesInQueue) {
  74. EXPECT_EQ(0ul, permission_request_queue_.Count(&request1_));
  75. permission_request_queue_.Push(&request1_);
  76. permission_request_queue_.Push(&request1_);
  77. permission_request_queue_.Push(&request2_);
  78. EXPECT_EQ(2ul, permission_request_queue_.Count(&request1_));
  79. }
  80. TEST_P(PermissionRequestQueueTest, OnlyEmptyWithoutRequests) {
  81. EXPECT_TRUE(permission_request_queue_.IsEmpty());
  82. permission_request_queue_.Push(&request1_);
  83. EXPECT_FALSE(permission_request_queue_.IsEmpty());
  84. permission_request_queue_.Pop();
  85. EXPECT_TRUE(permission_request_queue_.IsEmpty());
  86. }
  87. TEST_P(PermissionRequestQueueTest, ShouldFindDuplicateRequest) {
  88. permission_request_queue_.Push(&request1_);
  89. permission_request_queue_.Push(&request2_);
  90. EXPECT_EQ(&request2_, permission_request_queue_.FindDuplicate(&request2_));
  91. }
  92. TEST_P(PermissionRequestQueueTest, ShouldNotFindDuplicateIfNotPresent) {
  93. permission_request_queue_.Push(&request1_);
  94. EXPECT_EQ(nullptr, permission_request_queue_.FindDuplicate(&request2_));
  95. }
  96. TEST_P(PermissionRequestQueueTest, PeekedElementIsNextPoppedElement) {
  97. permission_request_queue_.Push(&request1_);
  98. permission_request_queue_.Push(&request2_);
  99. PermissionRequest* peekedElement = permission_request_queue_.Peek();
  100. EXPECT_EQ(peekedElement, permission_request_queue_.Pop());
  101. }
  102. TEST_P(PermissionRequestQueueTest, VerifyPushOrder) {
  103. permission_request_queue_.Push(&request1_);
  104. permission_request_queue_.Push(&request2_);
  105. permission_request_queue_.Push(&request2_);
  106. if (GetParam() == NoChip) {
  107. EXPECT_EQ(permission_request_queue_.Pop(), &request1_);
  108. EXPECT_EQ(permission_request_queue_.Pop(), &request2_);
  109. EXPECT_EQ(permission_request_queue_.Pop(), &request2_);
  110. } else { // QuietChipOnly, ChipOnly, QuietChipAndChip
  111. EXPECT_EQ(permission_request_queue_.Pop(), &request2_);
  112. EXPECT_EQ(permission_request_queue_.Pop(), &request2_);
  113. EXPECT_EQ(permission_request_queue_.Pop(), &request1_);
  114. }
  115. }
  116. INSTANTIATE_TEST_SUITE_P(
  117. All,
  118. PermissionRequestQueueTest,
  119. ::testing::Values(NoChip, QuietChipOnly, ChipOnly, QuietChipAndChip));
  120. } // namespace permissions