PLT-6115 adding unit tests for licensing (#5916)

* Adding test to license sql store

* PLT-6115 adding unit tests for licensing

* Removing unused test

app/license_test.go

@@ -0,0 +1,37 @@
+// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package app
+
+import (
+	//"github.com/mattermost/platform/model"
+	"github.com/mattermost/platform/utils"
+	"testing"
+)
+
+func TestLoadLicense(t *testing.T) {
+	Setup()
+
+	LoadLicense()
+	if utils.IsLicensed {
+		t.Fatal("shouldn't have a valid license")
+	}
+}
+
+func TestSaveLicense(t *testing.T) {
+	Setup()
+
+	b1 := []byte("junk")
+
+	if _, err := SaveLicense(b1); err == nil {
+		t.Fatal("shouldn't have saved license")
+	}
+}
+
+func TestRemoveLicense(t *testing.T) {
+	Setup()
+
+	if err := RemoveLicense(); err != nil {
+		t.Fatal("should have removed license")
+	}
+}

store/sql_license_store_test.go

@@ -22,6 +22,12 @@ func TestLicenseStoreSave(t *testing.T) {
 	if err := (<-store.License().Save(&l1)).Err; err != nil {
 		t.Fatal("shouldn't fail on trying to save existing license record", err)
 	}
+
+	l1.Id = ""
+
+	if err := (<-store.License().Save(&l1)).Err; err == nil {
+		t.Fatal("should fail on invalid license", err)
+	}
 }
 
 func TestLicenseStoreGet(t *testing.T) {
@@ -40,4 +46,8 @@ func TestLicenseStoreGet(t *testing.T) {
 			t.Fatal("license bytes didn't match")
 		}
 	}
+
+	if err := (<-store.License().Get("missing")).Err; err == nil {
+		t.Fatal("should fail on get license", err)
+	}
 }

utils/license_test.go

@@ -43,6 +43,8 @@ func TestValidateLicense(t *testing.T) {
 		t.Fatal("should have failed - bad license")
 	}
 
+	LoadLicense(b1)
+
 	b2 := []byte("junkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunkjunk")
 	if ok, _ := ValidateLicense(b2); ok {
 		t.Fatal("should have failed - bad license")
@@ -67,6 +69,22 @@ func TestClientLicenseEtag(t *testing.T) {
 	}
 }
 
+func TestGetSanitizedClientLicense(t *testing.T) {
+	l1 := &model.License{}
+	l1.Features = &model.Features{}
+	l1.Customer = &model.Customer{}
+	l1.Customer.Name = "TestName"
+	l1.StartsAt = model.GetMillis() - 1000
+	l1.ExpiresAt = model.GetMillis() + 100000
+	SetLicense(l1)
+
+	m := GetSanitizedClientLicense()
+
+	if _, ok := m["Name"]; ok {
+		t.Fatal("should have been sanatized")
+	}
+}
+
 func TestGetLicenseFileLocation(t *testing.T) {
 	fileName := GetLicenseFileLocation("")
 	if len(fileName) == 0 {