1 /*
2  * Copyright (C) 2021 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 NET_DETECTION_CALLBACK_TEST_H
17 #define NET_DETECTION_CALLBACK_TEST_H
18 
19 #include <condition_variable>
20 #include <mutex>
21 
22 #include "net_detection_callback_stub.h"
23 
24 namespace OHOS {
25 namespace NetManagerStandard {
26 class NetDetectionCallbackTest : public NetDetectionCallbackStub {
27 public:
28     NetDetectionCallbackTest();
29     ~NetDetectionCallbackTest() override;
30     int32_t OnNetDetectionResultChanged(
31         NetDetectionResultCode detectionResult, const std::string &urlRedirect) override;
32 
33     void WaitFor(int timeoutSecond);
34 
GetNetDetectionResult()35     int32_t GetNetDetectionResult() const
36     {
37         return netDetectionResult_;
38     }
39 
GetUrlRedirect()40     std::string GetUrlRedirect() const
41     {
42         return urlRedirect_;
43     }
44 
45 private:
46     void NotifyAll();
47     std::mutex callbackMutex_;
48     std::condition_variable cv_;
49     int32_t netDetectionResult_ = 0;
50     std::string urlRedirect_;
51 };
52 } // namespace NetManagerStandard
53 } // namespace OHOS
54 #endif // NET_DETECTION_CALLBACK_TEST_H
55