fake_ranker_model_loader.h 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. // Copyright 2017 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. #ifndef COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_
  5. #define COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_
  6. #include <memory>
  7. #include "components/assist_ranker/proto/ranker_model.pb.h"
  8. #include "components/assist_ranker/ranker_model_loader.h"
  9. namespace assist_ranker {
  10. namespace testing {
  11. // Simplified RankerModelLoader for testing.
  12. class FakeRankerModelLoader : public RankerModelLoader {
  13. public:
  14. FakeRankerModelLoader(ValidateModelCallback validate_model_cb,
  15. OnModelAvailableCallback on_model_available_cb,
  16. std::unique_ptr<RankerModel> ranker_model);
  17. FakeRankerModelLoader(const FakeRankerModelLoader&) = delete;
  18. FakeRankerModelLoader& operator=(const FakeRankerModelLoader&) = delete;
  19. ~FakeRankerModelLoader() override;
  20. void NotifyOfRankerActivity() override;
  21. private:
  22. std::unique_ptr<RankerModel> ranker_model_;
  23. const ValidateModelCallback validate_model_cb_;
  24. const OnModelAvailableCallback on_model_available_cb_;
  25. };
  26. } // namespace testing
  27. } // namespace assist_ranker
  28. #endif // COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_