path: root/whisper/topic_test.go

package whisper

import (
    "bytes"
    "testing"
)

var topicCreationTests = []struct {
    data []byte
    hash [4]byte
}{
    {hash: [4]byte{0x8f, 0x9a, 0x2b, 0x7d}, data: []byte("test name")},
    {hash: [4]byte{0xf2, 0x6e, 0x77, 0x79}, data: []byte("some other test")},
}

func TestTopicCreation(t *testing.T) {
    // Create the topics individually
    for i, tt := range topicCreationTests {
        topic := NewTopic(tt.data)
        if bytes.Compare(topic[:], tt.hash[:]) != 0 {
            t.Errorf("binary test %d: hash mismatch: have %v, want %v.", i, topic, tt.hash)
        }
    }
    for i, tt := range topicCreationTests {
        topic := NewTopicFromString(string(tt.data))
        if bytes.Compare(topic[:], tt.hash[:]) != 0 {
            t.Errorf("textual test %d: hash mismatch: have %v, want %v.", i, topic, tt.hash)
        }
    }
    // Create the topics in batches
    binaryData := make([][]byte, len(topicCreationTests))
    for i, tt := range topicCreationTests {
        binaryData[i] = tt.data
    }
    textualData := make([]string, len(topicCreationTests))
    for i, tt := range topicCreationTests {
        textualData[i] = string(tt.data)
    }

    topics := NewTopics(binaryData...)
    for i, tt := range topicCreationTests {
        if bytes.Compare(topics[i][:], tt.hash[:]) != 0 {
            t.Errorf("binary batch test %d: hash mismatch: have %v, want %v.", i, topics[i], tt.hash)
        }
    }
    topics = NewTopicsFromStrings(textualData...)
    for i, tt := range topicCreationTests {
        if bytes.Compare(topics[i][:], tt.hash[:]) != 0 {
            t.Errorf("textual batch test %d: hash mismatch: have %v, want %v.", i, topics[i], tt.hash)
        }
    }
}

var topicMatcherCreationTest = struct {
    binary  [][][]byte
    textual [][]string
    matcher []map[[4]byte]struct{}
}{
    binary: [][][]byte{
        [][]byte{},
        [][]byte{
            []byte("Topic A"),
        },
        [][]byte{
            []byte("Topic B1"),
            []byte("Topic B2"),
            []byte("Topic B3"),
        },
    },
    textual: [][]string{
        []string{},
        []string{"Topic A"},
        []string{"Topic B1", "Topic B2", "Topic B3"},
    },
    matcher: []map[[4]byte]struct{}{
        map[[4]byte]struct{}{},
        map[[4]byte]struct{}{
            [4]byte{0x25, 0xfc, 0x95, 0x66}: struct{}{},
        },
        map[[4]byte]struct{}{
            [4]byte{0x93, 0x6d, 0xec, 0x09}: struct{}{},
            [4]byte{0x25, 0x23, 0x34, 0xd3}: struct{}{},
            [4]byte{0x6b, 0xc2, 0x73, 0xd1}: struct{}{},
        },
    },
}

func TestTopicMatcherCreation(t *testing.T) {
    test := topicMatcherCreationTest

    matcher := newTopicMatcherFromBinary(test.binary...)
    for i, cond := range matcher.conditions {
        for topic, _ := range cond {
            if _, ok := test.matcher[i][topic]; !ok {
                t.Errorf("condition %d; extra topic found: 0x%x", i, topic[:])
            }
        }
    }
    for i, cond := range test.matcher {
        for topic, _ := range cond {
            if _, ok := matcher.conditions[i][topic]; !ok {
                t.Errorf("condition %d; topic not found: 0x%x", i, topic[:])
            }
        }
    }

    matcher = newTopicMatcherFromStrings(test.textual...)
    for i, cond := range matcher.conditions {
        for topic, _ := range cond {
            if _, ok := test.matcher[i][topic]; !ok {
                t.Errorf("condition %d; extra topic found: 0x%x", i, topic[:])
            }
        }
    }
    for i, cond := range test.matcher {
        for topic, _ := range cond {
            if _, ok := matcher.conditions[i][topic]; !ok {
                t.Errorf("condition %d; topic not found: 0x%x", i, topic[:])
            }
        }
    }
}

var topicMatcherTests = []struct {
    filter [][]string
    topics []string
    match  bool
}{
    // Empty topic matcher should match everything
    {
        filter: [][]string{},
        topics: []string{},
        match:  true,
    },
    {
        filter: [][]string{},
        topics: []string{"a", "b", "c"},
        match:  true,
    },
    // Fixed topic matcher should match strictly, but only prefix
    {
        filter: [][]string{[]string{"a"}, []string{"b"}},
        topics: []string{"a"},
        match:  false,
    },
    {
        filter: [][]string{[]string{"a"}, []string{"b"}},
        topics: []string{"a", "b"},
        match:  true,
    },
    {
        filter: [][]string{[]string{"a"}, []string{"b"}},
        topics: []string{"a", "b", "c"},
        match:  true,
    },
    // Multi-matcher should match any from a sub-group
    {
        filter: [][]string{[]string{"a1", "a2"}},
        topics: []string{"a"},
        match:  false,
    },
    {
        filter: [][]string{[]string{"a1", "a2"}},
        topics: []string{"a1"},
        match:  true,
    },
    {
        filter: [][]string{[]string{"a1", "a2"}},
        topics: []string{"a2"},
        match:  true,
    },
    // Wild-card condition should match anything
    {
        filter: [][]string{[]string{}, []string{"b"}},
        topics: []string{"a"},
        match:  false,
    },
    {
        filter: [][]string{[]string{}, []string{"b"}},
        topics: []string{"a", "b"},
        match:  true,
    },
    {
        filter: [][]string{[]string{}, []string{"b"}},
        topics: []string{"b", "b"},
        match:  true,
    },
}

func TestTopicMatcher(t *testing.T) {
    for i, tt := range topicMatcherTests {
        topics := NewTopicsFromStrings(tt.topics...)

        matcher := newTopicMatcherFromStrings(tt.filter...)
        if match := matcher.Matches(topics); match != tt.match {
            t.Errorf("test %d: match mismatch: have %v, want %v", i, match, tt.match)
        }
    }
}