/* 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 <ztest.h>

#include "charge_manager.h"
#include "ec_commands.h"
#include "test/drivers/test_state.h"

ZTEST_SUITE(charge_manager, drivers_predicate_post_main, NULL, NULL, NULL,
	    NULL);

/**
 * Test the default implementation of board_fill_source_power_info(). The fill
 * function should reset all the power info values. If the test binary overrides
 * board_fill_source_power_info(), then this test can be removed.
 */
ZTEST_USER(charge_manager, test_default_fill_power_info)
{
	struct ec_response_usb_pd_power_info info = {
		.meas = {
			.voltage_now = 10,
			.voltage_max = 10,
			.current_max = 10,
			.current_lim = 10,
		},
		.max_power = 10,
	};

	board_fill_source_power_info(0, &info);
	zassert_equal(info.meas.voltage_now, 0, NULL);
	zassert_equal(info.meas.voltage_max, 0, NULL);
	zassert_equal(info.meas.current_max, 0, NULL);
	zassert_equal(info.meas.current_lim, 0, NULL);
	zassert_equal(info.max_power, 0, NULL);
}

/**
 * Test the default implementation of board_charge_port_is_connected(). This
 * function should always return 1 regardless of input.
 */
ZTEST_USER(charge_manager, test_default_charge_port_is_connected)
{
	zassert_true(board_charge_port_is_connected(-1), NULL);
	zassert_true(board_charge_port_is_connected(0), NULL);
	zassert_true(board_charge_port_is_connected(1), NULL);
	zassert_true(board_charge_port_is_connected(500), NULL);
}

ZTEST_USER(charge_manager, test_default_charge_port_is_sink)
{
	zassert_true(board_charge_port_is_sink(-1), NULL);
	zassert_true(board_charge_port_is_sink(0), NULL);
	zassert_true(board_charge_port_is_sink(1), NULL);
	zassert_true(board_charge_port_is_sink(500), NULL);
}