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
|
/* Copyright 2022 The Chromium OS Authors. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include <zephyr/shell/shell.h>
#include <zephyr/ztest.h>
#include "charge_manager.h"
#include "console.h"
#include "emul/emul_isl923x.h"
#include "emul/tcpc/emul_tcpci.h"
#include "emul/tcpc/emul_tcpci_partner_snk.h"
#include "tcpm/tcpci.h"
#include "test/drivers/test_state.h"
#include "test/drivers/utils.h"
struct console_cmd_charge_manager_fixture {
struct tcpci_partner_data sink_5v_3a;
struct tcpci_snk_emul_data sink_ext;
const struct emul *tcpci_emul;
const struct emul *charger_emul;
};
static void *console_cmd_charge_manager_setup(void)
{
static struct console_cmd_charge_manager_fixture test_fixture;
/* Get references for the emulators */
test_fixture.tcpci_emul = EMUL_GET_USBC_BINDING(0, tcpc);
test_fixture.charger_emul = EMUL_GET_USBC_BINDING(0, chg);
/* Initialized the sink to request 5V and 3A */
tcpci_partner_init(&test_fixture.sink_5v_3a, PD_REV20);
test_fixture.sink_5v_3a.extensions = tcpci_snk_emul_init(
&test_fixture.sink_ext, &test_fixture.sink_5v_3a, NULL);
test_fixture.sink_ext.pdo[1] =
PDO_FIXED(5000, 3000, PDO_FIXED_UNCONSTRAINED);
return &test_fixture;
}
static void console_cmd_charge_manager_after(void *state)
{
struct console_cmd_charge_manager_fixture *fixture = state;
shell_execute_cmd(get_ec_shell(), "chgoverride -1");
disconnect_sink_from_port(fixture->tcpci_emul);
}
ZTEST_SUITE(console_cmd_charge_manager, drivers_predicate_post_main,
console_cmd_charge_manager_setup, NULL,
console_cmd_charge_manager_after, NULL);
/**
* Test the chgsup (charge supplier info) command. This command only prints to
* console some information which is not yet possible to verify. So just check
* that the console command ran successfully.
*/
ZTEST_USER(console_cmd_charge_manager, test_chgsup)
{
zassert_ok(shell_execute_cmd(get_ec_shell(), "chgsup"), NULL);
}
/**
* Test chgoverride command with no arguments. This should just print the
* current override port.
*/
ZTEST_USER(console_cmd_charge_manager, test_chgoverride_missing_port)
{
zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride"), NULL);
}
ZTEST_USER(console_cmd_charge_manager, test_chgoverride_off_from_off)
{
zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride -1"), NULL);
zassert_equal(charge_manager_get_override(), OVERRIDE_OFF, NULL);
}
ZTEST_USER(console_cmd_charge_manager, test_chgoverride_disable_from_off)
{
zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride -2"), NULL);
zassert_equal(charge_manager_get_override(), OVERRIDE_DONT_CHARGE,
NULL);
}
ZTEST_USER(console_cmd_charge_manager, test_chgoverride_0_from_off)
{
zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride 0"), NULL);
zassert_equal(charge_manager_get_override(), 0, NULL);
}
ZTEST_USER_F(console_cmd_charge_manager, test_chgoverride_0_from_sink)
{
test_set_chipset_to_g3();
k_sleep(K_SECONDS(1));
/* Set chipset to ON, this will set TCPM to DRP */
test_set_chipset_to_s0();
/* TODO(b/214401892): Check why need to give time TCPM to spin */
k_sleep(K_SECONDS(1));
connect_sink_to_port(&fixture->sink_5v_3a, fixture->tcpci_emul,
fixture->charger_emul);
zassert_equal(shell_execute_cmd(get_ec_shell(), "chgoverride 0"),
EC_ERROR_INVAL, NULL);
}
ZTEST_USER(console_cmd_charge_manager, test_chgoverride_invalid_port)
{
char cmd[256];
zassume_true(sprintf(cmd, "chgoverride %d", CHARGE_PORT_COUNT) > 0,
NULL);
zassert_equal(shell_execute_cmd(get_ec_shell(), cmd), EC_ERROR_PARAM1,
NULL);
}
|