github.com/haalcala/mattermost-server-change-repo@v0.0.0-20210713015153-16753fbeee5f/store/sqlstore/upgrade_test.go (about)

     1  // Copyright (c) 2015-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/stretchr/testify/require"
    10  
    11  	"github.com/mattermost/mattermost-server/v5/store"
    12  )
    13  
    14  func TestStoreUpgradeDotRelease(t *testing.T) {
    15  	StoreTest(t, func(t *testing.T, ss store.Store) {
    16  		sqlStore := ss.(*SqlStore)
    17  		saveSchemaVersion(sqlStore, "5.32.1")
    18  		err := upgradeDatabase(sqlStore, CurrentSchemaVersion)
    19  		require.NoError(t, err)
    20  		require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
    21  	})
    22  }
    23  
    24  func TestStoreUpgrade(t *testing.T) {
    25  	StoreTest(t, func(t *testing.T, ss store.Store) {
    26  		sqlStore := ss.(*SqlStore)
    27  
    28  		t.Run("invalid currentModelVersion", func(t *testing.T) {
    29  			err := upgradeDatabase(sqlStore, "notaversion")
    30  			require.EqualError(t, err, "failed to parse current model version notaversion: No Major.Minor.Patch elements found")
    31  		})
    32  
    33  		t.Run("upgrade from invalid version", func(t *testing.T) {
    34  			saveSchemaVersion(sqlStore, "invalid")
    35  			err := upgradeDatabase(sqlStore, "5.8.0")
    36  			require.EqualError(t, err, "failed to parse database schema version invalid: No Major.Minor.Patch elements found")
    37  			require.Equal(t, "invalid", sqlStore.GetCurrentSchemaVersion())
    38  		})
    39  
    40  		t.Run("upgrade from unsupported version", func(t *testing.T) {
    41  			saveSchemaVersion(sqlStore, "2.0.0")
    42  			err := upgradeDatabase(sqlStore, "5.8.0")
    43  			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.")
    44  			require.Equal(t, "2.0.0", sqlStore.GetCurrentSchemaVersion())
    45  		})
    46  
    47  		t.Run("upgrade from earliest supported version", func(t *testing.T) {
    48  			saveSchemaVersion(sqlStore, Version300)
    49  			err := upgradeDatabase(sqlStore, CurrentSchemaVersion)
    50  			require.NoError(t, err)
    51  			require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
    52  		})
    53  
    54  		t.Run("upgrade from no existing version", func(t *testing.T) {
    55  			saveSchemaVersion(sqlStore, "")
    56  			err := upgradeDatabase(sqlStore, CurrentSchemaVersion)
    57  			require.NoError(t, err)
    58  			require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
    59  		})
    60  
    61  		t.Run("upgrade schema running earlier minor version", func(t *testing.T) {
    62  			saveSchemaVersion(sqlStore, "5.1.0")
    63  			err := upgradeDatabase(sqlStore, "5.8.0")
    64  			require.NoError(t, err)
    65  			// Assert CurrentSchemaVersion, not 5.8.0, since the migrations will move
    66  			// past 5.8.0 regardless of the input parameter.
    67  			require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
    68  		})
    69  
    70  		t.Run("upgrade schema running later minor version", func(t *testing.T) {
    71  			saveSchemaVersion(sqlStore, "5.99.0")
    72  			err := upgradeDatabase(sqlStore, "5.8.0")
    73  			require.NoError(t, err)
    74  			require.Equal(t, "5.99.0", sqlStore.GetCurrentSchemaVersion())
    75  		})
    76  
    77  		t.Run("upgrade schema running earlier major version", func(t *testing.T) {
    78  			saveSchemaVersion(sqlStore, "4.1.0")
    79  			err := upgradeDatabase(sqlStore, CurrentSchemaVersion)
    80  			require.NoError(t, err)
    81  			require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
    82  		})
    83  
    84  		t.Run("upgrade schema running later major version", func(t *testing.T) {
    85  			saveSchemaVersion(sqlStore, "6.0.0")
    86  			err := upgradeDatabase(sqlStore, "5.8.0")
    87  			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.")
    88  			require.Equal(t, "6.0.0", sqlStore.GetCurrentSchemaVersion())
    89  		})
    90  	})
    91  }
    92  
    93  func TestSaveSchemaVersion(t *testing.T) {
    94  	StoreTest(t, func(t *testing.T, ss store.Store) {
    95  		sqlStore := ss.(*SqlStore)
    96  
    97  		t.Run("set earliest version", func(t *testing.T) {
    98  			saveSchemaVersion(sqlStore, Version300)
    99  			props, err := ss.System().Get()
   100  			require.NoError(t, err)
   101  
   102  			require.Equal(t, Version300, props["Version"])
   103  			require.Equal(t, Version300, sqlStore.GetCurrentSchemaVersion())
   104  		})
   105  
   106  		t.Run("set current version", func(t *testing.T) {
   107  			saveSchemaVersion(sqlStore, CurrentSchemaVersion)
   108  			props, err := ss.System().Get()
   109  			require.NoError(t, err)
   110  
   111  			require.Equal(t, CurrentSchemaVersion, props["Version"])
   112  			require.Equal(t, CurrentSchemaVersion, sqlStore.GetCurrentSchemaVersion())
   113  		})
   114  	})
   115  }