status-im/status-go

package localnotifications

import (
    "database/sql"
    "testing"

    "github.com/stretchr/testify/require"

    "github.com/status-im/status-go/appdatabase"
    "github.com/status-im/status-go/t/helpers"
)

func setupAppTestDb(t *testing.T) (*sql.DB, func()) {
    db, cleanup, err := helpers.SetupTestSQLDB(appdatabase.DbInitializer{}, "local-notifications-tests-")
    require.NoError(t, err)
    return db, func() { require.NoError(t, cleanup()) }
}

func setupTestDB(t *testing.T, db *sql.DB) (*Database, func()) {
    return NewDB(db, 1777), func() {
        require.NoError(t, db.Close())
    }
}

func TestWalletPreferences(t *testing.T) {
    appDB, appStop := setupAppTestDb(t)
    defer appStop()

    db, stop := setupTestDB(t, appDB)
    defer stop()

    enabled := true
    require.NoError(t, db.ChangeWalletPreference(enabled))
    rst, err := db.GetWalletPreference()
    require.NoError(t, err)
    require.Equal(t, enabled, rst.Enabled)

    enabled = false
    require.NoError(t, db.ChangeWalletPreference(enabled))
    rst, err = db.GetWalletPreference()
    require.Equal(t, enabled, rst.Enabled)

    require.NoError(t, err)
}

func TestPreferences(t *testing.T) {
    appDB, appStop := setupAppTestDb(t)
    defer appStop()

    db, stop := setupTestDB(t, appDB)
    defer stop()

    enabled := true

    require.NoError(t, db.ChangeWalletPreference(enabled))
    rst, err := db.GetPreferences()

    require.Equal(t, 1, len(rst))
    require.Equal(t, enabled, rst[0].Enabled)

    require.NoError(t, err)
}