mbed-os/TESTS/filesystem/parallel/main.cpp

#include "mbed.h"
#include "greentea-client/test_env.h"
#include "unity.h"
#include "utest.h"
#include <stdlib.h>
#include <errno.h>

using namespace utest::v1;

// test configuration
#ifndef MBED_TEST_FILESYSTEM
#define MBED_TEST_FILESYSTEM LittleFileSystem
#endif

#ifndef MBED_TEST_FILESYSTEM_DECL
#define MBED_TEST_FILESYSTEM_DECL MBED_TEST_FILESYSTEM fs("fs")
#endif

#ifndef MBED_TEST_BLOCKDEVICE
#define MBED_TEST_BLOCKDEVICE SPIFBlockDevice
#define MBED_TEST_BLOCKDEVICE_DECL SPIFBlockDevice bd(PTE2, PTE4, PTE1, PTE5)
#endif

#ifndef MBED_TEST_BLOCKDEVICE_DECL
#define MBED_TEST_BLOCKDEVICE_DECL MBED_TEST_BLOCKDEVICE bd
#endif

#ifndef MBED_TEST_FILES
#define MBED_TEST_FILES 4
#endif

#ifndef MBED_TEST_DIRS
#define MBED_TEST_DIRS 4
#endif

#ifndef MBED_TEST_BUFFER
#define MBED_TEST_BUFFER 8192
#endif

#ifndef MBED_TEST_TIMEOUT
#define MBED_TEST_TIMEOUT 120
#endif


// declarations
#define STRINGIZE(x) STRINGIZE2(x)
#define STRINGIZE2(x) #x
#define INCLUDE(x) STRINGIZE(x.h)

#include INCLUDE(MBED_TEST_FILESYSTEM)
#include INCLUDE(MBED_TEST_BLOCKDEVICE)

MBED_TEST_FILESYSTEM_DECL;
MBED_TEST_BLOCKDEVICE_DECL;

Dir dir[MBED_TEST_DIRS];
File file[MBED_TEST_FILES];
DIR *dd[MBED_TEST_DIRS];
FILE *fd[MBED_TEST_FILES];
struct dirent ent;
struct dirent *ed;
size_t size;
uint8_t buffer[MBED_TEST_BUFFER];
uint8_t rbuffer[MBED_TEST_BUFFER];
uint8_t wbuffer[MBED_TEST_BUFFER];


// tests

void test_parallel_tests() {
    int res = bd.init();
    TEST_ASSERT_EQUAL(0, res);

    {
        res = MBED_TEST_FILESYSTEM::format(&bd);
        TEST_ASSERT_EQUAL(0, res);
    }

    res = bd.deinit();
    TEST_ASSERT_EQUAL(0, res);
}

void test_parallel_file_test() {
    int res = bd.init();
    TEST_ASSERT_EQUAL(0, res);

    {
        res = fs.mount(&bd);
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "a", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
        res = file[1].open(&fs, "b", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
        res = file[2].open(&fs, "c", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
        res = file[3].open(&fs, "d", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 10; i++) {
            res = file[0].write((const void*)"a", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[1].write((const void*)"b", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[2].write((const void*)"c", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[3].write((const void*)"d", 1);
            TEST_ASSERT_EQUAL(1, res);
        }
    
        file[0].close();
        file[1].close();
        file[2].close();
        file[3].close();
        res = dir[0].open(&fs, "/");
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, ".");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "..");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "a");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "b");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "c");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "d");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].close();
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "a", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
        res = file[1].open(&fs, "b", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
        res = file[2].open(&fs, "c", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
        res = file[3].open(&fs, "d", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 10; i++) {
            res = file[0].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('a', res);
            res = file[1].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('b', res);
            res = file[2].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('c', res);
            res = file[3].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('d', res);
        }
    
        file[0].close();
        file[1].close();
        file[2].close();
        file[3].close();
        res = fs.unmount();
        TEST_ASSERT_EQUAL(0, res);
    }

    res = bd.deinit();
    TEST_ASSERT_EQUAL(0, res);
}

