добавил тестирование

main
parent d9f93205d2
commit 8345ba6dd3

@ -1,18 +1,18 @@
package seaweed
/***
Пакет seaweed предназначен для облегчения работы с БД sea
и weed.
Текущая версия: v1.0.6
Текущая версия: v1.0.7
***/
package seaweed
import (
"os"
"fmt"
"errors"
"flag"
"fmt"
"os"
_ "github.com/go-sql-driver/mysql"
_ "github.com/go-sql-driver/mysql"
"github.com/jmoiron/sqlx"
"github.com/kataras/pio"
@ -24,22 +24,23 @@ var p = pio.NewTextPrinter("color", os.Stdout)
// ============= SEA ============
type Sea struct {
db *sqlx.DB
db *sqlx.DB
debug bool
}
// Connect to database and connect to weed
func Connect() (*Sea, error) {
debug := flag.Bool("debug", false, "enable debug mode")
fmt.Printf("debug: %v\n", *debug)
user := os.Getenv("SEA_USER")
if user == "" {
return nil, errors.New("Не установлена переменная SEA_USER")
return nil, errors.New("не установлена переменная SEA_USER")
}
password := os.Getenv("SEA_PASSWORD")
if password == "" {
return nil, errors.New("Не установлена переменная SEA_PASSWORD")
return nil, errors.New("не установлена переменная SEA_PASSWORD")
}
db, err := openDB(user, password)
@ -50,7 +51,7 @@ func Connect() (*Sea, error) {
if *debug {
p.Println(pio.Rich("БД sea успешно открыта", pio.Blue))
}
sea := Sea{db: db, debug: *debug}
return &sea, nil
}
@ -84,4 +85,3 @@ func (s *Sea) Albums() ([]Album, error) {
}
return albums, nil
}

@ -0,0 +1,36 @@
/***
Пакет seaweed предназначен для облегчения работы с БД sea
и weed.
Текущая версия: v1.0.7
***/
package seaweed
import (
"reflect"
"testing"
_ "github.com/go-sql-driver/mysql"
)
func TestConnect(t *testing.T) {
tests := []struct {
name string
want *Sea
wantErr bool
}{
// TODO: Add test cases.
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := Connect()
if (err != nil) != tt.wantErr {
t.Errorf("Connect() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("Connect() = %v, want %v", got, tt.want)
}
})
}
}
Loading…
Cancel
Save