Add memory usage to integration tests [UPDATED]
[cps.git] / integration-test / src / test / groovy / org / onap / cps / integration / performance / base / PerfTestBase.groovy
1 /*
2  *  ============LICENSE_START=======================================================
3  *  Copyright (C) 2023 Nordix Foundation
4  *  ================================================================================
5  *  Licensed under the Apache License, Version 2.0 (the 'License');
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *        http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an 'AS IS' BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  *  SPDX-License-Identifier: Apache-2.0
18  *  ============LICENSE_END=========================================================
19  */
20
21 package org.onap.cps.integration.performance.base
22
23 import org.onap.cps.integration.base.CpsIntegrationSpecBase
24 import org.springframework.util.StopWatch
25
26 abstract class PerfTestBase extends CpsIntegrationSpecBase {
27
28     static def LARGE_SCHEMA_SET = 'largeSchemaSet'
29     static def PERFORMANCE_RECORD = []
30
31     def cleanupSpec() {
32         println('#############################################################################')
33         printTitle()
34         println('#############################################################################')
35         PERFORMANCE_RECORD.sort().each {
36             println(it)
37         }
38         println('#############################################################################')
39         PERFORMANCE_RECORD.clear()
40     }
41
42     def setup() {
43         if (!isInitialised()) {
44             setupPerformanceInfraStructure()
45             createInitialData()
46         }
47     }
48
49     abstract def printTitle()
50
51     abstract def isInitialised()
52
53     abstract def setupPerformanceInfraStructure()
54
55     abstract def createInitialData()
56
57     def recordAndAssertResourceUsage(String shortTitle, thresholdInMs, recordedTimeInMs, memoryLimit, memoryUsageInMB) {
58         def pass = recordedTimeInMs <= thresholdInMs
59         if (shortTitle.length() > 40) {
60             shortTitle = shortTitle.substring(0, 40)
61         }
62         def record = String.format('%2d.%-40s limit%,8d took %,8d ms %,8.2f MB used ', PERFORMANCE_RECORD.size() + 1, shortTitle, thresholdInMs, recordedTimeInMs, memoryUsageInMB)
63         record += pass ? 'PASS' : 'FAIL'
64         PERFORMANCE_RECORD.add(record)
65         assert recordedTimeInMs <= thresholdInMs
66         assert memoryUsageInMB <= memoryLimit
67         return true
68     }
69 }