From 35410a9b538e5d33777f4ef8d757d2e176bf102e Mon Sep 17 00:00:00 2001 From: Seppo Takalo Date: Tue, 19 Nov 2019 15:54:51 +0200 Subject: [PATCH] Add module tests for TDBStore --- .../storage/kvstore/TDBStore/moduletest.cpp | 126 ++++++++++++++++++ .../storage/kvstore/TDBStore/unittest.cmake | 27 ++++ 2 files changed, 153 insertions(+) create mode 100644 UNITTESTS/moduletests/storage/kvstore/TDBStore/moduletest.cpp create mode 100644 UNITTESTS/moduletests/storage/kvstore/TDBStore/unittest.cmake diff --git a/UNITTESTS/moduletests/storage/kvstore/TDBStore/moduletest.cpp b/UNITTESTS/moduletests/storage/kvstore/TDBStore/moduletest.cpp new file mode 100644 index 0000000000..15c1f4fb16 --- /dev/null +++ b/UNITTESTS/moduletests/storage/kvstore/TDBStore/moduletest.cpp @@ -0,0 +1,126 @@ +/* Copyright (c) 2019 ARM Limited + * SPDX-License-Identifier: Apache-2.0 + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "gtest/gtest.h" +#include "features/storage/blockdevice/HeapBlockDevice.h" +#include "features/storage/blockdevice/FlashSimBlockDevice.h" +#include "features/storage/kvstore/tdbstore/TDBStore.h" +#include + +#define BLOCK_SIZE (512) +#define DEVICE_SIZE (BLOCK_SIZE*200) + +using namespace mbed; + +class TDBStoreModuleTest : public testing::Test { +protected: + HeapBlockDevice heap{DEVICE_SIZE}; + FlashSimBlockDevice flash{&heap}; + TDBStore tdb{&flash}; + + virtual void SetUp() + { + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + EXPECT_EQ(tdb.reset(), MBED_SUCCESS); + } + + virtual void TearDown() + { + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + } +}; + +TEST_F(TDBStoreModuleTest, init) +{ + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + EXPECT_EQ(tdb.init(), MBED_SUCCESS); +} + +TEST_F(TDBStoreModuleTest, set_get) +{ + char buf[100]; + size_t size; + EXPECT_EQ(tdb.set("key", "data", 5, 0), MBED_SUCCESS); + EXPECT_EQ(tdb.get("key", buf, 100, &size), MBED_SUCCESS); + EXPECT_EQ(size, 5); + EXPECT_STREQ("data", buf); +} + +TEST_F(TDBStoreModuleTest, erased_get_set) +{ + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + EXPECT_EQ(flash.init(), MBED_SUCCESS); + EXPECT_EQ(flash.erase(0, flash.size()), MBED_SUCCESS); + EXPECT_EQ(flash.deinit(), MBED_SUCCESS); + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + char buf[100]; + size_t size; + EXPECT_EQ(tdb.set("key", "data", 5, 0), MBED_SUCCESS); + EXPECT_EQ(tdb.get("key", buf, 100, &size), MBED_SUCCESS); + EXPECT_EQ(size, 5); + EXPECT_STREQ("data", buf); +} + +TEST_F(TDBStoreModuleTest, set_deinit_init_get) +{ + char buf[100]; + size_t size; + for (int i = 0; i < 100; ++i) { + EXPECT_EQ(tdb.set("key", "data", 5, 0), MBED_SUCCESS); + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + EXPECT_EQ(tdb.get("key", buf, 100, &size), MBED_SUCCESS); + EXPECT_EQ(size, 5); + EXPECT_STREQ("data", buf); + EXPECT_EQ(tdb.remove("key"), MBED_SUCCESS); + } +} + +TEST_F(TDBStoreModuleTest, corrupted_set_deinit_init_get) +{ + char buf[100]; + char *block = new char[BLOCK_SIZE]; + size_t size; + EXPECT_EQ(heap.init(), MBED_SUCCESS); // Extra init, so the heap will not be deinitialized + + srand(0); // Prefer to have always the same pattern + + for (int i = 0; i < 100; ++i) { + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + // Corrupt the first part of the storage + for (int i = 0; i < heap.size()/BLOCK_SIZE/2; i++) { + for (int j = 0; j < BLOCK_SIZE; j++) { + block[j] = rand(); + } + EXPECT_EQ(heap.program(block, BLOCK_SIZE * i, BLOCK_SIZE), MBED_SUCCESS); + } + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + for (int j = 0; j < 100; ++j) { + // Use random data, so the data has to be updated + EXPECT_EQ(tdb.set("key", block+j, 50, 0), MBED_SUCCESS); + EXPECT_EQ(tdb.deinit(), MBED_SUCCESS); + EXPECT_EQ(tdb.init(), MBED_SUCCESS); + EXPECT_EQ(tdb.get("key", buf, 100, &size), MBED_SUCCESS); + EXPECT_EQ(size, 50); + EXPECT_EQ(0, memcmp(buf, block+j, size)); + } + EXPECT_EQ(tdb.remove("key"), MBED_SUCCESS); + } + + EXPECT_EQ(heap.deinit(), MBED_SUCCESS); + delete[] block; +} diff --git a/UNITTESTS/moduletests/storage/kvstore/TDBStore/unittest.cmake b/UNITTESTS/moduletests/storage/kvstore/TDBStore/unittest.cmake new file mode 100644 index 0000000000..198df4988c --- /dev/null +++ b/UNITTESTS/moduletests/storage/kvstore/TDBStore/unittest.cmake @@ -0,0 +1,27 @@ + +#################### +# UNIT TESTS +#################### + +set(unittest-includes ${unittest-includes} + . + .. + ../features/frameworks/mbed-trace/mbed-trace +) + +set(unittest-sources + ../features/storage/blockdevice/FlashSimBlockDevice.cpp + ../features/storage/blockdevice/HeapBlockDevice.cpp + ../features/storage/blockdevice/BufferedBlockDevice.cpp + ../features/storage/kvstore/tdbstore/TDBStore.cpp + ../features/frameworks/mbed-trace/source/mbed_trace.c + stubs/mbed_atomic_stub.c + stubs/mbed_assert_stub.cpp + stubs/mbed_error.c +) + +set(unittest-test-sources + moduletests/storage/kvstore/TDBStore/moduletest.cpp +) + +set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DBYPASS_NVSTORE_CHECK")