data_management_service_test.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. package service
  2. import (
  3. "context"
  4. "path/filepath"
  5. "testing"
  6. infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
  7. "github.com/stretchr/testify/require"
  8. )
  9. func TestDataManagementService_GetAgentHealth_Deprecated(t *testing.T) {
  10. t.Parallel()
  11. socketPath := filepath.Join(t.TempDir(), "unused.sock")
  12. svc := NewDataManagementServiceWithOptions(socketPath, 0)
  13. health := svc.GetAgentHealth(context.Background())
  14. require.False(t, health.Enabled)
  15. require.Equal(t, DataManagementDeprecatedReason, health.Reason)
  16. require.Equal(t, socketPath, health.SocketPath)
  17. require.Nil(t, health.Agent)
  18. }
  19. func TestDataManagementService_EnsureAgentEnabled_Deprecated(t *testing.T) {
  20. t.Parallel()
  21. socketPath := filepath.Join(t.TempDir(), "unused.sock")
  22. svc := NewDataManagementServiceWithOptions(socketPath, 100)
  23. err := svc.EnsureAgentEnabled(context.Background())
  24. require.Error(t, err)
  25. statusCode, status := infraerrors.ToHTTP(err)
  26. require.Equal(t, 503, statusCode)
  27. require.Equal(t, DataManagementDeprecatedReason, status.Reason)
  28. require.Equal(t, socketPath, status.Metadata["socket_path"])
  29. }