package core

import (
	"testing"

	"github.com/evcc-io/evcc/util"
)

func TestSitePower(t *testing.T) {
	tc := []struct {
		maxGrid, grid, battery, site float64
	}{
		// {0, 0, 0, 0},    // silent night
		// {0, 0, 1, 1},    // battery discharging
		// {0, 0, -1, -1},  // battery charging -> negative result cannot occur in reality
		// {0, 1, 0, 1},    // grid import
		// {0, 1, 1, 2},    // grid import + battery discharging
		// {0, -1, 0, -1},  // grid export
		// {0, -1, -1, -2}, // grid export + battery charging
		{0, 1, -1, 0},   // grid import + battery charging -> should not happen
		{0.5, 1, -1, 1}, // grid import + DC battery charging
	}

	log := util.NewLogger("foo")

	for _, tc := range tc {
		res := sitePower(log, tc.maxGrid, tc.grid, tc.battery, 0)
		if res != tc.site {
			t.Errorf("sitePower wanted %.f, got %.f", tc.site, res)
		}
	}
}

func TestGreenShare(t *testing.T) {
	tc := []struct {
		title             string
		grid, pv, battery float64
		share             float64
	}{
		{"half grid, half pv",
			2500, 2500, 0,
			0.5},
		{"full pv",
			0, 5000, 0,
			1},
		{"full grid",
			5000, 0, 0,
			0},
		{"half grid, half battery",
			2500, 0, 2500,
			0.5},
		{"full pv, pv export",
			-5000, 10000, 0,
			1},
		{"full pv, pv export, battery charge",
			-2500, 10000, -2500,
			1},
		{"double charge speed, full grid",
			10000, 0, 0,
			0},
	}

	for _, tc := range tc {
		t.Logf("%+v", tc)

		s := &Site{
			gridPower:    tc.grid,
			pvPower:      tc.pv,
			batteryPower: tc.battery,
		}

		share := s.greenShare()
		if share != tc.share {
			t.Errorf("greenShare wanted %.f, got %.f", tc.share, share)
		}
	}
}