Esempio n. 1
0
func TestQos(t *testing.T) {
	highWater := 16
	lowWater := 4

	callEnterHighWatertimes := 0
	callEnterLowWatertimes := 0

	qos := tunnel.NewQos(highWater, lowWater, func() {
		callEnterHighWatertimes += 1
	}, func() {
		callEnterLowWatertimes += 1
	})

	qos.SetWater(highWater)
	qos.SetWater(lowWater)

	if callEnterHighWatertimes != 1 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterHighWatertimes, 1)
	}
	if callEnterLowWatertimes != 1 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterLowWatertimes, 1)
	}
	qos.SetWater(lowWater)
	qos.SetWater(highWater)
	qos.SetWater(highWater - 1)
	qos.SetWater(lowWater)
	if callEnterHighWatertimes != 2 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterHighWatertimes, 2)
	}
	if callEnterLowWatertimes != 2 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterLowWatertimes, 2)
	}
}
Esempio n. 2
0
func TestQosRemote(t *testing.T) {
	highWater := 16
	lowWater := 4

	callEnterHighWatertimes := 0
	callEnterLowWatertimes := 0

	qos := tunnel.NewQos(highWater, lowWater, func() {
		callEnterHighWatertimes += 1
	}, func() {
		callEnterLowWatertimes += 1
	})

	qos.Balance()
	qos.SetRemoteFlag(true)
	go qos.SetRemoteFlag(false)
	qos.Balance()

	qos.SetRemoteFlag(true)
	go qos.Close()
	qos.Balance()

	if callEnterHighWatertimes != 0 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterHighWatertimes, 0)
	}
	if callEnterLowWatertimes != 0 {
		t.Errorf("unexpected call enter low times:%d(%d)", callEnterLowWatertimes, 0)
	}
}