github.com/mysteriumnetwork/node@v0.0.0-20240516044423-365054f76801/core/connection/registry_test.go (about)

     1  /*
     2   * Copyright (C) 2018 The "MysteriumNetwork/node" Authors.
     3   *
     4   * This program is free software: you can redistribute it and/or modify
     5   * it under the terms of the GNU General Public License as published by
     6   * the Free Software Foundation, either version 3 of the License, or
     7   * (at your option) any later version.
     8   *
     9   * This program is distributed in the hope that it will be useful,
    10   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12   * GNU General Public License for more details.
    13   *
    14   * You should have received a copy of the GNU General Public License
    15   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    16   */
    17  
    18  package connection
    19  
    20  import (
    21  	"testing"
    22  
    23  	"github.com/stretchr/testify/assert"
    24  )
    25  
    26  var _ Creator = (&Registry{}).CreateConnection
    27  
    28  var (
    29  	serviceType = "serviceType"
    30  )
    31  
    32  func TestRegistry_Factory(t *testing.T) {
    33  	registry := NewRegistry()
    34  	assert.Len(t, registry.creators, 0)
    35  }
    36  
    37  func TestRegistry_Register(t *testing.T) {
    38  	registry := Registry{
    39  		creators: map[string]Factory{},
    40  	}
    41  
    42  	registry.Register(serviceType, func() (connection Connection, err error) {
    43  		return &connectionMock{}, nil
    44  	})
    45  	assert.Len(t, registry.creators, 1)
    46  }
    47  
    48  func TestRegistry_CreateConnection_NonExisting(t *testing.T) {
    49  	registry := &Registry{}
    50  
    51  	connection, err := registry.CreateConnection(serviceType)
    52  	assert.Equal(t, ErrUnsupportedServiceType, err)
    53  	assert.Nil(t, connection)
    54  }
    55  
    56  func TestRegistry_CreateConnection_Existing(t *testing.T) {
    57  	mock := &connectionMock{}
    58  	registry := Registry{
    59  		creators: map[string]Factory{
    60  			"fake-service": func() (connection Connection, err error) {
    61  				return mock, nil
    62  			},
    63  		},
    64  	}
    65  
    66  	connection, err := registry.CreateConnection("fake-service")
    67  	assert.NoError(t, err)
    68  	assert.Equal(t, mock, connection)
    69  }