storj.io/minio@v0.0.0-20230509071714-0cbc90f649b1/cmd/bitrot_test.go (about)

     1  /*
     2   * MinIO Cloud Storage, (C) 2018 MinIO, Inc.
     3   *
     4   * Licensed under the Apache License, Version 2.0 (the "License");
     5   * you may not use this file except in compliance with the License.
     6   * You may obtain a copy of the License at
     7   *
     8   *     http://www.apache.org/licenses/LICENSE-2.0
     9   *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   * See the License for the specific language governing permissions and
    14   * limitations under the License.
    15   */
    16  
    17  package cmd
    18  
    19  import (
    20  	"context"
    21  	"io"
    22  	"io/ioutil"
    23  	"os"
    24  	"testing"
    25  )
    26  
    27  func testBitrotReaderWriterAlgo(t *testing.T, bitrotAlgo BitrotAlgorithm) {
    28  	tmpDir, err := ioutil.TempDir("", "")
    29  	if err != nil {
    30  		t.Fatal(err)
    31  	}
    32  	defer os.RemoveAll(tmpDir)
    33  
    34  	volume := "testvol"
    35  	filePath := "testfile"
    36  
    37  	disk, err := newLocalXLStorage(tmpDir)
    38  	if err != nil {
    39  		t.Fatal(err)
    40  	}
    41  
    42  	disk.MakeVol(context.Background(), volume)
    43  
    44  	writer := newBitrotWriter(disk, volume, filePath, 35, bitrotAlgo, 10, false)
    45  
    46  	_, err = writer.Write([]byte("aaaaaaaaaa"))
    47  	if err != nil {
    48  		t.Fatal(err)
    49  	}
    50  	_, err = writer.Write([]byte("aaaaaaaaaa"))
    51  	if err != nil {
    52  		t.Fatal(err)
    53  	}
    54  	_, err = writer.Write([]byte("aaaaaaaaaa"))
    55  	if err != nil {
    56  		t.Fatal(err)
    57  	}
    58  	_, err = writer.Write([]byte("aaaaa"))
    59  	if err != nil {
    60  		t.Fatal(err)
    61  	}
    62  	writer.(io.Closer).Close()
    63  
    64  	reader := newBitrotReader(disk, nil, volume, filePath, 35, bitrotAlgo, bitrotWriterSum(writer), 10)
    65  	b := make([]byte, 10)
    66  	if _, err = reader.ReadAt(b, 0); err != nil {
    67  		t.Fatal(err)
    68  	}
    69  	if _, err = reader.ReadAt(b, 10); err != nil {
    70  		t.Fatal(err)
    71  	}
    72  	if _, err = reader.ReadAt(b, 20); err != nil {
    73  		t.Fatal(err)
    74  	}
    75  	if _, err = reader.ReadAt(b[:5], 30); err != nil {
    76  		t.Fatal(err)
    77  	}
    78  }
    79  
    80  func TestAllBitrotAlgorithms(t *testing.T) {
    81  	for bitrotAlgo := range bitrotAlgorithms {
    82  		testBitrotReaderWriterAlgo(t, bitrotAlgo)
    83  	}
    84  }