optimization_metadata_unittest.cc 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. // Copyright 2020 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 "components/optimization_guide/core/optimization_metadata.h"
  5. #include "components/optimization_guide/proto/loading_predictor_metadata.pb.h"
  6. #include "testing/gtest/include/gtest/gtest.h"
  7. namespace optimization_guide {
  8. TEST(OptimizationMetadataTest, ParsedMetadataAnyMetadataNotPopulatedTest) {
  9. OptimizationMetadata optimization_metadata;
  10. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  11. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  12. EXPECT_FALSE(parsed_metadata.has_value());
  13. }
  14. TEST(OptimizationMetadataTest, ParsedMetadataNoTypeURLTest) {
  15. proto::Any any_metadata;
  16. proto::LoadingPredictorMetadata metadata;
  17. proto::Resource* subresource = metadata.add_subresources();
  18. subresource->set_url("https://example.com/");
  19. subresource->set_resource_type(proto::ResourceType::RESOURCE_TYPE_CSS);
  20. subresource->set_preconnect_only(true);
  21. metadata.SerializeToString(any_metadata.mutable_value());
  22. OptimizationMetadata optimization_metadata;
  23. optimization_metadata.set_any_metadata(any_metadata);
  24. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  25. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  26. EXPECT_FALSE(parsed_metadata.has_value());
  27. }
  28. TEST(OptimizationMetadataTest, ParsedMetadataMismatchedTypeTest) {
  29. proto::Any any_metadata;
  30. any_metadata.set_type_url("type.googleapis.com/com.foo.Whatever");
  31. proto::LoadingPredictorMetadata metadata;
  32. proto::Resource* subresource = metadata.add_subresources();
  33. subresource->set_url("https://example.com/");
  34. subresource->set_resource_type(proto::ResourceType::RESOURCE_TYPE_CSS);
  35. subresource->set_preconnect_only(true);
  36. metadata.SerializeToString(any_metadata.mutable_value());
  37. OptimizationMetadata optimization_metadata;
  38. optimization_metadata.set_any_metadata(any_metadata);
  39. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  40. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  41. EXPECT_FALSE(parsed_metadata.has_value());
  42. }
  43. TEST(OptimizationMetadataTest, ParsedMetadataNotSerializableTest) {
  44. proto::Any any_metadata;
  45. any_metadata.set_type_url(
  46. "type.googleapis.com/com.foo.LoadingPredictorMetadata");
  47. any_metadata.set_value("12345678garbage");
  48. OptimizationMetadata optimization_metadata;
  49. optimization_metadata.set_any_metadata(any_metadata);
  50. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  51. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  52. EXPECT_FALSE(parsed_metadata.has_value());
  53. }
  54. TEST(OptimizationMetadataTest, ParsedMetadataTest) {
  55. proto::Any any_metadata;
  56. any_metadata.set_type_url(
  57. "type.googleapis.com/com.foo.LoadingPredictorMetadata");
  58. proto::LoadingPredictorMetadata metadata;
  59. proto::Resource* subresource = metadata.add_subresources();
  60. subresource->set_url("https://example.com/");
  61. subresource->set_resource_type(proto::ResourceType::RESOURCE_TYPE_CSS);
  62. subresource->set_preconnect_only(true);
  63. metadata.SerializeToString(any_metadata.mutable_value());
  64. OptimizationMetadata optimization_metadata;
  65. optimization_metadata.set_any_metadata(any_metadata);
  66. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  67. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  68. EXPECT_TRUE(parsed_metadata.has_value());
  69. ASSERT_EQ(parsed_metadata->subresources_size(), 1);
  70. const proto::Resource& parsed_subresource = parsed_metadata->subresources(0);
  71. EXPECT_EQ(parsed_subresource.url(), "https://example.com/");
  72. EXPECT_EQ(parsed_subresource.resource_type(),
  73. proto::ResourceType::RESOURCE_TYPE_CSS);
  74. EXPECT_TRUE(parsed_subresource.preconnect_only());
  75. }
  76. TEST(OptimizationMetadataTest, SetAnyMetadataForTestingTest) {
  77. proto::LoadingPredictorMetadata metadata;
  78. proto::Resource* subresource = metadata.add_subresources();
  79. subresource->set_url("https://example.com/");
  80. subresource->set_resource_type(proto::ResourceType::RESOURCE_TYPE_CSS);
  81. subresource->set_preconnect_only(true);
  82. OptimizationMetadata optimization_metadata;
  83. optimization_metadata.SetAnyMetadataForTesting(metadata);
  84. absl::optional<proto::LoadingPredictorMetadata> parsed_metadata =
  85. optimization_metadata.ParsedMetadata<proto::LoadingPredictorMetadata>();
  86. EXPECT_TRUE(parsed_metadata.has_value());
  87. ASSERT_EQ(parsed_metadata->subresources_size(), 1);
  88. const proto::Resource& parsed_subresource = parsed_metadata->subresources(0);
  89. EXPECT_EQ(parsed_subresource.url(), "https://example.com/");
  90. EXPECT_EQ(parsed_subresource.resource_type(),
  91. proto::ResourceType::RESOURCE_TYPE_CSS);
  92. EXPECT_TRUE(parsed_subresource.preconnect_only());
  93. }
  94. } // namespace optimization_guide