package storage

import (
	"errors"
	"testing"

	gomock "github.com/golang/mock/gomock"
	"github.com/qdm12/gluetun/internal/models"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func Test_extractServersFromBytes(t *testing.T) {
	t.Parallel()

	testCases := map[string]struct {
		b         []byte
		hardcoded models.AllServers
		logged    []string
		persisted models.AllServers
		err       error
	}{
		"no data": {
			err: errors.New("cannot decode versions: unexpected end of JSON input"),
		},
		"empty JSON": {
			b:   []byte("{}"),
			err: errors.New("cannot decode servers for provider: Cyberghost: unexpected end of JSON input"),
		},
		"different versions": {
			b: []byte(`{}`),
			hardcoded: models.AllServers{
				Cyberghost:   models.CyberghostServers{Version: 1},
				Expressvpn:   models.ExpressvpnServers{Version: 1},
				Fastestvpn:   models.FastestvpnServers{Version: 1},
				HideMyAss:    models.HideMyAssServers{Version: 1},
				Ipvanish:     models.IpvanishServers{Version: 1},
				Ivpn:         models.IvpnServers{Version: 1},
				Mullvad:      models.MullvadServers{Version: 1},
				Nordvpn:      models.NordvpnServers{Version: 1},
				Privado:      models.PrivadoServers{Version: 1},
				Pia:          models.PiaServers{Version: 1},
				Privatevpn:   models.PrivatevpnServers{Version: 1},
				Protonvpn:    models.ProtonvpnServers{Version: 1},
				Purevpn:      models.PurevpnServers{Version: 1},
				Surfshark:    models.SurfsharkServers{Version: 1},
				Torguard:     models.TorguardServers{Version: 1},
				VPNUnlimited: models.VPNUnlimitedServers{Version: 1},
				Vyprvpn:      models.VyprvpnServers{Version: 1},
				Wevpn:        models.WevpnServers{Version: 1},
				Windscribe:   models.WindscribeServers{Version: 1},
			},
			logged: []string{
				"Cyberghost servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Expressvpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Fastestvpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"HideMyAss servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Ipvanish servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Ivpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Mullvad servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Nordvpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Privado servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Pia servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Privatevpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Protonvpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Purevpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Surfshark servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Torguard servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"VPNUnlimited servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Vyprvpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Wevpn servers from file discarded because they have version 0 and hardcoded servers have version 1",
				"Windscribe servers from file discarded because they have version 0 and hardcoded servers have version 1",
			},
		},
		"same versions": {
			b: []byte(`{
				"cyberghost": {"version": 1, "timestamp": 1},
				"expressvpn": {"version": 1, "timestamp": 1},
				"fastestvpn": {"version": 1, "timestamp": 1},
				"hidemyass": {"version": 1, "timestamp": 1},
				"ipvanish": {"version": 1, "timestamp": 1},
				"ivpn": {"version": 1, "timestamp": 1},
				"mullvad": {"version": 1, "timestamp": 1},
				"nordvpn": {"version": 1, "timestamp": 1},
				"privado": {"version": 1, "timestamp": 1},
				"pia": {"version": 1, "timestamp": 1},
				"privatevpn": {"version": 1, "timestamp": 1},
				"protonvpn": {"version": 1, "timestamp": 1},
				"purevpn": {"version": 1, "timestamp": 1},
				"surfshark": {"version": 1, "timestamp": 1},
				"torguard": {"version": 1, "timestamp": 1},
				"vpnunlimited": {"version": 1, "timestamp": 1},
				"vyprvpn": {"version": 1, "timestamp": 1},
				"wevpn": {"version": 1, "timestamp": 1},
				"windscribe": {"version": 1, "timestamp": 1}
			}`),
			hardcoded: models.AllServers{
				Cyberghost:   models.CyberghostServers{Version: 1},
				Expressvpn:   models.ExpressvpnServers{Version: 1},
				Fastestvpn:   models.FastestvpnServers{Version: 1},
				HideMyAss:    models.HideMyAssServers{Version: 1},
				Ipvanish:     models.IpvanishServers{Version: 1},
				Ivpn:         models.IvpnServers{Version: 1},
				Mullvad:      models.MullvadServers{Version: 1},
				Nordvpn:      models.NordvpnServers{Version: 1},
				Privado:      models.PrivadoServers{Version: 1},
				Pia:          models.PiaServers{Version: 1},
				Privatevpn:   models.PrivatevpnServers{Version: 1},
				Protonvpn:    models.ProtonvpnServers{Version: 1},
				Purevpn:      models.PurevpnServers{Version: 1},
				Surfshark:    models.SurfsharkServers{Version: 1},
				Torguard:     models.TorguardServers{Version: 1},
				VPNUnlimited: models.VPNUnlimitedServers{Version: 1},
				Vyprvpn:      models.VyprvpnServers{Version: 1},
				Wevpn:        models.WevpnServers{Version: 1},
				Windscribe:   models.WindscribeServers{Version: 1},
			},
			persisted: models.AllServers{
				Cyberghost:   models.CyberghostServers{Version: 1, Timestamp: 1},
				Expressvpn:   models.ExpressvpnServers{Version: 1, Timestamp: 1},
				Fastestvpn:   models.FastestvpnServers{Version: 1, Timestamp: 1},
				HideMyAss:    models.HideMyAssServers{Version: 1, Timestamp: 1},
				Ipvanish:     models.IpvanishServers{Version: 1, Timestamp: 1},
				Ivpn:         models.IvpnServers{Version: 1, Timestamp: 1},
				Mullvad:      models.MullvadServers{Version: 1, Timestamp: 1},
				Nordvpn:      models.NordvpnServers{Version: 1, Timestamp: 1},
				Privado:      models.PrivadoServers{Version: 1, Timestamp: 1},
				Pia:          models.PiaServers{Version: 1, Timestamp: 1},
				Privatevpn:   models.PrivatevpnServers{Version: 1, Timestamp: 1},
				Protonvpn:    models.ProtonvpnServers{Version: 1, Timestamp: 1},
				Purevpn:      models.PurevpnServers{Version: 1, Timestamp: 1},
				Surfshark:    models.SurfsharkServers{Version: 1, Timestamp: 1},
				Torguard:     models.TorguardServers{Version: 1, Timestamp: 1},
				VPNUnlimited: models.VPNUnlimitedServers{Version: 1, Timestamp: 1},
				Vyprvpn:      models.VyprvpnServers{Version: 1, Timestamp: 1},
				Wevpn:        models.WevpnServers{Version: 1, Timestamp: 1},
				Windscribe:   models.WindscribeServers{Version: 1, Timestamp: 1},
			},
		},
	}

	for name, testCase := range testCases {
		testCase := testCase
		t.Run(name, func(t *testing.T) {
			t.Parallel()
			ctrl := gomock.NewController(t)

			logger := NewMockInfoErrorer(ctrl)
			for _, logged := range testCase.logged {
				logger.EXPECT().Info(logged)
			}

			s := &Storage{
				logger: logger,
			}

			servers, err := s.extractServersFromBytes(testCase.b, testCase.hardcoded)

			if testCase.err != nil {
				require.Error(t, err)
				assert.Equal(t, testCase.err.Error(), err.Error())
			} else {
				assert.NoError(t, err)
			}

			assert.Equal(t, testCase.persisted, servers)
		})
	}
}