s/vgo/dep/

build/manifest/vendor/github.com/mattermost/mattermost-server/store/sqlstore/license_store.go

@@ -0,0 +1,60 @@
+// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package sqlstore
+
+import (
+	"net/http"
+
+	"github.com/mattermost/mattermost-server/model"
+	"github.com/mattermost/mattermost-server/store"
+)
+
+type SqlLicenseStore struct {
+	SqlStore
+}
+
+func NewSqlLicenseStore(sqlStore SqlStore) store.LicenseStore {
+	ls := &SqlLicenseStore{sqlStore}
+
+	for _, db := range sqlStore.GetAllConns() {
+		table := db.AddTableWithName(model.LicenseRecord{}, "Licenses").SetKeys(false, "Id")
+		table.ColMap("Id").SetMaxSize(26)
+		table.ColMap("Bytes").SetMaxSize(10000)
+	}
+
+	return ls
+}
+
+func (ls SqlLicenseStore) CreateIndexesIfNotExists() {
+}
+
+func (ls SqlLicenseStore) Save(license *model.LicenseRecord) store.StoreChannel {
+	return store.Do(func(result *store.StoreResult) {
+		license.PreSave()
+		if result.Err = license.IsValid(); result.Err != nil {
+			return
+		}
+
+		// Only insert if not exists
+		if err := ls.GetReplica().SelectOne(&model.LicenseRecord{}, "SELECT * FROM Licenses WHERE Id = :Id", map[string]interface{}{"Id": license.Id}); err != nil {
+			if err := ls.GetMaster().Insert(license); err != nil {
+				result.Err = model.NewAppError("SqlLicenseStore.Save", "store.sql_license.save.app_error", nil, "license_id="+license.Id+", "+err.Error(), http.StatusInternalServerError)
+			} else {
+				result.Data = license
+			}
+		}
+	})
+}
+
+func (ls SqlLicenseStore) Get(id string) store.StoreChannel {
+	return store.Do(func(result *store.StoreResult) {
+		if obj, err := ls.GetReplica().Get(model.LicenseRecord{}, id); err != nil {
+			result.Err = model.NewAppError("SqlLicenseStore.Get", "store.sql_license.get.app_error", nil, "license_id="+id+", "+err.Error(), http.StatusInternalServerError)
+		} else if obj == nil {
+			result.Err = model.NewAppError("SqlLicenseStore.Get", "store.sql_license.get.missing.app_error", nil, "license_id="+id, http.StatusNotFound)
+		} else {
+			result.Data = obj.(*model.LicenseRecord)
+		}
+	})
+}

build/manifest/vendor/github.com/mattermost/mattermost-server/store/storetest/license_store.go

@@ -0,0 +1,56 @@
+// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package storetest
+
+import (
+	"testing"
+
+	"github.com/mattermost/mattermost-server/model"
+	"github.com/mattermost/mattermost-server/store"
+)
+
+func TestLicenseStore(t *testing.T, ss store.Store) {
+	t.Run("Save", func(t *testing.T) { testLicenseStoreSave(t, ss) })
+	t.Run("Get", func(t *testing.T) { testLicenseStoreGet(t, ss) })
+}
+
+func testLicenseStoreSave(t *testing.T, ss store.Store) {
+	l1 := model.LicenseRecord{}
+	l1.Id = model.NewId()
+	l1.Bytes = "junk"
+
+	if err := (<-ss.License().Save(&l1)).Err; err != nil {
+		t.Fatal("couldn't save license record", err)
+	}
+
+	if err := (<-ss.License().Save(&l1)).Err; err != nil {
+		t.Fatal("shouldn't fail on trying to save existing license record", err)
+	}
+
+	l1.Id = ""
+
+	if err := (<-ss.License().Save(&l1)).Err; err == nil {
+		t.Fatal("should fail on invalid license", err)
+	}
+}
+
+func testLicenseStoreGet(t *testing.T, ss store.Store) {
+	l1 := model.LicenseRecord{}
+	l1.Id = model.NewId()
+	l1.Bytes = "junk"
+
+	store.Must(ss.License().Save(&l1))
+
+	if r := <-ss.License().Get(l1.Id); r.Err != nil {
+		t.Fatal("couldn't get license", r.Err)
+	} else {
+		if r.Data.(*model.LicenseRecord).Bytes != l1.Bytes {
+			t.Fatal("license bytes didn't match")
+		}
+	}
+
+	if err := (<-ss.License().Get("missing")).Err; err == nil {
+		t.Fatal("should fail on get license", err)
+	}
+}

build/manifest/vendor/github.com/mattermost/mattermost-server/store/storetest/mocks/LicenseStore.go

@@ -0,0 +1,46 @@
+// Code generated by mockery v1.0.0
+
+// Regenerate this file using `make store-mocks`.
+
+package mocks
+
+import mock "github.com/stretchr/testify/mock"
+import model "github.com/mattermost/mattermost-server/model"
+import store "github.com/mattermost/mattermost-server/store"
+
+// LicenseStore is an autogenerated mock type for the LicenseStore type
+type LicenseStore struct {
+	mock.Mock
+}
+
+// Get provides a mock function with given fields: id
+func (_m *LicenseStore) Get(id string) store.StoreChannel {
+	ret := _m.Called(id)
+
+	var r0 store.StoreChannel
+	if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok {
+		r0 = rf(id)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(store.StoreChannel)
+		}
+	}
+
+	return r0
+}
+
+// Save provides a mock function with given fields: license
+func (_m *LicenseStore) Save(license *model.LicenseRecord) store.StoreChannel {
+	ret := _m.Called(license)
+
+	var r0 store.StoreChannel
+	if rf, ok := ret.Get(0).(func(*model.LicenseRecord) store.StoreChannel); ok {
+		r0 = rf(license)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(store.StoreChannel)
+		}
+	}
+
+	return r0
+}