package arm

import (
	"context"
	"fmt"
	"testing"

	"github.com/hashicorp/packer/builder/azure/common/constants"
	"github.com/hashicorp/packer/helper/multistep"
)

func TestStepGetIPAddressShouldFailIfGetFails(t *testing.T) {
	endpoints := []EndpointType{PublicEndpoint, PublicEndpointInPrivateNetwork}

	for _, endpoint := range endpoints {
		var testSubject = &StepGetIPAddress{
			get: func(context.Context, string, string, string) (string, error) {
				return "", fmt.Errorf("!! Unit Test FAIL !!")
			},
			endpoint: endpoint,
			say:      func(message string) {},
			error:    func(e error) {},
		}

		stateBag := createTestStateBagStepGetIPAddress()

		var result = testSubject.Run(context.Background(), stateBag)
		if result != multistep.ActionHalt {
			t.Fatalf("Expected the step to return 'ActionHalt', but got '%d'.", result)
		}

		if _, ok := stateBag.GetOk(constants.Error); ok == false {
			t.Fatalf("Expected the step to set stateBag['%s'], but it was not.", constants.Error)
		}
	}
}

func TestStepGetIPAddressShouldPassIfGetPasses(t *testing.T) {
	endpoints := []EndpointType{PublicEndpoint, PublicEndpointInPrivateNetwork}

	for _, endpoint := range endpoints {
		var testSubject = &StepGetIPAddress{
			get:      func(context.Context, string, string, string) (string, error) { return "", nil },
			endpoint: endpoint,
			say:      func(message string) {},
			error:    func(e error) {},
		}

		stateBag := createTestStateBagStepGetIPAddress()

		var result = testSubject.Run(context.Background(), stateBag)
		if result != multistep.ActionContinue {
			t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result)
		}

		if _, ok := stateBag.GetOk(constants.Error); ok == true {
			t.Fatalf("Expected the step to not set stateBag['%s'], but it was.", constants.Error)
		}
	}
}

func TestStepGetIPAddressShouldTakeStepArgumentsFromStateBag(t *testing.T) {
	var actualResourceGroupName string
	var actualIPAddressName string
	var actualNicName string
	endpoints := []EndpointType{PublicEndpoint, PublicEndpointInPrivateNetwork}

	for _, endpoint := range endpoints {
		var testSubject = &StepGetIPAddress{
			get: func(ctx context.Context, resourceGroupName string, ipAddressName string, nicName string) (string, error) {
				actualResourceGroupName = resourceGroupName
				actualIPAddressName = ipAddressName
				actualNicName = nicName

				return "127.0.0.1", nil
			},
			endpoint: endpoint,
			say:      func(message string) {},
			error:    func(e error) {},
		}

		stateBag := createTestStateBagStepGetIPAddress()
		var result = testSubject.Run(context.Background(), stateBag)

		if result != multistep.ActionContinue {
			t.Fatalf("Expected the step to return 'ActionContinue', but got '%d'.", result)
		}

		var expectedResourceGroupName = stateBag.Get(constants.ArmResourceGroupName).(string)
		var expectedIPAddressName = stateBag.Get(constants.ArmPublicIPAddressName).(string)
		var expectedNicName = stateBag.Get(constants.ArmNicName).(string)

		if actualIPAddressName != expectedIPAddressName {
			t.Fatal("Expected StepGetIPAddress to source 'constants.ArmIPAddressName' from the state bag, but it did not.")
		}

		if actualResourceGroupName != expectedResourceGroupName {
			t.Fatal("Expected StepGetIPAddress to source 'constants.ArmResourceGroupName' from the state bag, but it did not.")
		}

		if actualNicName != expectedNicName {
			t.Fatalf("Expected StepGetIPAddress to source 'constants.ArmNetworkInterfaceName' from the state bag, but it did not.")
		}

		expectedIPAddress, ok := stateBag.GetOk(constants.SSHHost)
		if !ok {
			t.Fatalf("Expected the state bag to have a value for '%s', but it did not.", constants.SSHHost)
		}

		if expectedIPAddress != "127.0.0.1" {
			t.Fatalf("Expected the value of stateBag[%s] to be '127.0.0.1', but got '%s'.", constants.SSHHost, expectedIPAddress)
		}
	}
}

func createTestStateBagStepGetIPAddress() multistep.StateBag {
	stateBag := new(multistep.BasicStateBag)

	stateBag.Put(constants.ArmPublicIPAddressName, "Unit Test: PublicIPAddressName")
	stateBag.Put(constants.ArmNicName, "Unit Test: NicName")
	stateBag.Put(constants.ArmResourceGroupName, "Unit Test: ResourceGroupName")

	return stateBag
}
