Luciano Milani
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package client_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/techxcar/backend/internal/client"
|
|
)
|
|
|
|
func getTestConn(t *testing.T) *pgxpool.Conn {
|
|
t.Helper()
|
|
dsn := os.Getenv("TEST_DATABASE_URL")
|
|
if dsn == "" {
|
|
t.Skip("TEST_DATABASE_URL not set")
|
|
}
|
|
pool, err := pgxpool.New(context.Background(), dsn)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { pool.Close() })
|
|
|
|
conn, err := pool.Acquire(context.Background())
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { conn.Release() })
|
|
|
|
_, err = conn.Exec(context.Background(), "SET search_path = tenant_test, public")
|
|
require.NoError(t, err)
|
|
return conn
|
|
}
|
|
|
|
func TestClientCRUD(t *testing.T) {
|
|
conn := getTestConn(t)
|
|
ctx := context.Background()
|
|
|
|
c, err := client.CreateClient(ctx, conn, "João Silva", "123456789", "912345678", "joao@example.com", "Rua A", "")
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, c.ID)
|
|
assert.Equal(t, "João Silva", c.Name)
|
|
|
|
list, err := client.ListClients(ctx, conn)
|
|
require.NoError(t, err)
|
|
assert.GreaterOrEqual(t, len(list), 1)
|
|
|
|
got, err := client.GetClient(ctx, conn, c.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, c.ID, got.ID)
|
|
|
|
updated, err := client.UpdateClient(ctx, conn, c.ID, "João Santos", "987654321", "921000000", "joao2@example.com", "Rua B", "nota")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "João Santos", updated.Name)
|
|
|
|
err = client.DeleteClient(ctx, conn, c.ID)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestVehicleCRUD(t *testing.T) {
|
|
conn := getTestConn(t)
|
|
ctx := context.Background()
|
|
|
|
c, err := client.CreateClient(ctx, conn, "Test Client", "", "", "", "", "")
|
|
require.NoError(t, err)
|
|
|
|
v, err := client.CreateVehicle(ctx, conn, c.ID, "AA-00-AA", "Toyota", "Yaris", 2020, "", 50000, "")
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, v.ID)
|
|
assert.Equal(t, "AA-00-AA", v.Plate)
|
|
|
|
list, err := client.ListVehiclesByClient(ctx, conn, c.ID)
|
|
require.NoError(t, err)
|
|
assert.Len(t, list, 1)
|
|
|
|
updated, err := client.UpdateVehicle(ctx, conn, v.ID, "BB-11-BB", "Toyota", "Yaris", 2021, "", 60000, "nota")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "BB-11-BB", updated.Plate)
|
|
}
|
|
|