github.com/nitinawathare/ethereumassignment3@v0.0.0-20211021213010-f07344c2b868/go-ethereum/swarm/storage/localstore/mode_has_test.go (about)

     1  // Copyright 2019 The go-ethereum Authors
     2  // This file is part of the go-ethereum library.
     3  //
     4  // The go-ethereum library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser 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  // The go-ethereum library 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 Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package localstore
    18  
    19  import (
    20  	"testing"
    21  )
    22  
    23  // TestHas validates that Hasser is returning true for
    24  // the stored chunk and false for one that is not stored.
    25  func TestHas(t *testing.T) {
    26  	db, cleanupFunc := newTestDB(t, nil)
    27  	defer cleanupFunc()
    28  
    29  	chunk := generateTestRandomChunk()
    30  
    31  	err := db.NewPutter(ModePutUpload).Put(chunk)
    32  	if err != nil {
    33  		t.Fatal(err)
    34  	}
    35  
    36  	hasser := db.NewHasser()
    37  
    38  	has, err := hasser.Has(chunk.Address())
    39  	if err != nil {
    40  		t.Fatal(err)
    41  	}
    42  	if !has {
    43  		t.Error("chunk not found")
    44  	}
    45  
    46  	missingChunk := generateTestRandomChunk()
    47  
    48  	has, err = hasser.Has(missingChunk.Address())
    49  	if err != nil {
    50  		t.Fatal(err)
    51  	}
    52  	if has {
    53  		t.Error("unexpected chunk is found")
    54  	}
    55  }