package validate

import (
	"context"
	"testing"

	"github.com/databricks/cli/bundle"
	"github.com/databricks/cli/bundle/config"
	"github.com/databricks/cli/bundle/config/resources"
	"github.com/databricks/cli/libs/diag"
	"github.com/databricks/databricks-sdk-go/service/jobs"
	"github.com/stretchr/testify/require"
)

func TestJobClusterKeyDefined(t *testing.T) {
	b := &bundle.Bundle{
		Config: config.Root{
			Resources: config.Resources{
				Jobs: map[string]*resources.Job{
					"job1": {
						JobSettings: &jobs.JobSettings{
							Name: "job1",
							JobClusters: []jobs.JobCluster{
								{JobClusterKey: "do-not-exist"},
							},
							Tasks: []jobs.Task{
								{JobClusterKey: "do-not-exist"},
							},
						},
					},
				},
			},
		},
	}

	diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
	require.Len(t, diags, 0)
	require.NoError(t, diags.Error())
}

func TestJobClusterKeyNotDefined(t *testing.T) {
	b := &bundle.Bundle{
		Config: config.Root{
			Resources: config.Resources{
				Jobs: map[string]*resources.Job{
					"job1": {
						JobSettings: &jobs.JobSettings{
							Name: "job1",
							Tasks: []jobs.Task{
								{JobClusterKey: "do-not-exist"},
							},
						},
					},
				},
			},
		},
	}

	diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
	require.Len(t, diags, 1)
	require.NoError(t, diags.Error())
	require.Equal(t, diags[0].Severity, diag.Warning)
	require.Equal(t, diags[0].Summary, "job_cluster_key do-not-exist is not defined")
}

func TestJobClusterKeyDefinedInDifferentJob(t *testing.T) {
	b := &bundle.Bundle{
		Config: config.Root{
			Resources: config.Resources{
				Jobs: map[string]*resources.Job{
					"job1": {
						JobSettings: &jobs.JobSettings{
							Name: "job1",
							Tasks: []jobs.Task{
								{JobClusterKey: "do-not-exist"},
							},
						},
					},
					"job2": {
						JobSettings: &jobs.JobSettings{
							Name: "job2",
							JobClusters: []jobs.JobCluster{
								{JobClusterKey: "do-not-exist"},
							},
						},
					},
				},
			},
		},
	}

	diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
	require.Len(t, diags, 1)
	require.NoError(t, diags.Error())
	require.Equal(t, diags[0].Severity, diag.Warning)
	require.Equal(t, diags[0].Summary, "job_cluster_key do-not-exist is not defined")
}