package metrics

import "testing"

// TestReconnectCount connects three times with a disconnect between each, the
// way the streamer does, and expects every connection after the first to count
// one reconnect.
func TestReconnectCount(t *testing.T) {
	tracker := New()

	tracker.SetConnected(true)
	tracker.SetConnected(false)
	tracker.SetConnected(true)
	tracker.SetConnected(false)
	tracker.SetConnected(true)

	const wantReconnects = 2
	if got := tracker.GetReconnectCount(); got != wantReconnects {
		t.Fatalf("reconnect count = %d, want %d", got, wantReconnects)
	}
}