go-sqlmock でテストをとりあえず書く

package user

import (
	"database/sql"
	"strconv"
	"time"

	"github.com/pkg/errors"

	"github.com/k1350/sololog/internal/domain/model/user"
	iu "github.com/k1350/sololog/internal/domain/repository/user"
	ae "github.com/k1350/sololog/internal/errors"
)

type UserRepository struct {
	*sql.DB
}

func NewUserRepository(db *sql.DB) iu.IUserRepository {
	return &UserRepository{db}
}

func (r *UserRepository) IsExist(uid string) (bool, error) {
	cSqlString := `
		SELECT 
		 count(1)
		FROM users 
		WHERE uid = ? `

	var i string
	if err := r.QueryRow(cSqlString, uid).Scan(&i); err != nil {
		err = errors.Wrap(ae.DbError, err.Error())
		return false, err
	}
	count, err := strconv.Atoi(i)
	if err != nil {
		err = errors.Wrap(ae.DbError, err.Error())
		return false, err
	}
	return count > 0, nil
}

func (r *UserRepository) Create(user *user.User) error {
	iSqlString := `
		INSERT INTO users
		(uid, last_logined, created_at, updated_at)
		VALUES
		(?, ?, ?, ?)
		`

	_, err := r.Exec(iSqlString, user.Uid, user.LastLogined, user.CreatedAt, user.UpdatedAt)
	if err != nil {
		err = errors.Wrap(ae.DbError, err.Error())
		return err
	}
	return nil
}

package user

import (
	"fmt"
	"regexp"
	"testing"
	"time"

	"github.com/DATA-DOG/go-sqlmock"

	"github.com/k1350/sololog/internal/domain/model/user"
)

func TestIsExist(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		fmt.Println("failed to open sqlmock database:", err)
	}
	defer db.Close()

	rows := sqlmock.NewRows([]string{"count"}).
		AddRow(1)

	mock.ExpectQuery(regexp.QuoteMeta("SELECT count(1) FROM users WHERE uid = ?")).
		WithArgs("one").
		WillReturnRows(rows)

	repo := NewUserRepository(db)
	_, err = repo.IsExist("one")

	if err != nil {
		t.Error("got err:", err)
	}

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

func TestCreate(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		fmt.Println("failed to open sqlmock database:", err)
	}
	defer db.Close()

	now := time.Now()

	mock.ExpectExec(regexp.QuoteMeta("INSERT INTO users (uid, last_logined, created_at, updated_at) VALUES (?, ?, ?, ?)")).
		WithArgs("one", now, now, now).
		WillReturnResult(sqlmock.NewResult(1, 1))

	u := user.NewUser("one", now, now, now)

	repo := NewUserRepository(db)
	err = repo.Create(u)

	if err != nil {
		t.Error("got err:", err)
	}

	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}