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") }