func newLogger(faculty string, args ...string) *Logger {
cnfBld := conf.Builder{Defaults: conf.DEFAULTS}
cnf, err := cnfBld.Build()
if err != nil {
panic(fmt.Sprintf("failed to create conf: %s", err.Error()))
}
cnf2 := cnf.Clone(args...)
lg := NewLogger(faculty, cnf2)
return lg
}
func TestHeartbeaterStartupShutdown(t *testing.T) {
cnfBld := conf.Builder{
Values: conf.TEST_VALUES(),
Defaults: conf.DEFAULTS,
}
cnf, err := cnfBld.Build()
if err != nil {
t.Fatalf("failed to create conf: %s", err.Error())
}
lg := common.NewLogger("heartbeater", cnf)
hb := NewHeartbeater("ExampleHeartbeater", 1, lg)
if hb.String() != "ExampleHeartbeater" {
t.Fatalf("hb.String() returned %s instead of %s\n", hb.String(), "ExampleHeartbeater")
}
hb.Shutdown()
}
// Run the helper process which TestSignals spawns.
func runHelperProcess() {
cnfMap := map[string]string{
conf.HTRACE_LOG_LEVEL: "TRACE",
conf.HTRACE_LOG_PATH: "", // log to stdout
}
cnfBld := conf.Builder{Values: cnfMap, Defaults: conf.DEFAULTS}
cnf, err := cnfBld.Build()
if err != nil {
fmt.Printf("Error building configuration: %s\n", err.Error())
os.Exit(1)
}
InstallSignalHandlers(cnf)
fmt.Fprintf(os.Stderr, "Signal handler installed.\n")
// Wait for a signal to be delivered
for {
time.Sleep(time.Hour * 100)
}
}
func TestHeartbeaterSendsHeartbeats(t *testing.T) {
cnfBld := conf.Builder{
Values: conf.TEST_VALUES(),
Defaults: conf.DEFAULTS,
}
cnf, err := cnfBld.Build()
if err != nil {
t.Fatalf("failed to create conf: %s", err.Error())
}
lg := common.NewLogger("heartbeater", cnf)
// The minimum amount of time which the heartbeater test should take
MINIMUM_TEST_DURATION := time.Millisecond * (NUM_TEST_HEARTBEATS * HEARTBEATER_PERIOD)
duration := MINIMUM_TEST_DURATION
for duration <= MINIMUM_TEST_DURATION {
start := time.Now()
testHeartbeaterSendsHeartbeatsImpl(t, lg)
end := time.Now()
duration = end.Sub(start)
lg.Debugf("Measured duration: %v; minimum expected duration: %v\n",
duration, MINIMUM_TEST_DURATION)
}
}
func TestMetricsSinkPerHostEviction(t *testing.T) {
cnfBld := conf.Builder{
Values: conf.TEST_VALUES(),
Defaults: conf.DEFAULTS,
}
cnfBld.Values[conf.HTRACE_METRICS_MAX_ADDR_ENTRIES] = "2"
cnf, err := cnfBld.Build()
if err != nil {
t.Fatalf("failed to create conf: %s", err.Error())
}
msink := NewMetricsSink(cnf)
msink.UpdatePersisted("192.168.0.100", 20, 10)
msink.UpdatePersisted("192.168.0.101", 20, 10)
msink.UpdatePersisted("192.168.0.102", 20, 10)
msink.lock.Lock()
defer msink.lock.Unlock()
if len(msink.HostSpanMetrics) != 2 {
for k, v := range msink.HostSpanMetrics {
fmt.Printf("WATERMELON: [%s] = [%s]\n", k, v)
}
t.Fatalf("Expected len(msink.HostSpanMetrics) to be 2, but got %d\n",
len(msink.HostSpanMetrics))
}
}
func (bld *MiniHTracedBuilder) Build() (*MiniHTraced, error) {
var err error
var store *dataStore
var rsv *RestServer
var hsv *HrpcServer
if bld.Name == "" {
bld.Name = "HTraceTest"
}
if bld.Cnf == nil {
bld.Cnf = make(map[string]string)
}
if bld.DataDirs == nil {
bld.DataDirs = make([]string, 2)
}
for idx := range bld.DataDirs {
if bld.DataDirs[idx] == "" {
bld.DataDirs[idx], err = ioutil.TempDir(os.TempDir(),
fmt.Sprintf("%s%d", bld.Name, idx+1))
if err != nil {
return nil, err
}
}
}
// Copy the default test configuration values.
for k, v := range conf.TEST_VALUES() {
_, hasVal := bld.Cnf[k]
if !hasVal {
bld.Cnf[k] = v
}
}
bld.Cnf[conf.HTRACE_DATA_STORE_DIRECTORIES] =
strings.Join(bld.DataDirs, conf.PATH_LIST_SEP)
cnfBld := conf.Builder{Values: bld.Cnf, Defaults: conf.DEFAULTS}
cnf, err := cnfBld.Build()
if err != nil {
return nil, err
}
lg := common.NewLogger("mini.htraced", cnf)
defer func() {
if err != nil {
if store != nil {
store.Close()
}
for idx := range bld.DataDirs {
if !bld.KeepDataDirsOnClose {
if bld.DataDirs[idx] != "" {
os.RemoveAll(bld.DataDirs[idx])
}
}
}
if rsv != nil {
rsv.Close()
}
lg.Infof("Failed to create MiniHTraced %s: %s\n", bld.Name, err.Error())
lg.Close()
}
}()
store, err = CreateDataStore(cnf, bld.WrittenSpans)
if err != nil {
return nil, err
}
rstListener, listenErr := net.Listen("tcp", cnf.Get(conf.HTRACE_WEB_ADDRESS))
if listenErr != nil {
return nil, listenErr
}
defer func() {
if rstListener != nil {
rstListener.Close()
}
}()
rsv, err = CreateRestServer(cnf, store, rstListener)
if err != nil {
return nil, err
}
rstListener = nil
hsv, err = CreateHrpcServer(cnf, store, bld.HrpcTestHooks)
if err != nil {
return nil, err
}
lg.Infof("Created MiniHTraced %s\n", bld.Name)
return &MiniHTraced{
Name: bld.Name,
Cnf: cnf,
DataDirs: bld.DataDirs,
Store: store,
Rsv: rsv,
Hsv: hsv,
Lg: lg,
KeepDataDirsOnClose: bld.KeepDataDirsOnClose,
}, nil
}
func (bld *MiniHTracedBuilder) Build() (*MiniHTraced, error) {
var err error
var store *dataStore
var rsv *RestServer
var hsv *HrpcServer
if bld.Name == "" {
bld.Name = "HTraceTest"
}
if bld.Cnf == nil {
bld.Cnf = make(map[string]string)
}
if bld.DataDirs == nil {
bld.DataDirs = make([]string, 2)
}
for idx := range bld.DataDirs {
if bld.DataDirs[idx] == "" {
bld.DataDirs[idx], err = ioutil.TempDir(os.TempDir(),
fmt.Sprintf("%s%d", bld.Name, idx+1))
if err != nil {
return nil, err
}
}
}
bld.Cnf[conf.HTRACE_DATA_STORE_DIRECTORIES] =
strings.Join(bld.DataDirs, conf.PATH_LIST_SEP)
bld.Cnf[conf.HTRACE_WEB_ADDRESS] = ":0" // use a random port for the REST server
bld.Cnf[conf.HTRACE_HRPC_ADDRESS] = ":0" // use a random port for the HRPC server
bld.Cnf[conf.HTRACE_LOG_LEVEL] = "TRACE"
cnfBld := conf.Builder{Values: bld.Cnf, Defaults: conf.DEFAULTS}
cnf, err := cnfBld.Build()
if err != nil {
return nil, err
}
lg := common.NewLogger("mini.htraced", cnf)
defer func() {
if err != nil {
if store != nil {
store.Close()
}
for idx := range bld.DataDirs {
if bld.DataDirs[idx] != "" {
os.RemoveAll(bld.DataDirs[idx])
}
}
if rsv != nil {
rsv.Close()
}
lg.Infof("Failed to create MiniHTraced %s: %s\n", bld.Name, err.Error())
lg.Close()
}
}()
store, err = CreateDataStore(cnf, bld.WrittenSpans)
if err != nil {
return nil, err
}
rsv, err = CreateRestServer(cnf, store)
if err != nil {
return nil, err
}
hsv, err = CreateHrpcServer(cnf, store)
if err != nil {
return nil, err
}
lg.Infof("Created MiniHTraced %s\n", bld.Name)
return &MiniHTraced{
Name: bld.Name,
Cnf: cnf,
DataDirs: bld.DataDirs,
Store: store,
Rsv: rsv,
Hsv: hsv,
Lg: lg,
KeepDataDirsOnClose: bld.KeepDataDirsOnClose,
}, nil
}