51 lines
1.0 KiB
Go

package mocks
import (
"time"
"git.32bit.cafe/32bitcafe/guestbook/internal/models"
)
var mockWebsite = models.Website{
ID: 1,
ShortId: 1,
Name: "Example",
SiteUrl: "example.com",
AuthorName: "John Test",
UserId: 1,
Created: time.Now(),
Guestbook: mockGuestbook,
}
type WebsiteModel struct{}
func (m *WebsiteModel) Insert(shortId uint64, userId int64, siteName, siteUrl, authorName string) (int64, error) {
return 2, nil
}
func (m *WebsiteModel) Get(shortId uint64) (models.Website, error) {
switch shortId {
case 1:
return mockWebsite, nil
default:
return models.Website{}, models.ErrNoRecord
}
}
func (m *WebsiteModel) GetAllUser(userId int64) ([]models.Website, error) {
return []models.Website{mockWebsite}, nil
}
func (m *WebsiteModel) GetById(id int64) (models.Website, error) {
switch id {
case 1:
return mockWebsite, nil
default:
return models.Website{}, models.ErrNoRecord
}
}
func (m *WebsiteModel) GetAll() ([]models.Website, error) {
return []models.Website{mockWebsite}, nil
}