quarantine_service_unittest.cc 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. // Copyright 2019 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 <memory>
  5. #include <utility>
  6. #include "base/bind.h"
  7. #include "base/callback.h"
  8. #include "base/files/file_path.h"
  9. #include "base/files/file_util.h"
  10. #include "base/files/scoped_temp_dir.h"
  11. #include "base/run_loop.h"
  12. #include "base/test/task_environment.h"
  13. #include "components/services/quarantine/public/mojom/quarantine.mojom.h"
  14. #include "components/services/quarantine/quarantine_impl.h"
  15. #include "mojo/public/cpp/bindings/remote.h"
  16. #include "testing/gmock/include/gmock/gmock.h"
  17. #include "testing/gtest/include/gtest/gtest.h"
  18. namespace quarantine {
  19. const char kTestData[] = "It's okay to have a trailing nul.";
  20. const char kInternetURL[] = "http://example.com/some-url";
  21. const char kInternetReferrerURL[] = "http://example.com/some-other-url";
  22. class QuarantineServiceTest : public testing::Test {
  23. public:
  24. QuarantineServiceTest() = default;
  25. QuarantineServiceTest(const QuarantineServiceTest&) = delete;
  26. QuarantineServiceTest& operator=(const QuarantineServiceTest&) = delete;
  27. ~QuarantineServiceTest() override = default;
  28. void OnFileQuarantined(const base::FilePath& test_file,
  29. base::OnceClosure quit_closure,
  30. mojom::QuarantineFileResult result) {
  31. base::DeleteFile(test_file);
  32. result_ = result;
  33. std::move(quit_closure).Run();
  34. }
  35. protected:
  36. base::test::TaskEnvironment task_environment_;
  37. mojo::Remote<mojom::Quarantine> quarantine_;
  38. mojom::QuarantineFileResult result_;
  39. private:
  40. QuarantineImpl service_{quarantine_.BindNewPipeAndPassReceiver()};
  41. };
  42. TEST_F(QuarantineServiceTest, QuarantineFile) {
  43. base::ScopedTempDir test_dir;
  44. ASSERT_TRUE(test_dir.CreateUniqueTempDir());
  45. base::FilePath test_file = test_dir.GetPath().AppendASCII("foo.class");
  46. ASSERT_EQ(static_cast<int>(std::size(kTestData)),
  47. base::WriteFile(test_file, kTestData, std::size(kTestData)));
  48. base::RunLoop run_loop;
  49. quarantine_->QuarantineFile(
  50. test_file, GURL(kInternetURL), GURL(kInternetReferrerURL), std::string(),
  51. base::BindOnce(&QuarantineServiceTest::OnFileQuarantined,
  52. base::Unretained(this), test_file,
  53. run_loop.QuitClosure()));
  54. run_loop.Run();
  55. EXPECT_EQ(result_, mojom::QuarantineFileResult::OK);
  56. }
  57. } // namespace quarantine