status-im/status-go

package node

import (
    "testing"

    "github.com/stretchr/testify/require"

    "github.com/status-im/status-go/params"
)

func TestWakuLightModeEnabledSetsEmptyBloomFilter(t *testing.T) {
    statusNode, err := createAndStartStatusNode(&params.NodeConfig{
        WakuConfig: params.WakuConfig{
            Enabled:     true,
            LightClient: true,
        },
    })
    require.NoError(t, err)
    defer func() {
        err := statusNode.Stop()
        require.NoError(t, err)
    }()

    waku := statusNode.WakuService()
    require.NotNil(t, waku)

    bloomFilter := waku.BloomFilter()
    expectedEmptyBloomFilter := make([]byte, 64)
    require.NotNil(t, bloomFilter)
    require.Equal(t, expectedEmptyBloomFilter, bloomFilter)
}

func TestWakuLightModeEnabledSetsNilBloomFilter(t *testing.T) {
    statusNode, err := createAndStartStatusNode(&params.NodeConfig{
        WakuConfig: params.WakuConfig{
            Enabled:     true,
            LightClient: false,
        },
    })
    require.NoError(t, err)
    defer func() {
        err := statusNode.Stop()
        require.NoError(t, err)
    }()

    waku := statusNode.WakuService()
    require.NotNil(t, waku)
    require.Nil(t, waku.BloomFilter())
}