void test_parallel_remove_file_test() {
    int res = bd.init();
    TEST_ASSERT_EQUAL(0, res);

    {
        res = fs.mount(&bd);
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "e", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 5; i++) {
            res = file[0].write((const void*)"e", 1);
            TEST_ASSERT_EQUAL(1, res);
        }
        res = fs.remove("a");
        TEST_ASSERT_EQUAL(0, res);
        res = fs.remove("b");
        TEST_ASSERT_EQUAL(0, res);
        res = fs.remove("c");
        TEST_ASSERT_EQUAL(0, res);
        res = fs.remove("d");
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 5; i++) {
            res = file[0].write((const void*)"e", 1);
            TEST_ASSERT_EQUAL(1, res);
        }
    
        file[0].close();
        res = dir[0].open(&fs, "/");
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, ".");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "..");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "e");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].close();
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "e", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 10; i++) {
            res = file[0].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('e', res);
        }
    
        file[0].close();
        res = fs.unmount();
        TEST_ASSERT_EQUAL(0, res);
    }

    res = bd.deinit();
    TEST_ASSERT_EQUAL(0, res);
}

void test_remove_inconveniently_test() {
    int res = bd.init();
    TEST_ASSERT_EQUAL(0, res);

    {
        res = fs.mount(&bd);
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "e", O_WRONLY | O_TRUNC);
        TEST_ASSERT_EQUAL(0, res);
        res = file[1].open(&fs, "f", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
        res = file[2].open(&fs, "g", O_WRONLY | O_CREAT);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 5; i++) {
            res = file[0].write((const void*)"e", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[1].write((const void*)"f", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[2].write((const void*)"g", 1);
            TEST_ASSERT_EQUAL(1, res);
        }
        res = fs.remove("f");
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 5; i++) {
            res = file[0].write((const void*)"e", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[1].write((const void*)"f", 1);
            TEST_ASSERT_EQUAL(1, res);
            res = file[2].write((const void*)"g", 1);
            TEST_ASSERT_EQUAL(1, res);
        }
    
        file[0].close();
        file[1].close();
        file[2].close();
        res = dir[0].open(&fs, "/");
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, ".");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "..");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_DIR, res);
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "e");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(1, res);
        res = strcmp(ent.d_name, "g");
        TEST_ASSERT_EQUAL(0, res);
        res = ent.d_type;
        TEST_ASSERT_EQUAL(DT_REG, res);
    
        res = dir[0].read(&ent);
        TEST_ASSERT_EQUAL(0, res);
        res = dir[0].close();
        TEST_ASSERT_EQUAL(0, res);
        res = file[0].open(&fs, "e", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
        res = file[1].open(&fs, "g", O_RDONLY);
        TEST_ASSERT_EQUAL(0, res);
    
        for (int i = 0; i < 10; i++) {
            res = file[0].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('e', res);
            res = file[1].read(buffer, 1);
            TEST_ASSERT_EQUAL(1, res);
            res = buffer[0];
            TEST_ASSERT_EQUAL('g', res);
        }
    
        file[0].close();
        file[1].close();
        res = fs.unmount();
        TEST_ASSERT_EQUAL(0, res);
    }

    res = bd.deinit();
    TEST_ASSERT_EQUAL(0, res);
}



// test setup
utest::v1::status_t test_setup(const size_t number_of_cases) {
    GREENTEA_SETUP(MBED_TEST_TIMEOUT, "default_auto");
    return verbose_test_setup_handler(number_of_cases);
}

Case cases[] = {
    Case("Parallel tests", test_parallel_tests),
    Case("Parallel file test", test_parallel_file_test),
    Case("Parallel remove file test", test_parallel_remove_file_test),
    Case("Remove inconveniently test", test_remove_inconveniently_test),
};

Specification specification(test_setup, cases);

int main() {
    return !Harness::run(specification);
}