Add module tests for TDBStore

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 <stdlib.h>
+
+#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;
+}

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")