package mock_db

import (
	"context"

	"gorm.io/gorm"
)

type fakeTransProvider struct {
	db *gorm.DB
}

func NewFakeTransProvider(db *gorm.DB) *fakeTransProvider {
	return &fakeTransProvider{
		db: db,
	}
}

func (p *fakeTransProvider) ReadDB(ctx context.Context) *gorm.DB {
	return p.db
}

func (p *fakeTransProvider) WriteDB(ctx context.Context) *gorm.DB {
	return p.db
}

func (p *fakeTransProvider) Transaction(ctx context.Context, cb func(ctx context.Context) error) error {
	return cb(ctx)
}