refactor: use shared mocks for tests

This commit is contained in:
talksik
2026-04-22 09:06:10 -07:00
parent 2e9ff742a0
commit 110f4594be
10 changed files with 304 additions and 174 deletions
+10 -10
View File
@@ -6,12 +6,12 @@ import (
"os"
"testing"
"github.com/flowy-live/llink/internal/billing"
"github.com/flowy-live/llink/internal/network"
"github.com/flowy-live/llink/internal/particle"
"github.com/flowy-live/llink/internal/testhelper"
"github.com/flowy-live/llink/internal/testhelper/mocks/mocknetwork"
"github.com/jackc/pgx/v5/pgxpool"
"github.com/stretchr/testify/assert"
"go.uber.org/mock/gomock"
)
var dbPool *pgxpool.Pool
@@ -34,7 +34,7 @@ func getStreamStatus(data json.RawMessage) string {
func TestParticleService_CreateAndGet(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network first
@@ -73,7 +73,7 @@ func TestParticleService_CreateAndGet(t *testing.T) {
func TestParticleService_NestedParticles(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network
@@ -119,7 +119,7 @@ func TestParticleService_NestedParticles(t *testing.T) {
func TestParticleService_CustomVisibility(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network with a member
@@ -161,7 +161,7 @@ func TestParticleService_CustomVisibility(t *testing.T) {
func TestParticleService_UpdateAndDelete(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network
@@ -199,7 +199,7 @@ func TestParticleService_UpdateAndDelete(t *testing.T) {
func TestParticleService_ListRootParticles(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network
@@ -225,7 +225,7 @@ func TestParticleService_ListRootParticles(t *testing.T) {
func TestParticleService_OpenCloseStream(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network
@@ -273,7 +273,7 @@ func TestParticleService_OpenCloseStream(t *testing.T) {
func TestParticleService_NotAStream(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network
@@ -302,7 +302,7 @@ func TestParticleService_NotAStream(t *testing.T) {
func TestParticleService_AccessInheritance(t *testing.T) {
ctx := context.Background()
networkSvc := network.NewService(dbPool, nil, billing.Noop(), nil)
networkSvc := testhelper.NewNetworkService(t, dbPool)
svc := particle.NewService(dbPool, networkSvc)
// Create a network with members