1 /*
2  * Copyright (C) 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef LIBTEXTCLASSIFIER_ANNOTATOR_DATETIME_TESTING_DATETIME_COMPONENT_BUILDER_H_
18 #define LIBTEXTCLASSIFIER_ANNOTATOR_DATETIME_TESTING_DATETIME_COMPONENT_BUILDER_H_
19 
20 #include "annotator/types.h"
21 
22 namespace libtextclassifier3 {
23 
24 // Builder class to construct the DatetimeComponents and make the test readable.
25 class DatetimeComponentsBuilder {
26  public:
27   DatetimeComponentsBuilder Add(DatetimeComponent::ComponentType type,
28                                 int value);
29 
30   DatetimeComponentsBuilder Add(
31       DatetimeComponent::ComponentType type, int value,
32       DatetimeComponent::RelativeQualifier relative_qualifier,
33       int relative_count);
34 
35   std::vector<DatetimeComponent> Build();
36 
37  private:
38   DatetimeComponentsBuilder AddComponent(
39       const DatetimeComponent& datetime_component);
40   std::vector<DatetimeComponent> datetime_components_;
41 };
42 }  // namespace libtextclassifier3
43 
44 #endif  // LIBTEXTCLASSIFIER_ANNOTATOR_DATETIME_TESTING_DATETIME_COMPONENT_BUILDER_H_
45