github.com/SmartMeshFoundation/Spectrum@v0.0.0-20220621030607-452a266fee1e/cmd/utils/fdlimit_test.go (about)

     1  // Copyright 2016 The Spectrum Authors
     2  // This file is part of Spectrum.
     3  //
     4  // Spectrum is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // Spectrum is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU General Public License
    15  // along with Spectrum. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package utils
    18  
    19  import (
    20  	"fmt"
    21  	"testing"
    22  )
    23  
    24  // TestFileDescriptorLimits simply tests whether the file descriptor allowance
    25  // per this process can be retrieved.
    26  func TestFileDescriptorLimits(t *testing.T) {
    27  	target := 4096
    28  	hardlimit, err := getFdMaxLimit()
    29  	if err != nil {
    30  		t.Fatal(err)
    31  	}
    32  	if hardlimit < target {
    33  		t.Skip(fmt.Sprintf("system limit is less than desired test target: %d < %d", hardlimit, target))
    34  	}
    35  
    36  	if limit, err := getFdLimit(); err != nil || limit <= 0 {
    37  		t.Fatalf("failed to retrieve file descriptor limit (%d): %v", limit, err)
    38  	}
    39  	if err := raiseFdLimit(uint64(target)); err != nil {
    40  		t.Fatalf("failed to raise file allowance")
    41  	}
    42  	if limit, err := getFdLimit(); err != nil || limit < target {
    43  		t.Fatalf("failed to retrieve raised descriptor limit (have %v, want %v): %v", limit, target, err)
    44  	}
    45  }