diff options
| -rw-r--r-- | app/license_test.go | 37 | ||||
| -rw-r--r-- | store/sql_license_store_test.go | 10 | ||||
| -rw-r--r-- | utils/license_test.go | 18 |
3 files changed, 65 insertions, 0 deletions
diff --git a/app/license_test.go b/app/license_test.go new file mode 100644 index 000000000..d7d851589 --- /dev/null +++ b/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") + } +} diff --git a/store/sql_license_store_test.go b/store/sql_license_store_test.go index ad24a6af7..d91cff481 100644 --- a/store/sql_license_store_test.go +++ b/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) + } } diff --git a/utils/license_test.go b/utils/license_test.go index 5c0122bc0..7a30070b3 100644 --- a/utils/license_test.go +++ b/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 { |