github.com/aigarnetwork/aigar@v0.0.0-20191115204914-d59a6eb70f8e/whisper/whisperv6/api_test.go (about)

     1  //  Copyright 2018 The go-ethereum Authors
     2  //  Copyright 2019 The go-aigar Authors
     3  //  This file is part of the go-aigar library.
     4  //
     5  //  The go-aigar library is free software: you can redistribute it and/or modify
     6  //  it under the terms of the GNU Lesser General Public License as published by
     7  //  the Free Software Foundation, either version 3 of the License, or
     8  //  (at your option) any later version.
     9  //
    10  //  The go-aigar library is distributed in the hope that it will be useful,
    11  //  but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    13  //  GNU Lesser General Public License for more details.
    14  //
    15  //  You should have received a copy of the GNU Lesser General Public License
    16  //  along with the go-aigar library. If not, see <http://www.gnu.org/licenses/>.
    17  
    18  package whisperv6
    19  
    20  import (
    21  	"bytes"
    22  	"testing"
    23  	"time"
    24  )
    25  
    26  func TestMultipleTopicCopyInNewMessageFilter(t *testing.T) {
    27  	w := New(nil)
    28  
    29  	keyID, err := w.GenerateSymKey()
    30  	if err != nil {
    31  		t.Fatalf("Error generating symmetric key: %v", err)
    32  	}
    33  	api := PublicWhisperAPI{
    34  		w:        w,
    35  		lastUsed: make(map[string]time.Time),
    36  	}
    37  
    38  	t1 := [4]byte{0xde, 0xea, 0xbe, 0xef}
    39  	t2 := [4]byte{0xca, 0xfe, 0xde, 0xca}
    40  
    41  	crit := Criteria{
    42  		SymKeyID: keyID,
    43  		Topics:   []TopicType{TopicType(t1), TopicType(t2)},
    44  	}
    45  
    46  	_, err = api.NewMessageFilter(crit)
    47  	if err != nil {
    48  		t.Fatalf("Error creating the filter: %v", err)
    49  	}
    50  
    51  	found := false
    52  	candidates := w.filters.getWatchersByTopic(TopicType(t1))
    53  	for _, f := range candidates {
    54  		if len(f.Topics) == 2 {
    55  			if bytes.Equal(f.Topics[0], t1[:]) && bytes.Equal(f.Topics[1], t2[:]) {
    56  				found = true
    57  			}
    58  		}
    59  	}
    60  
    61  	if !found {
    62  		t.Fatalf("Could not find filter with both topics")
    63  	}
    64  }