refactor sqlite queue

This commit is contained in:
2025-01-02 17:32:33 -06:00
parent c2c49f6bf5
commit f112171145
17 changed files with 204 additions and 153 deletions

View File

@@ -1,12 +1,15 @@
package api
import (
"context"
"github.com/stretchr/testify/assert"
"go-nkode/internal/db"
"go-nkode/internal/email"
"go-nkode/internal/entities"
"go-nkode/internal/models"
"go-nkode/internal/repository"
"go-nkode/internal/security"
sqlite_queue "go-nkode/internal/sqlc"
"log"
"os"
"testing"
)
@@ -15,22 +18,31 @@ func TestNKodeAPI(t *testing.T) {
//db1 := NewInMemoryDb()
//testNKodeAPI(t, &db1)
dbFile := os.Getenv("TEST_DB")
db2, err := db.NewSqliteDB(dbFile)
dbPath := os.Getenv("TEST_DB")
ctx := context.Background()
sqliteDb, err := sqlite_queue.OpenSqliteDb(dbPath)
assert.NoError(t, err)
defer db2.Close()
testNKodeAPI(t, db2)
//if _, err := os.Stat(dbFile); err == nil {
// err = os.Remove(dbFile)
queue, err := sqlite_queue.NewQueue(sqliteDb, ctx)
assert.NoError(t, err)
queue.Start()
defer func(queue *sqlite_queue.Queue) {
if err := queue.Stop(); err != nil {
log.Fatal(err)
}
}(queue)
sqlitedb := repository.NewSqliteRepository(queue, ctx)
testNKodeAPI(t, &sqlitedb)
//if _, err := os.Stat(dbPath); err == nil {
// err = os.Remove(dbPath)
// assert.NoError(t, err)
//} else {
// assert.NoError(t, err)
//}
}
func testNKodeAPI(t *testing.T, db db.CustomerUserRepository) {
func testNKodeAPI(t *testing.T, db repository.CustomerUserRepository) {
bufferSize := 100
emailsPerSec := 14
testClient := email.TestEmailClient{}