Function TestRequestLimit

lib/model/model_test.go:2964–3009

(t *testing.T)

}

func TestRequestLimit(t *testing.T) {
	wrapper, fcfg := newDefaultCfgWrapper(t)
	ffs := fcfg.Filesystem()

	file := "tmpfile"
	fd, err := ffs.Create(file)
	must(t, err)
	fd.Close()
	waiter, err := wrapper.Modify(func(cfg *config.Configuration) {
		_, i, _ := cfg.Device(device1)
		cfg.Devices[i].MaxRequestKiB = 1
	})
	must(t, err)
	waiter.Wait()
	m, conn := setupModelWithConnectionFromWrapper(t, wrapper)
	defer cleanupModel(m)
	m.ScanFolder("default")

	befReq := time.Now()
	first, err := m.Request(conn, &protocol.Request{Folder: "default", Name: file, Size: 2000})
	if err != nil {
		t.Fatalf("First request failed: %v", err)
	}
	reqDur := time.Since(befReq)
	returned := make(chan struct{})
	go func() {
		second, err := m.Request(conn, &protocol.Request{Folder: "default", Name: file, Size: 2000})
		if err != nil {
			t.Errorf("Second request failed: %v", err)
		}
		close(returned)
		second.Close()
	}()
	time.Sleep(10 * reqDur)
	select {
	case <-returned:
		t.Fatalf("Second request returned before first was done")
	default:
	}
	first.Close()
	select {
	case <-returned:
	case <-time.After(time.Second):
		t.Fatalf("Second request did not return after first was done")
	}
}

// TestConnCloseOnRestart checks that there is no deadlock when calling Close
// on a protocol connection that has a blocking reader (blocking writer can't