github/drone
5
0
Fork 0
mirror of https://github.com/harness/drone.git synced 2025-05-11 14:40:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move testing related code

Browse Source
This commit is contained in:
Nurahmadie 2014-03-27 13:57:53 +00:00
parent fe58917a7c
commit 202ea95368
1 changed files with 76 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
pkg/database/migrate/sqlite_test.go → pkg/database/migrate/testing/sqlite_test.go
View File

@ -3,12 +3,14 @@ package migrate_test
import ( import (
"database/sql" "database/sql"
"fmt" "fmt"
"log"
"os" "os"
"strings" "strings"
"testing" "testing"
. "github.com/drone/drone/pkg/database/migrate" . "github.com/drone/drone/pkg/database/migrate"
_ "github.com/go-sql-driver/mysql"
_ "github.com/mattn/go-sqlite3" _ "github.com/mattn/go-sqlite3"
"github.com/russross/meddler" "github.com/russross/meddler"
) )
@ -476,6 +478,10 @@ func TestIndexOperations(t *testing.T) {
} }
var esquel []*sqliteMaster var esquel []*sqliteMaster
switch {
case driver == "mysql":
default:
// Query sqlite_master, check if index is exists. // Query sqlite_master, check if index is exists.
query := `SELECT sql FROM sqlite_master WHERE type='index' and tbl_name='samples'` query := `SELECT sql FROM sqlite_master WHERE type='index' and tbl_name='samples'`
if err := meddler.QueryAll(db, &esquel, query); err != nil { if err := meddler.QueryAll(db, &esquel, query); err != nil {
@ -514,6 +520,7 @@ func TestIndexOperations(t *testing.T) {
if len(esquel2) != 1 { if len(esquel2) != 1 {
t.Errorf("Expect row length equal to %d, got %d", 1, len(esquel2)) t.Errorf("Expect row length equal to %d, got %d", 1, len(esquel2))
} }
}
} }
func TestColumnRedundancy(t *testing.T) { func TestColumnRedundancy(t *testing.T) {
@ -527,15 +534,25 @@ func TestColumnRedundancy(t *testing.T) {
t.Fatalf("Can not migrate: %q", err) t.Fatalf("Can not migrate: %q", err)
} }
var tableSql string var dummy, query, tableSql string
query := `SELECT sql FROM sqlite_master where type='table' and name='samples'` switch {
case driver == "mysql":
query = `SHOW CREATE TABLE samples`
if err := db.QueryRow(query).Scan(&dummy, &tableSql); err != nil {
t.Fatalf("Can not query table's definition: %q", err)
}
if !strings.Contains(tableSql, "`repository`") {
t.Errorf("Expect column with name repository")
}
default:
query = `SELECT sql FROM sqlite_master where type='table' and name='samples'`
if err := db.QueryRow(query).Scan(&tableSql); err != nil { if err := db.QueryRow(query).Scan(&tableSql); err != nil {
t.Fatalf("Can not query sqlite_master: %q", err) t.Fatalf("Can not query sqlite_master: %q", err)
} }
if !strings.Contains(tableSql, "repository ") { if !strings.Contains(tableSql, "repository ") {
t.Errorf("Expect column with name repository") t.Errorf("Expect column with name repository")
} }
}
} }
func TestChangeColumnType(t *testing.T) { func TestChangeColumnType(t *testing.T) {
@ -549,15 +566,25 @@ func TestChangeColumnType(t *testing.T) {
t.Fatalf("Can not migrate: %q", err) t.Fatalf("Can not migrate: %q", err)
} }
var tableSql string var dummy, tableSql, query string
query := `SELECT sql FROM sqlite_master where type='table' and name='samples'` switch {
case driver == "mysql":
query = `SHOW CREATE TABLE samples`
if err := db.QueryRow(query).Scan(&dummy, &tableSql); err != nil {
t.Fatalf("Can not query table's definition: %q", err)
}
if !strings.Contains(tableSql, "`email` varchar(512)") {
t.Errorf("Expect email type to changed: %q", tableSql)
}
default:
query = `SELECT sql FROM sqlite_master where type='table' and name='samples'`
if err := db.QueryRow(query).Scan(&tableSql); err != nil { if err := db.QueryRow(query).Scan(&tableSql); err != nil {
t.Fatalf("Can not query sqlite_master: %q", err) t.Fatalf("Can not query sqlite_master: %q", err)
} }
if !strings.Contains(tableSql, "email varchar(512) UNIQUE") { if !strings.Contains(tableSql, "email varchar(512) UNIQUE") {
t.Errorf("Expect email type to changed: %q", tableSql) t.Errorf("Expect email type to changed: %q", tableSql)
} }
}
} }
func init() { func init() {
@ -573,15 +600,15 @@ func setUp() error {
var err error var err error
Driver = SQLite Driver = SQLite
if db, err = sql.Open(driver, dsn); err != nil { if db, err = sql.Open(driver, dsn); err != nil {
panic("Can't connect to database: ") log.Fatalf("Can't connect to database: %q", err)
} }
if driver == "mysql" { if driver == "mysql" {
Driver = MySQL Driver = MySQL
if _, err := db.Exec(fmt.Sprintf("CREATE DATABASE %s", dbname)); err != nil { if _, err := db.Exec(fmt.Sprintf("CREATE DATABASE %s", dbname)); err != nil {
panic("Can't create database: ") panic("Can't create database")
} }
if _, err := db.Exec(fmt.Sprintf("USE %s", dbname)); err != nil { if _, err := db.Exec(fmt.Sprintf("USE %s", dbname)); err != nil {
panic("Can't use database: ") panic("Can't use database")
} }
} }
return err return err