control-components/components/ocs_control/test/test_fsm_block_storage.cpp at 56ed7c74d32f278cf61704eafe03d8f534ac4b8b · tendry-lab/control-components · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
 * SPDX-FileCopyrightText: 2025 Tendry Lab
 * SPDX-License-Identifier: Apache-2.0
 */

#include <cstring>

#include "unity.h"

#include "ocs_control/test/test_fsm_block_storage.h"
#include "ocs_status/macros.h"

namespace ocs {
namespace control {

TestFsmBlockStorage::TestFsmBlockStorage(status::StatusCode read_status,
                                         status::StatusCode write_status,
                                         status::StatusCode erase_status)
    : read_status(read_status)
    , write_status(write_status)
    , erase_status(erase_status) {
}

status::StatusCode TestFsmBlockStorage::probe(const char* key, size_t& size) {
    return status::StatusCode::Error;
}

status::StatusCode TestFsmBlockStorage::read(const char* key, void* data, size_t size) {
    if (read_status != status::StatusCode::OK) {
        return read_status;
    }

    if (strcmp(key, "prev_state") == 0) {
        TEST_ASSERT_EQUAL(sizeof(prev_state), size);
        *static_cast<FsmBlock::State*>(data) = prev_state;
    } else if (strcmp(key, "curr_state") == 0) {
        TEST_ASSERT_EQUAL(sizeof(curr_state), size);
        *static_cast<FsmBlock::State*>(data) = curr_state;
    } else if (strcmp(key, "prev_state_dur") == 0) {
        TEST_ASSERT_EQUAL(sizeof(prev_state_duration), size);
        *static_cast<system::Time*>(data) = prev_state_duration;
    } else if (strcmp(key, "curr_state_dur") == 0) {
        TEST_ASSERT_EQUAL(sizeof(curr_state_duration), size);
        *static_cast<system::Time*>(data) = curr_state_duration;
    } else if (strcmp(key, "write_count") == 0) {
        TEST_ASSERT_EQUAL(sizeof(write_count), size);
        *static_cast<decltype(write_count)*>(data) = write_count;
    } else {
        return status::StatusCode::Error;
    }

    return status::StatusCode::OK;
}

status::StatusCode
TestFsmBlockStorage::write(const char* key, const void* data, size_t size) {
    if (write_status != status::StatusCode::OK) {
        return write_status;
    }

    if (strcmp(key, "prev_state") == 0) {
        TEST_ASSERT_EQUAL(sizeof(prev_state), size);
        prev_state = *static_cast<const FsmBlock::State*>(data);
    } else if (strcmp(key, "curr_state") == 0) {
        TEST_ASSERT_EQUAL(sizeof(curr_state), size);
        curr_state = *static_cast<const FsmBlock::State*>(data);
    } else if (strcmp(key, "prev_state_dur") == 0) {
        TEST_ASSERT_EQUAL(sizeof(prev_state_duration), size);
        prev_state_duration = *static_cast<const system::Time*>(data);
    } else if (strcmp(key, "curr_state_dur") == 0) {
        TEST_ASSERT_EQUAL(sizeof(curr_state_duration), size);
        curr_state_duration = *static_cast<const system::Time*>(data);
    } else if (strcmp(key, "write_count") == 0) {
        TEST_ASSERT_EQUAL(sizeof(write_count), size);
        write_count = *static_cast<const decltype(write_count)*>(data);
    } else {
        return status::StatusCode::Error;
    }

    return status::StatusCode::OK;
}

status::StatusCode TestFsmBlockStorage::erase(const char* key) {
    if (erase_status != status::StatusCode::OK) {
        return erase_status;
    }

    if (strcmp(key, "prev_state") == 0) {
        prev_state = 0;
    } else if (strcmp(key, "curr_state") == 0) {
        curr_state = 0;
    } else if (strcmp(key, "prev_state_dur") == 0) {
        prev_state_duration = 0;
    } else if (strcmp(key, "curr_state_dur") == 0) {
        curr_state_duration = 0;
    } else if (strcmp(key, "write_count") == 0) {
        write_count = 0;
    } else {
        return status::StatusCode::Error;
    }

    return status::StatusCode::OK;
}

status::StatusCode TestFsmBlockStorage::erase_all() {
    if (erase_status != status::StatusCode::OK) {
        return erase_status;
    }
    OCS_STATUS_RETURN_ON_ERROR(erase("prev_state"));
    OCS_STATUS_RETURN_ON_ERROR(erase("curr_state"));
    OCS_STATUS_RETURN_ON_ERROR(erase("prev_state_dur"));
    OCS_STATUS_RETURN_ON_ERROR(erase("curr_state_dur"));
    OCS_STATUS_RETURN_ON_ERROR(erase("write_count"));

    return status::StatusCode::OK;
}

} // namespace control
} // namespace ocs