/* 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 "battery.h"
#include "charge_state.h"
#include "math_util.h"
#include "test/drivers/test_state.h"

#include <zephyr/ztest.h>

int battery_outside_charging_temperature(void);

struct charge_state_fixture {
	struct charge_state_data charge_state_data;
};

static void *setup(void)
{
	static struct charge_state_fixture fixture;

	return &fixture;
}

static void before(void *f)
{
	struct charge_state_fixture *fixture = f;

	fixture->charge_state_data = *charge_get_status();
}

static void after(void *f)
{
	struct charge_state_fixture *fixture = f;

	*charge_get_status() = fixture->charge_state_data;
}

ZTEST_SUITE(charge_state, drivers_predicate_post_main, setup, before, after,
	    NULL);

ZTEST(charge_state, test_battery_flag_bad_temperature)
{
	struct charge_state_data *curr = charge_get_status();

	curr->batt.flags |= BATT_FLAG_BAD_TEMPERATURE;
	zassert_ok(battery_outside_charging_temperature());
}

ZTEST(charge_state, test_battery_temperature_range)
{
	struct charge_state_data *curr = charge_get_status();
	const struct battery_info *batt_info = battery_get_info();

	curr->batt.flags &= ~BATT_FLAG_BAD_TEMPERATURE;

	/* Start off without a desired voltage/current */
	curr->batt.desired_voltage = 0;
	curr->batt.desired_current = 0;

	/* Temperature is too high */
	curr->batt.temperature =
		CELSIUS_TO_DECI_KELVIN(batt_info->start_charging_max_c + 1);
	zassert_equal(1, battery_outside_charging_temperature());

	/* Temperature is too low */
	curr->batt.temperature =
		CELSIUS_TO_DECI_KELVIN(batt_info->start_charging_min_c - 1);
	zassert_equal(1, battery_outside_charging_temperature());

	/* Temperature is just right */
	curr->batt.temperature =
		CELSIUS_TO_DECI_KELVIN((batt_info->start_charging_max_c +
					batt_info->start_charging_min_c) /
				       2);
	zassert_ok(battery_outside_charging_temperature());

	/* Set an arbitrary desired current */
	curr->batt.desired_current = 3;

	/* Temperature is too high */
	curr->batt.temperature =
		CELSIUS_TO_DECI_KELVIN(batt_info->charging_max_c + 1);
	zassert_equal(1, battery_outside_charging_temperature());

	/* Set an arbitrary desired voltage */
	curr->batt.desired_voltage = 5;

	/* Temperature is too low */
	curr->batt.temperature =
		CELSIUS_TO_DECI_KELVIN(batt_info->charging_min_c - 1);
	zassert_equal(1, battery_outside_charging_temperature());

	/* Temperature is just right */
	curr->batt.temperature = CELSIUS_TO_DECI_KELVIN(
		(batt_info->charging_max_c + batt_info->charging_min_c) / 2);
	zassert_ok(battery_outside_charging_temperature());
}

ZTEST(charge_state, test_current_limit_derating)
{
	int charger_current_limit;

	charge_set_input_current_limit(1000, 5000);
	zassert_ok(charger_get_input_current_limit(0, &charger_current_limit));
	/*
	 * ISL923x sets ICL in multiples of 20 mA, so 950 mA gets rounded down
	 * to the nearest multiple of 20.
	 */
	zassert_equal(
		charger_current_limit, 944,
		"%d%% derating of 1A should be 944 mA, but charger is set for %d mA",
		CONFIG_PLATFORM_EC_CHARGER_INPUT_CURRENT_DERATE_PCT,
		charger_current_limit);
}

ZTEST(charge_state, test_minimum_current_limit)
{
	int charger_current_limit;

	charge_set_input_current_limit(50, 5000);
	zassert_ok(charger_get_input_current_limit(0, &charger_current_limit));
	zassert_equal(charger_current_limit, 96,
		      "Minimum input current limit should be %d mA,"
		      " but current limit is %d (capped to %d)",
		      96, charger_current_limit,
		      CONFIG_PLATFORM_EC_CHARGER_MIN_INPUT_CURRENT_LIMIT);
}