client_test.go 820 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package client
  2. import (
  3. "context"
  4. "testing"
  5. "github.com/stretchr/testify/suite"
  6. "google.golang.org/grpc"
  7. "google.golang.org/grpc/credentials/insecure"
  8. )
  9. type ClientSuite struct {
  10. MockSuiteBase
  11. }
  12. func (s *ClientSuite) TestNewClient() {
  13. ctx, cancel := context.WithCancel(context.Background())
  14. defer cancel()
  15. s.Run("Use bufconn dailer, testing case", func() {
  16. c, err := New(ctx,
  17. &ClientConfig{
  18. Address: "bufnet",
  19. DialOptions: []grpc.DialOption{
  20. grpc.WithBlock(),
  21. grpc.WithTransportCredentials(insecure.NewCredentials()),
  22. grpc.WithContextDialer(s.mockDialer),
  23. },
  24. })
  25. s.NoError(err)
  26. s.NotNil(c)
  27. })
  28. s.Run("empty_addr", func() {
  29. _, err := New(ctx, &ClientConfig{})
  30. s.Error(err)
  31. s.T().Log(err)
  32. })
  33. }
  34. func TestClient(t *testing.T) {
  35. suite.Run(t, new(ClientSuite))
  36. }