func TestEnsureRegistered(t *testing.T) {
_, counter, err := fdcount.Matching("TCP")
if err != nil {
t.Fatalf("Unable to get starting fdcount: %v", err)
}
u := getUtil()
// Test with no existing record
name, ip := "cfl-test-entry", "127.0.0.1"
rec, proxying, err := u.EnsureRegistered(name, ip, nil)
if assert.NoError(t, err, "Should be able to register with no record") {
assert.NotNil(t, rec, "A new record should have been returned")
assert.True(t, proxying, "Proxying (orange cloud) should be on")
}
// Test with existing record, but not passing it in
rec, proxying, err = u.EnsureRegistered(name, ip, nil)
if assert.NoError(t, err, "Should be able to register with unspecified existing record") {
assert.NotNil(t, rec, "Existing record should have been returned")
assert.True(t, proxying, "Proxying (orange cloud) should be on")
// Test with existing record, passing it in
rec, proxying, err = u.EnsureRegistered(name, ip, rec)
if assert.NoError(t, err, "Should be able to register with specified existing record") {
assert.NotNil(t, rec, "Existing record should have been returned")
assert.True(t, proxying, "Proxying (orange cloud) should be on")
}
}
if rec != nil {
err := u.DestroyRecord(rec)
assert.NoError(t, err, "Should be able to destroy record")
}
assert.NoError(t, counter.AssertDelta(0), "All file descriptors should have been closed")
}
func TestTimeout(t *testing.T) {
text, timedOut, err := Do(10*time.Millisecond, func() (interface{}, error) {
time.Sleep(11 * time.Millisecond)
return expectedText, expectedErr
})
assert.True(t, timedOut, "Should have timed out")
assert.NotNil(t, err, "There should be an error")
assert.Nil(t, text, "Text should be nil")
assert.Equal(t, timeoutErrorString, err.Error(), "Error should contain correct string")
}
func TestOKWithServerName(t *testing.T) {
fdStart := countTCPFiles()
cwt, err := DialForTimings(new(net.Dialer), "tcp", ADDR, true, &tls.Config{
RootCAs: cert.PoolContainingCert(),
})
conn := cwt.Conn
assert.NoError(t, err, "Unable to dial")
serverName := <-receivedServerNames
assert.Equal(t, "localhost", serverName, "Unexpected ServerName on server")
assert.NotNil(t, cwt.ResolvedAddr, "Should have resolved addr")
closeAndCountFDs(t, conn, err, fdStart)
}
func TestRegister(t *testing.T) {
_, counter, err := fdcount.Matching("TCP")
if err != nil {
t.Fatalf("Unable to get starting fdcount: %v", err)
}
u := getUtil()
// Test with no existing record
name, ip := "dsp-test-entry", "127.0.0.1"
rec, err := u.Register(name, ip)
if assert.NoError(t, err, "Should be able to register with no record") {
assert.NotNil(t, rec, "A new record should have been returned")
}
if rec != nil {
err := u.DestroyRecord(rec)
assert.NoError(t, err, "Should be able to destroy record")
}
assert.NoError(t, counter.AssertDelta(0), "All file descriptors should have been closed")
}