34 lines
1.0 KiB
Go
34 lines
1.0 KiB
Go
package persistence_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"os"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/persistence"
|
|
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/scraper"
|
|
)
|
|
|
|
func TestVersionUpgrade(t *testing.T) {
|
|
require := require.New(t)
|
|
profile_path := "test_profiles/TestVersions"
|
|
if file_exists(profile_path) {
|
|
err := os.RemoveAll(profile_path)
|
|
require.NoError(err)
|
|
}
|
|
profile := create_or_load_profile(profile_path)
|
|
|
|
test_migration := "insert into tweets (id, user_id, text) values (21250554358298342, -1, 'awefjk')"
|
|
test_tweet_id := scraper.TweetID(21250554358298342)
|
|
|
|
require.False(profile.IsTweetInDatabase(test_tweet_id), "Test tweet shouldn't be in db yet")
|
|
|
|
persistence.MIGRATIONS = append(persistence.MIGRATIONS, test_migration)
|
|
err := profile.UpgradeFromXToY(persistence.ENGINE_DATABASE_VERSION, persistence.ENGINE_DATABASE_VERSION+1)
|
|
require.NoError(err)
|
|
|
|
require.True(profile.IsTweetInDatabase(test_tweet_id), "Migration should have created the tweet, but it didn't")
|
|
}
|