registrar_test.go

Documentation: github.com/go-kit/kit/sd/eureka

     1  package eureka
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  )
     7  
     8  func TestRegistrar(t *testing.T) {
     9  	connection := &testConnection{
    10  		errHeartbeat: errTest,
    11  	}
    12  
    13  	registrar1 := NewRegistrar(connection, instanceTest1, loggerTest)
    14  	registrar2 := NewRegistrar(connection, instanceTest2, loggerTest)
    15  
    16  	// Not registered.
    17  	registrar1.Deregister()
    18  	if want, have := 0, len(connection.instances); want != have {
    19  		t.Errorf("want %d, have %d", want, have)
    20  	}
    21  
    22  	// Register.
    23  	registrar1.Register()
    24  	if want, have := 1, len(connection.instances); want != have {
    25  		t.Errorf("want %d, have %d", want, have)
    26  	}
    27  
    28  	registrar2.Register()
    29  	if want, have := 2, len(connection.instances); want != have {
    30  		t.Errorf("want %d, have %d", want, have)
    31  	}
    32  
    33  	// Deregister.
    34  	registrar1.Deregister()
    35  	if want, have := 1, len(connection.instances); want != have {
    36  		t.Errorf("want %d, have %d", want, have)
    37  	}
    38  
    39  	// Already registered.
    40  	registrar1.Register()
    41  	if want, have := 2, len(connection.instances); want != have {
    42  		t.Errorf("want %d, have %d", want, have)
    43  	}
    44  	registrar1.Register()
    45  	if want, have := 2, len(connection.instances); want != have {
    46  		t.Errorf("want %d, have %d", want, have)
    47  	}
    48  
    49  	// Wait for a heartbeat failure.
    50  	time.Sleep(1010 * time.Millisecond)
    51  	if want, have := 2, len(connection.instances); want != have {
    52  		t.Errorf("want %d, have %d", want, have)
    53  	}
    54  	registrar1.Deregister()
    55  	if want, have := 1, len(connection.instances); want != have {
    56  		t.Errorf("want %d, have %d", want, have)
    57  	}
    58  }
    59  
    60  func TestBadRegister(t *testing.T) {
    61  	connection := &testConnection{
    62  		errRegister: errTest,
    63  	}
    64  
    65  	registrar := NewRegistrar(connection, instanceTest1, loggerTest)
    66  	registrar.Register()
    67  	if want, have := 0, len(connection.instances); want != have {
    68  		t.Errorf("want %d, have %d", want, have)
    69  	}
    70  }
    71  
    72  func TestBadDeregister(t *testing.T) {
    73  	connection := &testConnection{
    74  		errDeregister: errTest,
    75  	}
    76  
    77  	registrar := NewRegistrar(connection, instanceTest1, loggerTest)
    78  	registrar.Register()
    79  	if want, have := 1, len(connection.instances); want != have {
    80  		t.Errorf("want %d, have %d", want, have)
    81  	}
    82  	registrar.Deregister()
    83  	if want, have := 1, len(connection.instances); want != have {
    84  		t.Errorf("want %d, have %d", want, have)
    85  	}
    86  }
    87  
    88  func TestExpiredInstance(t *testing.T) {
    89  	connection := &testConnection{
    90  		errHeartbeat: errNotFound,
    91  	}
    92  
    93  	registrar := NewRegistrar(connection, instanceTest1, loggerTest)
    94  	registrar.Register()
    95  
    96  	// Wait for a heartbeat failure.
    97  	time.Sleep(1010 * time.Millisecond)
    98  
    99  	if want, have := 1, len(connection.instances); want != have {
   100  		t.Errorf("want %d, have %d", want, have)
   101  	}
   102  }
   103
View as plain text