test: Enterprise License tests

dgraph/cmd/zero/enterpriseExperiedKey.key

@@ -0,0 +1,18 @@
+-----BEGIN PGP MESSAGE-----
+	owEBwgI9/ZANAwAKAXomeeH3SyppAax7YgxsaWNlbnNlLmpzb25etF5+ewogICJ1
+	c2VyIjogIkRncmFwaCBUZXN0IEtleSIsCiAgIm1heF9ub2RlcyI6IDE4NDQ2NzQ0
+	MDczNzA5NTUxNjE1LAogICJleHBpcnkiOiAiMTk3MC0wMS0wMVQwMDowMDowMFoi
+	Cn0KiQIzBAABCgAdFiEED3lYS97wtaMT1MW+eiZ54fdLKmkFAl60Xn4ACgkQeiZ5
+	4fdLKmlVYQ//afX0H7Seb0ukxCMAcM8uqlBEGCEFv3K34avk1g8XRa6y4q/Ys0uU
+	DSaaDWdQ8IS5Q9SNlZBbJuqO6Pf1R01dEPTYQizWkDjYIBsY9xJnMZKEaA+F3bkn
+	8TXqI588+AvbqxHosz8cvh/nG+Ajk451rI9c2bqKB/FvH/zI6XyfUjqN+PvrqH0E
+	POA7nqSrWDemW4cMgNR4PhXehB/n2i3G6cPpwgCUd+N00N1f1mir/LmL6G5T4PrG
+	BmVz9fOdEr+U85PbMF9vOke9LkLQYdnF1hEV+7++t2/uoaLDYbxYhUnXpJZxwCBX
+	DQTievpyQF47HzuifvqUyxDSEsYiSGhhap1e/tvf1VaZoFUuTYQQpiV7+9K3UrL0
+	SnJ5TRWS7cEKBLyZszrakGpqVakvEUlWO4wG0Fp4kUa4YXw8t58oqHRn9aAtoqJd
+	UOLnq2semUttaySR4DHhjneO3/RoVm79/aaqMi/QNJzc9Tt9nY0AgcYlA3bVXmAZ
+	nM9Rgi6SaO5DxnRdhFzZyYQMb4onFDI0eYMOhPm+NmKWplkFXB+mKPKj5o/pcEb4
+	SWHt8fUAWDLsmcooIixDmSay14aBmF08hQ1vtJkY7/jo3hlK36GrLnNdN4IODqk/
+	I8mUd/jcj3NZtGWFoxKq4laK/ruoeoHnWMznJyMm75nzcU5QZU9yEEI=
+	=2lFm
+	-----END PGP MESSAGE-----

dgraph/cmd/zero/enterpriseInvalidKey.key

@@ -0,0 +1,18 @@
+-----BEGIN PGP MESSAGE-----
+	owEBwgI9/ZANAwAKAXomeeH3SyppAax7YgxsaWNlbnNlLmpzb25etF5+ewogICJ1
+	c2VyIjogIkRncmFwaCBUZXN0IEtleSIsCiAgIm1heF9ub2RlcyI6IDE4NDQ2NzQ0
+	MDczNzA5NTUxNjE1LAogICJleHBpcnkiOiAiMTk3MC0wMS0wMVQwMDowaaawMFoi
+	Cn0KiQIzBAABCgAdFiEED3lYS97wtaMT1MW+eiZ54fdLKmkFAl60Xn4ACgkQeiZ5
+	4fdLKmlVYQ//afX0H7Seb0ukxCMAcM8uqlBEGCEFv3K34avk1g8XRaaa4q/Ys0uU
+	DSaaDWdQ8IS5Q9SNlZBbJuqO6Pf1R01dEPTYQizWkDjYIBsY9xJnMZKEaA+F3bkn
+	8TXqI588+AvbqxHosz8cvh/nG+Ajk451rI9c2bqKB/FvH/zI6XyfUjqN+PvrqH0E
+	POA7nqSrWDemW4cMgNR4PhXehB/n2i3G6cPpwgCUd+N00N1f1mir/LmL6G5T4PrG
+	BmVz9fOdEr+U85aaaF9vOke9LkLQYdnF1hEV+7++t2/uoaLDYbxYaaaXpJZxwCBX
+	DQTievpyQF47HzuifvqUyxDSEsYiSGhhap1e/tvf1VaZoFUuTYQQaaa7+9K3UrL0
+	SnJ5TRWS7cEKBLyZszrakGpqVakvEUlWO4wG0Fp4kUa4YXw8t58oqHRn9aAtoqJd
+	UOLnq2semUttayaaaDHhjneO3/RoVm79/aaqMi/QNJzc9Ttaaa0AgcYlA3bVXmAZ
+	nM9Rgi6SaO5DxnRdhFzZyYQMb4onFDI0eYMOhPm+NmKWplkFXB+mKPKj5o/pcEb4
+	SWHt8fUAWDLsmaaaIixDmSay14aBmF08hQ1vtaaa7/jo3hlK36GrLnNdN4IODqk/
+	I8mUd/jcj3NZtGWFoxKq4laK/ruoeoHnWMznJyMm75nzcU5QZU9yEEI=
+	=2lFm
+	-----END PGP MESSAGE-----

dgraph/cmd/zero/enterpriseUnit_test.go

@@ -0,0 +1,50 @@
+package zero
+
+import (
+	"testing"
+
+	"github.com/dgraph-io/dgraph/protos/pb"
+	"github.com/stretchr/testify/require"
+)
+
+func TestEnterpriseLicenseUnit(t *testing.T) {
+
+	expiredKey := "enterpriseExperiedKey.Key"
+	invalidKey := "enterpriseInvalidKey.Key"
+
+	var tests = []struct {
+		name           string
+		licenseKeyPath string
+		expectError    bool
+		expectedOutput string
+	}{
+		{
+			"Using expired entrerprised license key should return an error",
+			expiredKey,
+			true,
+			`while extracting enterprise details from the license: while decoding license file: EOF`,
+		},
+		{
+			"Using invalid entrerprised license key should return an error",
+			invalidKey,
+			true,
+			`while extracting enterprise details from the license: while decoding license file: EOF`,
+		},
+	}
+	server := &Server{
+		state: &pb.MembershipState{
+			Groups: map[uint32]*pb.Group{1: {Members: map[uint64]*pb.Member{}}},
+		},
+	}
+	for _, tt := range tests {
+		t.Logf("Running: %s\n", tt.name)
+		err := server.applyLicenseFileReturnError(expiredKey)
+		if tt.expectError {
+			require.Error(t, err)
+			continue
+		}
+
+		require.NoError(t, err)
+	}
+
+}

dgraph/cmd/zero/license_ee.go

@@ -147,3 +147,17 @@ func (s *Server) applyLicenseFile(path string) {
 		glog.Infof("Unable to apply license at %v due to error %v", path, err)
 	}
 }
+
+func (s *Server) applyLicenseFileReturnError(path string) error {
+	content, err := ioutil.ReadFile(path)
+	if err != nil {
+		glog.Infof("Unable to apply license at %v due to error %v", path, err)
+		return err
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+	defer cancel()
+	if err = s.applyLicense(ctx, bytes.NewReader(content)); err != nil {
+		glog.Infof("Unable to apply license at %v due to error %v", path, err)
+	}
+	return err
+}