refactor sqlite queue
This commit is contained in:
@@ -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{}
|
||||
|
||||
Reference in New Issue
Block a user