1 /*
2  * Copyright (c) 2024 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef DSCHED_CONTINUE_STATE_TEST_H
17 #define DSCHED_CONTINUE_STATE_TEST_H
18 
19 #include "gtest/gtest.h"
20 
21 #include "dsched_continue_data_state.h"
22 #include "dsched_continue_sink_end_state.h"
23 #include "dsched_continue_sink_start_state.h"
24 #include "dsched_continue_sink_wait_end_state.h"
25 #include "dsched_continue_ability_state.h"
26 #include "dsched_continue_source_end_state.h"
27 #include "dsched_continue_source_start_state.h"
28 #include "dsched_continue_source_wait_end_state.h"
29 #include "dsched_continue_state_machine.h"
30 
31 namespace OHOS {
32 namespace DistributedSchedule {
33 
34 class DSchedContinueDataStateTest : public testing::Test {
35 public:
36     static void SetUpTestCase();
37     static void TearDownTestCase();
38     void SetUp();
39     void TearDown();
40     std::shared_ptr<DSchedContinue> CreateObject();
41     std::shared_ptr<DSchedContinueDataState> dataStateTest_;
42 };
43 
44 class DSchedContinueSinkEndStateTest : public testing::Test {
45 public:
46     static void SetUpTestCase();
47     static void TearDownTestCase();
48     void SetUp();
49     void TearDown();
50     std::shared_ptr<DSchedContinue> CreateObject();
51     std::shared_ptr<DSchedContinueSinkEndState> sinkEndStateTest_;
52 };
53 
54 class DSchedContinueSinkStartStateTest : public testing::Test {
55 public:
56     static void SetUpTestCase();
57     static void TearDownTestCase();
58     void SetUp();
59     void TearDown();
60     std::shared_ptr<DSchedContinue> CreateObject();
61     std::shared_ptr<DSchedContinueSinkStartState> sinkStartStateTest_;
62 };
63 
64 class DSchedContinueSinkWaitEndStateTest : public testing::Test {
65 public:
66     static void SetUpTestCase();
67     static void TearDownTestCase();
68     void SetUp();
69     void TearDown();
70     std::shared_ptr<DSchedContinue> CreateObject();
71     std::shared_ptr<DSchedContinueSinkWaitEndState> sinkWaitEndTest_;
72 };
73 
74 class DSchedContinueAbilityStateTest : public testing::Test {
75 public:
76     static void SetUpTestCase();
77     static void TearDownTestCase();
78     void SetUp();
79     void TearDown();
80     std::shared_ptr<DSchedContinue> CreateObject();
81     std::shared_ptr<DSchedContinueAbilityState> abilityStateTest_;
82 };
83 
84 class DSchedContinueEndStateTest : public testing::Test {
85 public:
86     static void SetUpTestCase();
87     static void TearDownTestCase();
88     void SetUp();
89     void TearDown();
90     std::shared_ptr<DSchedContinue> CreateObject();
91     std::shared_ptr<DSchedContinueEndState> srcEndStateTest_;
92 };
93 
94 class DSchedContinueSourceStartStateTest : public testing::Test {
95 public:
96     static void SetUpTestCase();
97     static void TearDownTestCase();
98     void SetUp();
99     void TearDown();
100     std::shared_ptr<DSchedContinue> CreateObject();
101     std::shared_ptr<DSchedContinueSourceStartState> srcStartStateTest_;
102 };
103 
104 class DSchedContinueWaitEndStateTest : public testing::Test {
105 public:
106     static void SetUpTestCase();
107     static void TearDownTestCase();
108     void SetUp();
109     void TearDown();
110     std::shared_ptr<DSchedContinue> CreateObject();
111     std::shared_ptr<DSchedContinueWaitEndState> srcWaitEndTest_;
112 };
113 
114 class DSchedContinueStateMachineTest : public testing::Test {
115 public:
116     static void SetUpTestCase();
117     static void TearDownTestCase();
118     void SetUp();
119     void TearDown();
120     std::shared_ptr<DSchedContinue> CreateObject();
121     std::shared_ptr<DSchedContinueStateMachine> stateMachineTest_;
122 };
123 } // namespace DistributedSchedule
124 } // namespace OHOS
125 #endif // DSCHED_CONTINUE_STATE_TEST_H