Lee Kai Xuan / mbed-os

Fork of mbed-os by erkin yucel

TESTS/mbed_drivers/stats/main.cpp@1:3deb71413561, 2017-07-20 (annotated)

Committer:
xuaner
Date:
Thu Jul 20 14:26:57 2017 +0000
Revision:
1:3deb71413561
Parent:
0:f269e3021894 
mbed_os

Who changed what in which revision?

UserRevisionLine numberNew contents of line
elessair 0:f269e3021894 1 /*
elessair 0:f269e3021894 2 * Copyright (c) 2013-2016, ARM Limited, All Rights Reserved
elessair 0:f269e3021894 3 * SPDX-License-Identifier: Apache-2.0
elessair 0:f269e3021894 4 *
elessair 0:f269e3021894 5 * Licensed under the Apache License, Version 2.0 (the "License"); you may
elessair 0:f269e3021894 6 * not use this file except in compliance with the License.
elessair 0:f269e3021894 7 * You may obtain a copy of the License at
elessair 0:f269e3021894 8 *
elessair 0:f269e3021894 9 * http://www.apache.org/licenses/LICENSE-2.0
elessair 0:f269e3021894 10 *
elessair 0:f269e3021894 11 * Unless required by applicable law or agreed to in writing, software
elessair 0:f269e3021894 12 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
elessair 0:f269e3021894 13 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
elessair 0:f269e3021894 14 * See the License for the specific language governing permissions and
elessair 0:f269e3021894 15 * limitations under the License.
elessair 0:f269e3021894 16 */
elessair 0:f269e3021894 17 #include "mbed.h"
elessair 0:f269e3021894 18 #include "greentea-client/test_env.h"
elessair 0:f269e3021894 19 #include "unity/unity.h"
elessair 0:f269e3021894 20 #include "utest/utest.h"
elessair 0:f269e3021894 21 #include "mbed_stats.h"
elessair 0:f269e3021894 22 #include <stdlib.h>
elessair 0:f269e3021894 23 #include <stdio.h>
elessair 0:f269e3021894 24
elessair 0:f269e3021894 25 #if !defined(MBED_HEAP_STATS_ENABLED) || !MBED_HEAP_STATS_ENABLED || defined(__ICCARM__)
elessair 0:f269e3021894 26 #error [NOT_SUPPORTED] test not supported
elessair 0:f269e3021894 27 #endif
elessair 0:f269e3021894 28
elessair 0:f269e3021894 29 using namespace utest::v1;
elessair 0:f269e3021894 30
elessair 0:f269e3021894 31 #define ALLOCATION_SIZE_DEFAULT 564
elessair 0:f269e3021894 32 #define ALLOCATION_SIZE_SMALL 124
elessair 0:f269e3021894 33 #define ALLOCATION_SIZE_LARGE 790
elessair 0:f269e3021894 34 #define ALLOCATION_SIZE_FAIL (1024 * 1024 *1024)
elessair 0:f269e3021894 35
elessair 0:f269e3021894 36 typedef void* (*malloc_cb_t) (uint32_t size);
elessair 0:f269e3021894 37
elessair 0:f269e3021894 38 static void* thunk_malloc(uint32_t size);
elessair 0:f269e3021894 39 static void* thunk_calloc_1(uint32_t size);
elessair 0:f269e3021894 40 static void* thunk_calloc_4(uint32_t size);
elessair 0:f269e3021894 41 static void* thunk_realloc(uint32_t size);
elessair 0:f269e3021894 42
elessair 0:f269e3021894 43 malloc_cb_t malloc_thunk_array[] = {
elessair 0:f269e3021894 44 thunk_malloc,
elessair 0:f269e3021894 45 thunk_calloc_1,
elessair 0:f269e3021894 46 thunk_calloc_4,
elessair 0:f269e3021894 47 thunk_realloc,
elessair 0:f269e3021894 48 };
elessair 0:f269e3021894 49
elessair 0:f269e3021894 50 void test_case_malloc_free_size()
elessair 0:f269e3021894 51 {
elessair 0:f269e3021894 52 printf("Initial print to setup stdio buffers\n");
elessair 0:f269e3021894 53 mbed_stats_heap_t stats_start;
elessair 0:f269e3021894 54 mbed_stats_heap_t stats_current;
elessair 0:f269e3021894 55 void *data;
elessair 0:f269e3021894 56
elessair 0:f269e3021894 57 mbed_stats_heap_get(&stats_start);
elessair 0:f269e3021894 58
elessair 0:f269e3021894 59 for (uint32_t i = 0; i < sizeof(malloc_thunk_array) / sizeof(malloc_cb_t); i++) {
elessair 0:f269e3021894 60
elessair 0:f269e3021894 61 // Allocate memory and assert size change
elessair 0:f269e3021894 62 data = malloc_thunk_array[i](ALLOCATION_SIZE_DEFAULT);
elessair 0:f269e3021894 63 TEST_ASSERT(data != NULL);
elessair 0:f269e3021894 64 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 65 uint32_t increase = stats_current.current_size - stats_start.current_size;
elessair 0:f269e3021894 66 TEST_ASSERT_EQUAL_UINT32(ALLOCATION_SIZE_DEFAULT, increase);
elessair 0:f269e3021894 67 TEST_ASSERT_EQUAL_UINT32(stats_start.total_size + ALLOCATION_SIZE_DEFAULT * (i + 1), stats_current.total_size);
elessair 0:f269e3021894 68 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_cnt + 1, stats_current.alloc_cnt);
elessair 0:f269e3021894 69 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_fail_cnt, stats_current.alloc_fail_cnt);
elessair 0:f269e3021894 70
elessair 0:f269e3021894 71 // Free memory and assert back to starting size
elessair 0:f269e3021894 72 free(data);
elessair 0:f269e3021894 73 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 74 TEST_ASSERT_EQUAL_UINT32(stats_start.current_size, stats_current.current_size);
elessair 0:f269e3021894 75 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_cnt, stats_current.alloc_cnt);
elessair 0:f269e3021894 76 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_fail_cnt, stats_current.alloc_fail_cnt);
elessair 0:f269e3021894 77 }
elessair 0:f269e3021894 78 }
elessair 0:f269e3021894 79
elessair 0:f269e3021894 80 void test_case_allocate_zero()
elessair 0:f269e3021894 81 {
elessair 0:f269e3021894 82 mbed_stats_heap_t stats_start;
elessair 0:f269e3021894 83 mbed_stats_heap_t stats_current;
elessair 0:f269e3021894 84 void *data;
elessair 0:f269e3021894 85
elessair 0:f269e3021894 86 mbed_stats_heap_get(&stats_start);
elessair 0:f269e3021894 87
elessair 0:f269e3021894 88 for (uint32_t i = 0; i < sizeof(malloc_thunk_array) / sizeof(malloc_cb_t); i++) {
elessair 0:f269e3021894 89
elessair 0:f269e3021894 90 // Allocate memory and assert size change
elessair 0:f269e3021894 91 data = malloc_thunk_array[i](0);
elessair 0:f269e3021894 92 // Return can be NULL
elessair 0:f269e3021894 93 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 94 TEST_ASSERT_EQUAL_UINT32(stats_start.current_size, stats_current.current_size);
elessair 0:f269e3021894 95 TEST_ASSERT_EQUAL_UINT32(stats_start.total_size, stats_current.total_size);
elessair 0:f269e3021894 96 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_fail_cnt, stats_current.alloc_fail_cnt);
elessair 0:f269e3021894 97
elessair 0:f269e3021894 98 // Free memory and assert back to starting size
elessair 0:f269e3021894 99 free(data);
elessair 0:f269e3021894 100 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 101 TEST_ASSERT_EQUAL_UINT32(stats_start.current_size, stats_current.current_size);
elessair 0:f269e3021894 102 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_cnt, stats_current.alloc_cnt);
elessair 0:f269e3021894 103 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_fail_cnt, stats_current.alloc_fail_cnt);
elessair 0:f269e3021894 104 }
elessair 0:f269e3021894 105 }
elessair 0:f269e3021894 106
elessair 0:f269e3021894 107 void test_case_allocate_fail()
elessair 0:f269e3021894 108 {
elessair 0:f269e3021894 109 mbed_stats_heap_t stats_start;
elessair 0:f269e3021894 110 mbed_stats_heap_t stats_current;
elessair 0:f269e3021894 111 void *data;
elessair 0:f269e3021894 112
elessair 0:f269e3021894 113 mbed_stats_heap_get(&stats_start);
elessair 0:f269e3021894 114
elessair 0:f269e3021894 115 for (uint32_t i = 0; i < sizeof(malloc_thunk_array) / sizeof(malloc_cb_t); i++) {
elessair 0:f269e3021894 116
elessair 0:f269e3021894 117 // Trigger a failure by trying to allocate a buffer that won't fit
elessair 0:f269e3021894 118 data = malloc_thunk_array[i](ALLOCATION_SIZE_FAIL);
elessair 0:f269e3021894 119 TEST_ASSERT(data == NULL);
elessair 0:f269e3021894 120 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 121 TEST_ASSERT_EQUAL_UINT32(stats_start.current_size, stats_current.current_size);
elessair 0:f269e3021894 122 TEST_ASSERT_EQUAL_UINT32(stats_start.total_size, stats_current.total_size);
elessair 0:f269e3021894 123 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_cnt, stats_current.alloc_cnt);
elessair 0:f269e3021894 124 TEST_ASSERT_EQUAL_UINT32(stats_start.alloc_fail_cnt + i + 1, stats_current.alloc_fail_cnt);
elessair 0:f269e3021894 125 }
elessair 0:f269e3021894 126 }
elessair 0:f269e3021894 127
elessair 0:f269e3021894 128 static void* thunk_malloc(uint32_t size)
elessair 0:f269e3021894 129 {
elessair 0:f269e3021894 130 printf("Malloc thunk\n");
elessair 0:f269e3021894 131 return malloc(size);
elessair 0:f269e3021894 132 }
elessair 0:f269e3021894 133
elessair 0:f269e3021894 134 static void* thunk_calloc_1(uint32_t size)
elessair 0:f269e3021894 135 {
elessair 0:f269e3021894 136 printf("Calloc thunk 1 byte\n");
elessair 0:f269e3021894 137 return calloc(size / 1, 1);
elessair 0:f269e3021894 138 }
elessair 0:f269e3021894 139
elessair 0:f269e3021894 140 static void* thunk_calloc_4(uint32_t size)
elessair 0:f269e3021894 141 {
elessair 0:f269e3021894 142 printf("Calloc thunk 4 bytes\n");
elessair 0:f269e3021894 143 return calloc(size / 4, 4);
elessair 0:f269e3021894 144 }
elessair 0:f269e3021894 145
elessair 0:f269e3021894 146
elessair 0:f269e3021894 147 static void* thunk_realloc(uint32_t size)
elessair 0:f269e3021894 148 {
elessair 0:f269e3021894 149 printf("Realloc thunk\n");
elessair 0:f269e3021894 150 return realloc(NULL, size);
elessair 0:f269e3021894 151 }
elessair 0:f269e3021894 152
elessair 0:f269e3021894 153 void test_case_realloc_size()
elessair 0:f269e3021894 154 {
elessair 0:f269e3021894 155 mbed_stats_heap_t stats_start;
elessair 0:f269e3021894 156 mbed_stats_heap_t stats_current;
elessair 0:f269e3021894 157 uint32_t increase;
elessair 0:f269e3021894 158 void *data;
elessair 0:f269e3021894 159
elessair 0:f269e3021894 160 mbed_stats_heap_get(&stats_start);
elessair 0:f269e3021894 161
elessair 0:f269e3021894 162 // Allocate memory and assert size change
elessair 0:f269e3021894 163 data = realloc(NULL, ALLOCATION_SIZE_DEFAULT);
elessair 0:f269e3021894 164 TEST_ASSERT(data != NULL);
elessair 0:f269e3021894 165 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 166 increase = stats_current.current_size - stats_start.current_size;
elessair 0:f269e3021894 167 TEST_ASSERT_EQUAL_UINT32(increase, ALLOCATION_SIZE_DEFAULT);
elessair 0:f269e3021894 168
elessair 0:f269e3021894 169 // Decrease size and assert size change
elessair 0:f269e3021894 170 data = realloc(data, ALLOCATION_SIZE_SMALL);
elessair 0:f269e3021894 171 TEST_ASSERT(data != NULL);
elessair 0:f269e3021894 172 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 173 increase = stats_current.current_size - stats_start.current_size;
elessair 0:f269e3021894 174 TEST_ASSERT_EQUAL_UINT32(increase, ALLOCATION_SIZE_SMALL);
elessair 0:f269e3021894 175
elessair 0:f269e3021894 176 // Increase size and assert size change
elessair 0:f269e3021894 177 data = realloc(data, ALLOCATION_SIZE_LARGE);
elessair 0:f269e3021894 178 TEST_ASSERT(data != NULL);
elessair 0:f269e3021894 179 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 180 increase = stats_current.current_size - stats_start.current_size;
elessair 0:f269e3021894 181 TEST_ASSERT_EQUAL_UINT32(increase, ALLOCATION_SIZE_LARGE);
elessair 0:f269e3021894 182
elessair 0:f269e3021894 183 // Free memory and assert back to starting size
elessair 0:f269e3021894 184 free(data);
elessair 0:f269e3021894 185 mbed_stats_heap_get(&stats_current);
elessair 0:f269e3021894 186 TEST_ASSERT_EQUAL_UINT32(stats_start.current_size, stats_current.current_size);
elessair 0:f269e3021894 187 }
elessair 0:f269e3021894 188
elessair 0:f269e3021894 189 Case cases[] = {
elessair 0:f269e3021894 190 Case("malloc and free size", test_case_malloc_free_size),
elessair 0:f269e3021894 191 Case("allocate size zero", test_case_allocate_zero),
elessair 0:f269e3021894 192 Case("allocation failure", test_case_allocate_fail),
elessair 0:f269e3021894 193 Case("realloc size", test_case_realloc_size),
elessair 0:f269e3021894 194 };
elessair 0:f269e3021894 195
elessair 0:f269e3021894 196 utest::v1::status_t greentea_test_setup(const size_t number_of_cases)
elessair 0:f269e3021894 197 {
elessair 0:f269e3021894 198 GREENTEA_SETUP(20, "default_auto");
elessair 0:f269e3021894 199 return greentea_test_setup_handler(number_of_cases);
elessair 0:f269e3021894 200 }
elessair 0:f269e3021894 201
elessair 0:f269e3021894 202 Specification specification(greentea_test_setup, cases, greentea_test_teardown_handler);
elessair 0:f269e3021894 203
elessair 0:f269e3021894 204 int main()
elessair 0:f269e3021894 205 {
elessair 0:f269e3021894 206 Harness::run(specification);
elessair 0:f269e3021894 207 }