// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

package dockerstatsreceiver

import (
	"errors"
	"testing"

	ctypes "github.com/docker/docker/api/types/container"
	"github.com/stretchr/testify/assert"
)

func Test_calculateCPULimit1(t *testing.T) {
	tests := []struct {
		name string
		args *ctypes.HostConfig
		want float64
		err  error
	}{
		{
			"Test CPULimit",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{
					NanoCPUs: 2500000000,
				},
			},
			2.5,
			nil,
		},
		{
			"Test CPUSetCpu",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{
					CpusetCpus: "0-2",
				},
			},
			3,
			nil,
		},
		{
			"Test CPUQuota",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{
					CPUQuota: 50000,
				},
			},
			0.5,
			nil,
		},
		{
			"Test CPUQuota Custom Period",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{
					CPUQuota:  300000,
					CPUPeriod: 200000,
				},
			},
			1.5,
			nil,
		},
		{
			"Test Default",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{
					NanoCPUs:   1800000000,
					CpusetCpus: "0-1",
					CPUQuota:   400000,
				},
			},
			1.8,
			nil,
		},
		{
			"Test No Values",
			&ctypes.HostConfig{
				Resources: ctypes.Resources{},
			},
			0,
			nil,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			want, err := calculateCPULimit(tt.args)
			assert.Equalf(t, tt.want, want, "calculateCPULimit(%v)", tt.args)
			assert.Equalf(t, tt.err, err, "calculateCPULimit(%v)", tt.args)
		})
	}
}

func Test_parseCPUSet(t *testing.T) {
	tests := []struct {
		input    string
		expected float64
		err      error
	}{
		{"0,2", 2, nil},
		{"0-2", 3, nil},
		{"0-2,4", 4, nil},
		{"0-2,4-5", 5, nil},
		{"a-b", 0, errors.New("invalid -cpuset-cpus value: strconv.Atoi: parsing \"a\": invalid syntax")},
		{"", 1, nil},
	}

	for _, test := range tests {
		result, err := parseCPUSet(test.input)

		if err != nil && test.err != nil {
			if err.Error() != test.err.Error() {
				t.Errorf("parseCPUSet(%s) returned error %v, expected %v", test.input, err, test.err)
			}
		} else if !errors.Is(err, test.err) {
			t.Errorf("parseCPUSet(%s) returned error %v, expected %v", test.input, err, test.err)
		}

		if result != test.expected {
			t.Errorf("parseCPUSet(%s) returned %f, expected %f", test.input, result, test.expected)
		}
	}
}