1234567891011121314151617181920212223242526272829303132333435363738394041 |
- // Copyright 2017 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #ifndef COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_
- #define COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_
- #include <memory>
- #include "components/assist_ranker/proto/ranker_model.pb.h"
- #include "components/assist_ranker/ranker_model_loader.h"
- namespace assist_ranker {
- namespace testing {
- // Simplified RankerModelLoader for testing.
- class FakeRankerModelLoader : public RankerModelLoader {
- public:
- FakeRankerModelLoader(ValidateModelCallback validate_model_cb,
- OnModelAvailableCallback on_model_available_cb,
- std::unique_ptr<RankerModel> ranker_model);
- FakeRankerModelLoader(const FakeRankerModelLoader&) = delete;
- FakeRankerModelLoader& operator=(const FakeRankerModelLoader&) = delete;
- ~FakeRankerModelLoader() override;
- void NotifyOfRankerActivity() override;
- private:
- std::unique_ptr<RankerModel> ranker_model_;
- const ValidateModelCallback validate_model_cb_;
- const OnModelAvailableCallback on_model_available_cb_;
- };
- } // namespace testing
- } // namespace assist_ranker
- #endif // COMPONENTS_ASSIST_RANKER_FAKE_RANKER_MODEL_LOADER_H_
|