diff --git a/seadb.go b/seadb.go index 305d96d..4456cd5 100644 --- a/seadb.go +++ b/seadb.go @@ -4,7 +4,7 @@ import ( "github.com/jmoiron/sqlx" ) -//SeaDB - картинки в БД +// SeaDB - картинки в БД type SeaDB struct { db *sqlx.DB } @@ -14,8 +14,8 @@ func NewSeaDB(db *sqlx.DB) (*SeaDB, error) { return &seadb, nil } -//Возвращает список url-ов маленьких gif-ов их БД -func (sea *SeaDB)SmallGifs() ([]string, error) { +// Возвращает список url-ов маленьких gif-ов из БД +func (sea *SeaDB) SmallGifs() ([]string, error) { cmd := ` SELECT newurl @@ -39,4 +39,3 @@ func (sea *SeaDB)SmallGifs() ([]string, error) { return urls, nil } - diff --git a/seadb_test.go b/seadb_test.go index aadf3e0..042dd15 100644 --- a/seadb_test.go +++ b/seadb_test.go @@ -1,10 +1,12 @@ package pic import ( - //"log" + "strings" "testing" - "github.com/stretchr/testify/assert" + "github.com/jmoiron/sqlx" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" ) var db = openDB() @@ -19,9 +21,17 @@ func openDB() *sqlx.DB { func TestGetSmallGifs(t *testing.T) { seadb, err := NewSeaDB(db) - assert.Nil(t, err) + // если ошибка - дальнейшая проверка бессмыслена + require.NoError(t, err) urls, err := seadb.SmallGifs() 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") + } }