summaryrefslogtreecommitdiff
path: root/zephyr/test/drivers/default/src/console_cmd/accelread.c
blob: 81ebf87e55efca265bfc995d8fd1692526e525ab (plain)
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
120
121
122
123
124
/* Copyright 2022 The ChromiumOS Authors
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#include <zephyr/fff.h>
#include <zephyr/shell/shell.h>
#include <zephyr/ztest.h>

#include "accelgyro.h"
#include "console.h"
#include "ec_commands.h"
#include "motion_sense.h"
#include "test/drivers/test_state.h"
#include "test/drivers/utils.h"

FAKE_VALUE_FUNC(int, mock_read, const struct motion_sensor_t *, int *);
FAKE_VALUE_FUNC(int, mock_set_data_rate, const struct motion_sensor_t *, int,
		int);
FAKE_VALUE_FUNC(int, mock_get_data_rate, const struct motion_sensor_t *);

struct console_cmd_accelread_fixture {
	const struct accelgyro_drv *sensor_0_drv;
	struct accelgyro_drv mock_drv;
};

static void *console_cmd_accelread_setup(void)
{
	static struct console_cmd_accelread_fixture fixture = {
		.mock_drv = {
			.read = mock_read,
			/*
			 * Data rate functions are required so that motion_sense
			 * task doesn't segfault.
			 */
			.set_data_rate = mock_set_data_rate,
			.get_data_rate = mock_get_data_rate,
		},
	};
	fixture.sensor_0_drv = motion_sensors[0].drv;

	return &fixture;
}

static void console_cmd_accelread_before(void *fixture)
{
	ARG_UNUSED(fixture);
	RESET_FAKE(mock_read);
	RESET_FAKE(mock_set_data_rate);
	RESET_FAKE(mock_get_data_rate);
	FFF_RESET_HISTORY();
}

static void console_cmd_accelread_after(void *fixture)
{
	struct console_cmd_accelread_fixture *this = fixture;

	motion_sensors[0].drv = this->sensor_0_drv;
}

ZTEST_SUITE(console_cmd_accelread, drivers_predicate_pre_main,
	    console_cmd_accelread_setup, console_cmd_accelread_before,
	    console_cmd_accelread_after, NULL);

ZTEST_USER(console_cmd_accelread, test_too_few_arguments)
{
	int rv = shell_execute_cmd(get_ec_shell(), "accelread");

	zassert_equal(EC_ERROR_PARAM_COUNT, rv, "Expected %d, but got %d",
		      EC_ERROR_PARAM_COUNT, rv);
}

ZTEST_USER(console_cmd_accelread, test_invalid_sensor_num)
{
	int rv;

	rv = shell_execute_cmd(get_ec_shell(), "accelread f");
	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
		      EC_ERROR_PARAM1, rv);

	rv = shell_execute_cmd(get_ec_shell(), "accelread -1");
	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
		      EC_ERROR_PARAM1, rv);

	rv = shell_execute_cmd(get_ec_shell(), "accelread 100");
	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
		      EC_ERROR_PARAM1, rv);
}

static struct console_cmd_accelread_fixture *current_fixture;

int mock_read_call_super(const struct motion_sensor_t *s, int *v)
{
	return current_fixture->sensor_0_drv->read(s, v);
}

ZTEST_USER_F(console_cmd_accelread, test_read)
{
	current_fixture = fixture;
	mock_read_fake.custom_fake = mock_read_call_super;
	mock_get_data_rate_fake.return_val = 100;
	motion_sensors[0].drv = &fixture->mock_drv;

	zassert_ok(shell_execute_cmd(get_ec_shell(), "accelread 0"), NULL);
	zassert_equal(1, mock_read_fake.call_count,
		      "Expected only 1 call to read, but got %d",
		      mock_read_fake.call_count);

	zassert_ok(shell_execute_cmd(get_ec_shell(), "accelread 0 2"), NULL);
	zassert_equal(3, mock_read_fake.call_count,
		      "Expected only 3 call to read, but got %d",
		      mock_read_fake.call_count);
}

ZTEST_USER_F(console_cmd_accelread, test_read_fail)
{
	mock_read_fake.return_val = 1;
	motion_sensors[0].drv = &fixture->mock_drv;

	zassert_ok(shell_execute_cmd(get_ec_shell(), "accelread 0"), NULL);
	zassert_equal(1, mock_read_fake.call_count,
		      "Expected only 1 call to read, but got %d",
		      mock_read_fake.call_count);
}