получение списка gif

main
parent 39320d775f
commit a630d8af74

@ -4,7 +4,7 @@ import (
"github.com/jmoiron/sqlx" "github.com/jmoiron/sqlx"
) )
//SeaDB - картинки в БД // SeaDB - картинки в БД
type SeaDB struct { type SeaDB struct {
db *sqlx.DB db *sqlx.DB
} }
@ -14,8 +14,8 @@ func NewSeaDB(db *sqlx.DB) (*SeaDB, error) {
return &seadb, nil return &seadb, nil
} }
//Возвращает список url-ов маленьких gif-ов их БД // Возвращает список url-ов маленьких gif-ов из БД
func (sea *SeaDB)SmallGifs() ([]string, error) { func (sea *SeaDB) SmallGifs() ([]string, error) {
cmd := ` cmd := `
SELECT SELECT
newurl newurl
@ -39,4 +39,3 @@ func (sea *SeaDB)SmallGifs() ([]string, error) {
return urls, nil return urls, nil
} }

@ -1,10 +1,12 @@
package pic package pic
import ( import (
//"log" "strings"
"testing" "testing"
"github.com/stretchr/testify/assert"
"github.com/jmoiron/sqlx" "github.com/jmoiron/sqlx"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
) )
var db = openDB() var db = openDB()
@ -19,9 +21,17 @@ func openDB() *sqlx.DB {
func TestGetSmallGifs(t *testing.T) { func TestGetSmallGifs(t *testing.T) {
seadb, err := NewSeaDB(db) seadb, err := NewSeaDB(db)
assert.Nil(t, err) // если ошибка - дальнейшая проверка бессмыслена
require.NoError(t, err)
urls, err := seadb.SmallGifs() urls, err := seadb.SmallGifs()
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, len(urls), 10, "должны получить 10 записей") assert.Equal(t, len(urls), 10, "должны получить ровно 10 записей")
// теперь проверяем правильность формата url
// должны быть в форме: http://192.168.0.105:9091/4,530f8ea86e3e.gif
for _, u := range urls {
assert.Equal(t, strings.HasPrefix(u, "http://192.168.0.105:9091/"), true, "должен начинаться с http://192.168.0.105:9091/")
assert.Equal(t, strings.HasSuffix(u, ".gif"), true, "должен заканчиваться на .gif")
}
} }

Loading…
Cancel
Save