Update client license etag to handle new features (#2716)

1 files changed, 28 insertions, 1 deletions

diff --git a/api/license_test.go b/api/license_test.go
index b34aeb7a6..0126d6e54 100644
--- a/api/license_test.go
+++ b/api/license_test.go
@@ -4,13 +4,14 @@
 package api
 
 import (
+	"github.com/mattermost/platform/utils"
 	"testing"
 )
 
 func TestGetLicenceConfig(t *testing.T) {
 	Setup()
 
-	if result, err := Client.GetClientLicenceConfig(); err != nil {
+	if result, err := Client.GetClientLicenceConfig(""); err != nil {
 		t.Fatal(err)
 	} else {
 		cfg := result.Data.(map[string]string)
@@ -18,5 +19,31 @@ func TestGetLicenceConfig(t *testing.T) {
 		if _, ok := cfg["IsLicensed"]; !ok {
 			t.Fatal(cfg)
 		}
+
+		// test etag caching
+		if cache_result, err := Client.GetClientLicenceConfig(result.Etag); err != nil {
+			t.Fatal(err)
+		} else if len(cache_result.Data.(map[string]string)) != 0 {
+			t.Log(cache_result.Data)
+			t.Fatal("cache should be empty")
+		}
+
+		utils.ClientLicense["IsLicensed"] = "true"
+
+		if cache_result, err := Client.GetClientLicenceConfig(result.Etag); err != nil {
+			t.Fatal(err)
+		} else if len(cache_result.Data.(map[string]string)) == 0 {
+			t.Fatal("result should not be empty")
+		}
+
+		utils.ClientLicense["SomeFeature"] = "true"
+
+		if cache_result, err := Client.GetClientLicenceConfig(result.Etag); err != nil {
+			t.Fatal(err)
+		} else if len(cache_result.Data.(map[string]string)) == 0 {
+			t.Fatal("result should not be empty")
+		}
+
+		utils.ClientLicense = map[string]string{"IsLicensed": "false"}
 	}
 }