simplehq / gorpish Goto Github PK

func NewDB() *gorpish.DB {
    sqlDb, _ := sql.Open("postgres", "")
    gorpMap := &gorp.DbMap{Db: sqlDb, Dialect: gorp.PostgresDialect{}}
    return &gorpish.DB{DbMap: gorpMap}
}

db := NewDB(gorp.PostgresDialect{})
// Use the db as a normal gorp.DbMap

import (
    "testing"
    "github.com/simplehq/gorpish/mocks"
)

func TestInsert(t *testing.T) {
    db := mocks.NewTestDB()

    db.On("Insert", 1).Return(nil)

    err := db.Insert(1)

    db.AssertExpectations(t)
}

import "github.com/simplehq/gorpish/mocks"

type MyTestDB struct {
    *mocks.TestDB
}

func (db *MyTestDB) Select(i interface{}, query string, args ...interface{}) ([]interface{}, error) {
    // Mock it out.
}