1 /*
2  * Copyright (c) 2021-2021 Huawei Device Co., Ltd. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without modification,
5  * are permitted provided that the following conditions are met:
6  *
7  * 1. Redistributions of source code must retain the above copyright notice, this list of
8  * conditions and the following disclaimer.
9  *
10  * 2. Redistributions in binary form must reproduce the above copyright notice, this list
11  * of conditions and the following disclaimer in the documentation and/or other materials
12  * provided with the distribution.
13  *
14  * 3. Neither the name of the copyright holder nor the names of its contributors may be used
15  * to endorse or promote products derived from this software without specific prior written
16  * permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
20  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
22  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
25  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
26  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
27  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
28  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30 
31 #include "osTest.h"
32 #include "It_los_lmk.h"
33 #include "los_list.h"
34 
release_OKnull35 STATIC UINT32 release_OK(VOID)
36 {
37     return LOS_OK;
38 }
39 
release_NOKnull40 STATIC UINT32 release_NOK(VOID)
41 {
42     return LOS_NOK;
43 }
44 
restore_OKnull45 STATIC UINT32 restore_OK(VOID)
46 {
47     return LOS_OK;
48 }
49 
restore_NOKnull50 STATIC UINT32 restore_NOK(VOID)
51 {
52     return LOS_NOK;
53 }
54 
TestCasenull55 STATIC UINT32 TestCase(VOID)
56 {
57     UINT32 ret;
58     LosLmkOpsNode firstOpsNode = {
59         .priority = LMK_PRIORITY_LOW,
60         .freeMem = release_NOK,
61         .restoreTask = restore_NOK,
62     };
63 
64     LosLmkOpsNode anotherOpsNode = {
65         .priority = LMK_PRIORITY_MEDIUM,
66         .freeMem = release_NOK,
67         .restoreTask = restore_NOK,
68     };
69 
70     LosLmkOpsNode thirdOpsNode = {
71         .priority = LMK_PRIORITY_HIGH,
72         .freeMem = release_OK,
73         .restoreTask = restore_OK,
74     };
75 
76     ret = LOS_LmkOpsNodeRegister(&firstOpsNode);
77     ICUNIT_ASSERT_EQUAL(ret, LOS_OK, ret);
78 
79     ret = LOS_LmkOpsNodeRegister(&anotherOpsNode);
80     ICUNIT_ASSERT_EQUAL(ret, LOS_OK, ret);
81 
82     ret = LOS_LmkTasksKill();
83     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_FREE_MEMORY_FAILURE, ret);
84     ret = LOS_LmkTasksRestore();
85     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_RESTORE_TASKS_FAILURE, ret);
86 
87     ret = LOS_LmkOpsNodeRegister(&thirdOpsNode);
88     ICUNIT_ASSERT_EQUAL(ret, LOS_OK, ret);
89     ret = LOS_LmkTasksKill();
90     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_FREE_MEMORY_FAILURE, ret);
91     ret = LOS_LmkTasksKill();
92     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_MEMORY_ALREADY_FREED, ret);
93     ret = LOS_LmkTasksRestore();
94     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_RESTORE_TASKS_FAILURE, ret);
95     ret = LOS_LmkTasksRestore();
96     ICUNIT_ASSERT_EQUAL(ret, LOS_ERRNO_LMK_RESTORE_NOT_NEEDED, ret);
97 
98     firstOpsNode.freeMem = release_OK;
99     firstOpsNode.restoreTask = restore_OK;
100     anotherOpsNode.freeMem = release_OK;
101     anotherOpsNode.restoreTask = restore_OK;
102     ret = LOS_LmkTasksKill();
103     ICUNIT_ASSERT_EQUAL(ret, LOS_OK, ret);
104     ret = LOS_LmkTasksRestore();
105     ICUNIT_ASSERT_EQUAL(ret, LOS_OK, ret);
106 
107     (VOID)LOS_LmkOpsNodeUnregister(&firstOpsNode);
108     (VOID)LOS_LmkOpsNodeUnregister(&anotherOpsNode);
109     (VOID)LOS_LmkOpsNodeUnregister(&thirdOpsNode);
110 
111     return LOS_OK;
112 }
113 
ItLosLmk003null114 VOID ItLosLmk003(VOID)
115 {
116     TEST_ADD_CASE("ItLosLmk003", TestCase, TEST_LOS, TEST_COMP, TEST_LEVEL0, TEST_FUNCTION);
117 }
118