github.com/teloshs/mattermost-server@v5.11.1+incompatible/store/sqlstore/upgrade_test.go (about)

     1  // Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
     2  // See License.txt for license information.
     3  
     4  package sqlstore
     5  
     6  import (
     7  	"testing"
     8  
     9  	"github.com/mattermost/mattermost-server/model"
    10  	"github.com/mattermost/mattermost-server/store"
    11  	"github.com/stretchr/testify/require"
    12  )
    13  
    14  func TestStoreUpgrade(t *testing.T) {
    15  	StoreTest(t, func(t *testing.T, ss store.Store) {
    16  		sqlStore := ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)
    17  
    18  		t.Run("invalid currentModelVersion", func(t *testing.T) {
    19  			err := UpgradeDatabase(sqlStore, "notaversion")
    20  			require.EqualError(t, err, "failed to parse current model version notaversion: No Major.Minor.Patch elements found")
    21  		})
    22  
    23  		t.Run("upgrade from invalid version", func(t *testing.T) {
    24  			saveSchemaVersion(sqlStore, "invalid")
    25  			err := UpgradeDatabase(sqlStore, "5.8.0")
    26  			require.EqualError(t, err, "failed to parse database schema version invalid: No Major.Minor.Patch elements found")
    27  			require.Equal(t, "invalid", sqlStore.GetCurrentSchemaVersion())
    28  		})
    29  
    30  		t.Run("upgrade from unsupported version", func(t *testing.T) {
    31  			saveSchemaVersion(sqlStore, "2.0.0")
    32  			err := UpgradeDatabase(sqlStore, "5.8.0")
    33  			require.EqualError(t, err, "Database schema version 2.0.0 is no longer supported. This Mattermost server supports automatic upgrades from schema version 3.0.0 through schema version 5.8.0. Please manually upgrade to at least version 3.0.0 before continuing.")
    34  			require.Equal(t, "2.0.0", sqlStore.GetCurrentSchemaVersion())
    35  		})
    36  
    37  		t.Run("upgrade from earliest supported version", func(t *testing.T) {
    38  			saveSchemaVersion(sqlStore, VERSION_3_0_0)
    39  			err := UpgradeDatabase(sqlStore, model.CurrentVersion)
    40  			require.NoError(t, err)
    41  			require.Equal(t, model.CurrentVersion, sqlStore.GetCurrentSchemaVersion())
    42  		})
    43  
    44  		t.Run("upgrade from no existing version", func(t *testing.T) {
    45  			saveSchemaVersion(sqlStore, "")
    46  			err := UpgradeDatabase(sqlStore, model.CurrentVersion)
    47  			require.NoError(t, err)
    48  			require.Equal(t, model.CurrentVersion, sqlStore.GetCurrentSchemaVersion())
    49  		})
    50  
    51  		t.Run("upgrade schema running earlier minor version", func(t *testing.T) {
    52  			saveSchemaVersion(sqlStore, "5.1.0")
    53  			err := UpgradeDatabase(sqlStore, "5.8.0")
    54  			require.NoError(t, err)
    55  			// Assert model.CurrentVersion, not 5.8.0, since the migrations will move
    56  			// past 5.8.0 regardless of the input parameter.
    57  			require.Equal(t, model.CurrentVersion, sqlStore.GetCurrentSchemaVersion())
    58  		})
    59  
    60  		t.Run("upgrade schema running later minor version", func(t *testing.T) {
    61  			saveSchemaVersion(sqlStore, "5.29.0")
    62  			err := UpgradeDatabase(sqlStore, "5.8.0")
    63  			require.NoError(t, err)
    64  			require.Equal(t, "5.29.0", sqlStore.GetCurrentSchemaVersion())
    65  		})
    66  
    67  		t.Run("upgrade schema running earlier major version", func(t *testing.T) {
    68  			saveSchemaVersion(sqlStore, "4.1.0")
    69  			err := UpgradeDatabase(sqlStore, model.CurrentVersion)
    70  			require.NoError(t, err)
    71  			require.Equal(t, model.CurrentVersion, sqlStore.GetCurrentSchemaVersion())
    72  		})
    73  
    74  		t.Run("upgrade schema running later major version", func(t *testing.T) {
    75  			saveSchemaVersion(sqlStore, "6.0.0")
    76  			err := UpgradeDatabase(sqlStore, "5.8.0")
    77  			require.EqualError(t, err, "Database schema version 6.0.0 is not supported. This Mattermost server supports only >=5.8.0, <6.0.0. Please upgrade to at least version 6.0.0 before continuing.")
    78  			require.Equal(t, "6.0.0", sqlStore.GetCurrentSchemaVersion())
    79  		})
    80  	})
    81  }
    82  
    83  func TestSaveSchemaVersion(t *testing.T) {
    84  	StoreTest(t, func(t *testing.T, ss store.Store) {
    85  		sqlStore := ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)
    86  
    87  		t.Run("set earliest version", func(t *testing.T) {
    88  			saveSchemaVersion(sqlStore, VERSION_3_0_0)
    89  			result := <-ss.System().Get()
    90  			require.Nil(t, result.Err)
    91  
    92  			props := result.Data.(model.StringMap)
    93  			require.Equal(t, VERSION_3_0_0, props["Version"])
    94  			require.Equal(t, VERSION_3_0_0, sqlStore.GetCurrentSchemaVersion())
    95  		})
    96  
    97  		t.Run("set current version", func(t *testing.T) {
    98  			saveSchemaVersion(sqlStore, model.CurrentVersion)
    99  			result := <-ss.System().Get()
   100  			require.Nil(t, result.Err)
   101  
   102  			props := result.Data.(model.StringMap)
   103  			require.Equal(t, model.CurrentVersion, props["Version"])
   104  			require.Equal(t, model.CurrentVersion, sqlStore.GetCurrentSchemaVersion())
   105  		})
   106  	})
   107  }