118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package lnwire
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
// TestFeaturesRemoteRequireError checks that we throw an error if remote peer
|
|
// has required feature which we don't support.
|
|
func TestFeaturesRemoteRequireError(t *testing.T) {
|
|
const (
|
|
first = "first"
|
|
second = "second"
|
|
)
|
|
|
|
localFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
})
|
|
|
|
remoteFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
{second, RequiredFlag},
|
|
})
|
|
|
|
if _, err := localFeatures.Compare(remoteFeatures); err == nil {
|
|
t.Fatal("error wasn't received")
|
|
}
|
|
}
|
|
|
|
// TestFeaturesLocalRequireError checks that we throw an error if local peer has
|
|
// required feature which remote peer don't support.
|
|
func TestFeaturesLocalRequireError(t *testing.T) {
|
|
const (
|
|
first = "first"
|
|
second = "second"
|
|
)
|
|
|
|
localFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
{second, RequiredFlag},
|
|
})
|
|
|
|
remoteFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
})
|
|
|
|
if _, err := localFeatures.Compare(remoteFeatures); err == nil {
|
|
t.Fatal("error wasn't received")
|
|
}
|
|
}
|
|
|
|
// TestOptionalFeature checks that if remote peer don't have the feature but
|
|
// on our side this feature is optional than we mark this feature as disabled.
|
|
func TestOptionalFeature(t *testing.T) {
|
|
const first = "first"
|
|
|
|
localFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
})
|
|
|
|
remoteFeatures := NewFeatureVector([]Feature{})
|
|
|
|
shared, err := localFeatures.Compare(remoteFeatures)
|
|
if err != nil {
|
|
t.Fatalf("error while feature vector compare: %v", err)
|
|
}
|
|
|
|
if shared.IsActive(first) {
|
|
t.Fatal("locally feature was set but remote peer notified us" +
|
|
" that it don't have it")
|
|
}
|
|
}
|
|
|
|
// TestSetRequireAfterInit checks that we can change the feature flag after
|
|
// initialization.
|
|
func TestSetRequireAfterInit(t *testing.T) {
|
|
const first = "first"
|
|
|
|
localFeatures := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
})
|
|
localFeatures.SetFeatureFlag(first, RequiredFlag)
|
|
remoteFeatures := NewFeatureVector([]Feature{})
|
|
|
|
_, err := localFeatures.Compare(remoteFeatures)
|
|
if err == nil {
|
|
t.Fatalf("feature was set as required but error wasn't "+
|
|
"returned: %v", err)
|
|
}
|
|
}
|
|
|
|
// TestDecodeEncodeFeaturesVector checks that feature vector might be
|
|
// successfully encoded and decoded.
|
|
func TestDecodeEncodeFeaturesVector(t *testing.T) {
|
|
const first = "first"
|
|
|
|
f := NewFeatureVector([]Feature{
|
|
{first, OptionalFlag},
|
|
})
|
|
|
|
var b bytes.Buffer
|
|
if err := f.Encode(&b); err != nil {
|
|
t.Fatalf("error while encoding feature vector: %v", err)
|
|
}
|
|
|
|
nf, err := NewFeatureVectorFromReader(&b)
|
|
if err != nil {
|
|
t.Fatalf("error while decoding feature vector: %v", err)
|
|
}
|
|
|
|
// Assert equality of the two instances.
|
|
if !reflect.DeepEqual(f.flags, nf.flags) {
|
|
t.Fatalf("encode/decode feature vector don't match %v vs "+
|
|
"%v", f.String(), nf.String())
|
|
}
|
|
}
|