mirror of https://github.com/portainer/portainer
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package datastore
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// isFileExist is helper function to check for file existence
|
|
func isFileExist(path string) bool {
|
|
matches, err := filepath.Glob(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return len(matches) > 0
|
|
}
|
|
|
|
func updateVersion(store *Store, v string) {
|
|
version, err := store.VersionService.Version()
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
|
|
version.SchemaVersion = v
|
|
|
|
err = store.VersionService.UpdateVersion(version)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
}
|
|
|
|
func updateEdition(store *Store, edition portainer.SoftwareEdition) {
|
|
version, err := store.VersionService.Version()
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
|
|
version.Edition = int(edition)
|
|
|
|
err = store.VersionService.UpdateVersion(version)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
}
|
|
|
|
// testVersion is a helper which tests current store version against wanted version
|
|
func testVersion(store *Store, versionWant string, t *testing.T) {
|
|
v, err := store.VersionService.Version()
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
if v.SchemaVersion != versionWant {
|
|
t.Errorf("Expect store version to be %s but was %s", versionWant, v.SchemaVersion)
|
|
}
|
|
}
|