github.com/litesolutions/justifay-api@v1.0.0-2.0.20220707114139-46f28a909481/mock/mockdb/user.go (about)

     1  package mockdb
     2  
     3  import (
     4  	"github.com/go-pg/pg/orm"
     5  	"github.com/litesolutions/justifay-api/model"
     6  )
     7  
     8  // User database mock
     9  type User struct {
    10  	// CreateFn          func(orm.DB, model.User) (*model.User, error)
    11  	// ViewFn            func(orm.DB, int64) (*model.User, error)
    12  	// ListFn            func(orm.DB, string, int, int) ([]model.User, error)
    13  	// DeleteFn          func(orm.DB, *model.User) error
    14  	// UpdateFn          func(orm.DB, *model.User) (*model.User, error)
    15  	FindByAuthFn      func(orm.DB, string) (*model.User, error)
    16  	FindByTokenFn     func(orm.DB, string) (*model.User, error)
    17  	UpdateLastLoginFn func(orm.DB, *model.User) error
    18  }
    19  
    20  // Create mock
    21  // func (u *User) Create(db orm.DB, usr model.User) (*model.User, error) {
    22  // 	return u.CreateFn(db, usr)
    23  // }
    24  //
    25  // // View mock
    26  // func (u *User) View(db orm.DB, id int64) (*model.User, error) {
    27  // 	return u.ViewFn(db, id)
    28  // }
    29  //
    30  // // List mock
    31  // func (u *User) List(db orm.DB, q string, limit, page int) ([]model.User, error) {
    32  // 	return u.ListFn(db, q, limit, page)
    33  // }
    34  //
    35  // // Delete mock
    36  // func (u *User) Delete(db orm.DB, usr *model.User) error {
    37  // 	return u.DeleteFn(db, usr)
    38  // }
    39  
    40  // Update mock
    41  // func (u *User) Update(db orm.DB, usr *model.User) (*model.User, error) {
    42  // 	return u.UpdateFn(db, usr)
    43  // }
    44  
    45  // FindByAuth mock
    46  func (u *User) FindByAuth(db orm.DB, auth string) (*model.User, error) {
    47  	return u.FindByAuthFn(db, auth)
    48  }
    49  
    50  // FindByToken mock
    51  func (u *User) FindByToken(db orm.DB, auth string) (*model.User, error) {
    52  	return u.FindByTokenFn(db, auth)
    53  }
    54  
    55  // UpdateLastLogin mock
    56  func (u *User) UpdateLastLogin(db orm.DB, usr *model.User) error {
    57  	return u.UpdateLastLoginFn(db, usr)
    58  }