download_http_utils_unittest.cc 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. // Copyright 2021 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/download/network/download_http_utils.h"
  5. #include "net/http/http_byte_range.h"
  6. #include "net/http/http_request_headers.h"
  7. #include "testing/gtest/include/gtest/gtest.h"
  8. namespace download {
  9. namespace {
  10. TEST(DownloadHttpUtils, ParseRangeHeader) {
  11. net::HttpRequestHeaders request_headers;
  12. auto byte_range = ParseRangeHeader(request_headers);
  13. EXPECT_FALSE(byte_range.has_value());
  14. request_headers.AddHeaderFromString("Range: bytes=5-10");
  15. byte_range = ParseRangeHeader(request_headers);
  16. EXPECT_EQ(5, byte_range->first_byte_position());
  17. EXPECT_EQ(10, byte_range->last_byte_position());
  18. request_headers.Clear();
  19. request_headers.AddHeaderFromString("Range: bytes=5-");
  20. byte_range = ParseRangeHeader(request_headers);
  21. EXPECT_EQ(5, byte_range->first_byte_position());
  22. EXPECT_EQ(-1, byte_range->last_byte_position());
  23. request_headers.Clear();
  24. request_headers.AddHeaderFromString("Range: bytes=-5");
  25. byte_range = ParseRangeHeader(request_headers);
  26. EXPECT_TRUE(byte_range->IsSuffixByteRange());
  27. EXPECT_FALSE(byte_range->HasFirstBytePosition());
  28. EXPECT_FALSE(byte_range->HasLastBytePosition());
  29. EXPECT_EQ(5, byte_range->suffix_length());
  30. request_headers.Clear();
  31. request_headers.AddHeaderFromString("Range: bytes=5-10, 11-12");
  32. byte_range = ParseRangeHeader(request_headers);
  33. EXPECT_FALSE(byte_range.has_value()) << "Multiple range are not supported.";
  34. }
  35. TEST(DownloadHttpUtils, ValidateRequestHeaders) {
  36. net::HttpRequestHeaders request_headers;
  37. EXPECT_TRUE(ValidateRequestHeaders(request_headers));
  38. request_headers.AddHeaderFromString("Range: bytes=5-10");
  39. EXPECT_TRUE(ValidateRequestHeaders(request_headers));
  40. request_headers.Clear();
  41. request_headers.AddHeaderFromString("Range: kales=5-10");
  42. EXPECT_FALSE(ValidateRequestHeaders(request_headers));
  43. }
  44. } // namespace
  45. } // namespace download