status-go/services/gif/gif_test.go
Ivan Belyakov 2df9df10ab fix(tests): moved test db setup to a common place 't/helpers', created
interface for initializing db, which is implemented for appdatabase and
walletdatabase. TBD for multiaccounts DB.
Unified DB initializion for all tests using helpers and new interface.
Reduced sqlcipher kdf iterations for all tests to 1.
2023-08-18 09:00:56 +02:00

137 lines
3.3 KiB
Go

package gif
import (
"database/sql"
"encoding/json"
"testing"
"github.com/stretchr/testify/require"
"github.com/status-im/status-go/appdatabase"
"github.com/status-im/status-go/multiaccounts/accounts"
"github.com/status-im/status-go/t/helpers"
)
func setupSQLTestDb(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) (*accounts.Database, func()) {
acc, err := accounts.NewDB(db)
require.NoError(t, err)
return acc, func() {
require.NoError(t, db.Close())
}
}
func TestSetTenorAPIKey(t *testing.T) {
appDB, appStop := setupSQLTestDb(t)
defer appStop()
db, stop := setupTestDB(t, appDB)
defer stop()
gifAPI := NewGifAPI(db)
require.NoError(t, gifAPI.SetTenorAPIKey("DU7DWZ27STB2"))
require.Equal(t, "DU7DWZ27STB2", tenorAPIKey)
}
func TestGetContentWithRetry(t *testing.T) {
appDB, appStop := setupSQLTestDb(t)
defer appStop()
db, stop := setupTestDB(t, appDB)
defer stop()
gifAPI := NewGifAPI(db)
require.NoError(t, gifAPI.SetTenorAPIKey(""))
require.Equal(t, "", tenorAPIKey)
gifs, err := gifAPI.GetContentWithRetry("trending?")
require.Error(t, err)
require.Equal(t, "", gifs)
require.NoError(t, gifAPI.SetTenorAPIKey("DU7DWZ27STB2"))
require.Equal(t, "DU7DWZ27STB2", tenorAPIKey)
gifs, err = gifAPI.GetContentWithRetry("trending?")
require.NoError(t, err)
require.NotEqual(t, "", gifs)
}
func TestFavoriteGifs(t *testing.T) {
appDB, appStop := setupSQLTestDb(t)
defer appStop()
db, stop := setupTestDB(t, appDB)
defer stop()
gifAPI := NewGifAPI(db)
require.NoError(t, gifAPI.SetTenorAPIKey("DU7DWZ27STB2"))
require.Equal(t, "DU7DWZ27STB2", tenorAPIKey)
recent := map[string]interface{}{
"id": "23833142",
"title": "",
"url": "https://media.tenor.com/images/b845ae14f43883e5cd6283e705f09efb/tenor.gif",
"tinyUrl": "https://media.tenor.com/images/2067bdc0375f9606dfb9fb4d2bfaafde/tenor.gif",
"height": 498,
"isFavorite": true,
}
newRecents := map[string]interface{}{
"items": recent,
}
inputJSON := map[string]interface{}{
"jsonrpc": "2.0",
"method": "gif_setTenorAPIKey",
"params": newRecents,
}
like, _ := json.Marshal(inputJSON)
source := (json.RawMessage)(like)
require.NoError(t, gifAPI.UpdateFavoriteGifs(source))
}
func TestRecentGifs(t *testing.T) {
appDB, appStop := setupSQLTestDb(t)
defer appStop()
db, stop := setupTestDB(t, appDB)
defer stop()
gifAPI := NewGifAPI(db)
require.NoError(t, gifAPI.SetTenorAPIKey("DU7DWZ27STB2"))
require.Equal(t, "DU7DWZ27STB2", tenorAPIKey)
recent := map[string]interface{}{
"id": "23833142",
"title": "",
"url": "https://media.tenor.com/images/b845ae14f43883e5cd6283e705f09efb/tenor.gif",
"tinyUrl": "https://media.tenor.com/images/2067bdc0375f9606dfb9fb4d2bfaafde/tenor.gif",
"height": 498,
"isFavorite": true,
}
newRecents := map[string]interface{}{
"items": recent,
}
inputJSON := map[string]interface{}{
"jsonrpc": "2.0",
"method": "gif_setTenorAPIKey",
"params": newRecents,
}
like, _ := json.Marshal(inputJSON)
source := (json.RawMessage)(like)
require.NoError(t, gifAPI.UpdateRecentGifs(source))
}