feat(provider): Add Eureka Provider

provider/eureka_test.go

@@ -0,0 +1,170 @@
+package provider
+
+import (
+	"github.com/ArthurHlt/go-eureka-client/eureka"
+	"testing"
+)
+
+func TestEurekaGetPort(t *testing.T) {
+	cases := []struct {
+		expectedPort string
+		instanceInfo eureka.InstanceInfo
+	}{
+		{
+			expectedPort: "80",
+			instanceInfo: eureka.InstanceInfo{
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: false,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+			},
+		},
+		{
+			expectedPort: "443",
+			instanceInfo: eureka.InstanceInfo{
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: true,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: false,
+				},
+			},
+		},
+	}
+
+	eurekaProvider := &Eureka{}
+	for _, c := range cases {
+		port := eurekaProvider.getPort(c.instanceInfo)
+		if port != c.expectedPort {
+			t.Fatalf("Should have been %s, got %s", c.expectedPort, port)
+		}
+	}
+}
+
+func TestEurekaGetProtocol(t *testing.T) {
+	cases := []struct {
+		expectedProtocol string
+		instanceInfo     eureka.InstanceInfo
+	}{
+		{
+			expectedProtocol: "http",
+			instanceInfo: eureka.InstanceInfo{
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: false,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+			},
+		},
+		{
+			expectedProtocol: "https",
+			instanceInfo: eureka.InstanceInfo{
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: true,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: false,
+				},
+			},
+		},
+	}
+
+	eurekaProvider := &Eureka{}
+	for _, c := range cases {
+		protocol := eurekaProvider.getProtocol(c.instanceInfo)
+		if protocol != c.expectedProtocol {
+			t.Fatalf("Should have been %s, got %s", c.expectedProtocol, protocol)
+		}
+	}
+}
+
+func TestEurekaGetWeight(t *testing.T) {
+	cases := []struct {
+		expectedWeight string
+		instanceInfo   eureka.InstanceInfo
+	}{
+		{
+			expectedWeight: "0",
+			instanceInfo: eureka.InstanceInfo{
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+				Metadata: &eureka.MetaData{
+					Map: map[string]string{},
+				},
+			},
+		},
+		{
+			expectedWeight: "10",
+			instanceInfo: eureka.InstanceInfo{
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+				Metadata: &eureka.MetaData{
+					Map: map[string]string{
+						"traefik.weight": "10",
+					},
+				},
+			},
+		},
+	}
+
+	eurekaProvider := &Eureka{}
+	for _, c := range cases {
+		weight := eurekaProvider.getWeight(c.instanceInfo)
+		if weight != c.expectedWeight {
+			t.Fatalf("Should have been %s, got %s", c.expectedWeight, weight)
+		}
+	}
+}
+
+func TestEurekaGetInstanceId(t *testing.T) {
+	cases := []struct {
+		expectedID   string
+		instanceInfo eureka.InstanceInfo
+	}{
+		{
+			expectedID: "MyInstanceId",
+			instanceInfo: eureka.InstanceInfo{
+				IpAddr: "10.11.12.13",
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: false,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+				Metadata: &eureka.MetaData{
+					Map: map[string]string{
+						"traefik.backend.id": "MyInstanceId",
+					},
+				},
+			},
+		},
+		{
+			expectedID: "10-11-12-13-80",
+			instanceInfo: eureka.InstanceInfo{
+				IpAddr: "10.11.12.13",
+				SecurePort: &eureka.Port{
+					Port: 443, Enabled: false,
+				},
+				Port: &eureka.Port{
+					Port: 80, Enabled: true,
+				},
+				Metadata: &eureka.MetaData{
+					Map: map[string]string{},
+				},
+			},
+		},
+	}
+
+	eurekaProvider := &Eureka{}
+	for _, c := range cases {
+		id := eurekaProvider.getInstanceID(c.instanceInfo)
+		if id != c.expectedID {
+			t.Fatalf("Should have been %s, got %s", c.expectedID, id)
+		}
+	}
+}