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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/* 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 <ztest.h>
#include "charge_manager.h"
#include "console.h"
#include "emul/emul_isl923x.h"
#include "emul/tcpc/emul_tcpci_partner_snk.h"
#include "tcpm/tcpci.h"
#include "test/drivers/test_state.h"
#include "test/drivers/utils.h"
static void connect_sink_to_port(const struct emul *charger_emul,
const struct emul *tcpci_emul,
struct tcpci_partner_data *partner)
{
isl923x_emul_set_adc_vbus(charger_emul, 0);
tcpci_emul_set_reg(tcpci_emul, TCPC_REG_POWER_STATUS,
TCPC_REG_POWER_STATUS_VBUS_DET);
tcpci_emul_set_reg(tcpci_emul, TCPC_REG_EXT_STATUS,
TCPC_REG_EXT_STATUS_SAFE0V);
tcpci_tcpc_alert(0);
zassume_ok(tcpci_partner_connect_to_tcpci(partner, tcpci_emul),
NULL);
/* Wait for PD negotiation and current ramp.
* TODO(b/213906889): Check message timing and contents.
*/
k_sleep(K_SECONDS(10));
}
static inline void disconnect_sink_from_port(const struct emul *tcpci_emul)
{
zassume_ok(tcpci_emul_disconnect_partner(tcpci_emul), NULL);
k_sleep(K_SECONDS(1));
}
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_binding(DT_LABEL(DT_NODELABEL(tcpci_emul)));
test_fixture.charger_emul =
emul_get_binding(DT_LABEL(DT_NODELABEL(isl923x_emul)));
tcpci_emul_set_rev(test_fixture.tcpci_emul, TCPCI_EMUL_REV2_0_VER1_1);
/* 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(this->charger_emul, this->tcpci_emul,
&this->sink_5v_3a);
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);
}
|
