mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-05 12:56:38 +00:00
35 lines
699 B
Go
35 lines
699 B
Go
|
package geosite_test
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/sagernet/sing-box/common/geosite"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
)
|
||
|
|
||
|
func TestGeosite(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
|
||
|
var buffer bytes.Buffer
|
||
|
err := geosite.Write(&buffer, map[string][]geosite.Item{
|
||
|
"test": {
|
||
|
{
|
||
|
Type: geosite.RuleTypeDomain,
|
||
|
Value: "example.org",
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
require.NoError(t, err)
|
||
|
reader, codes, err := geosite.NewReader(bytes.NewReader(buffer.Bytes()))
|
||
|
require.NoError(t, err)
|
||
|
require.Equal(t, []string{"test"}, codes)
|
||
|
items, err := reader.Read("test")
|
||
|
require.NoError(t, err)
|
||
|
require.Equal(t, []geosite.Item{{
|
||
|
Type: geosite.RuleTypeDomain,
|
||
|
Value: "example.org",
|
||
|
}}, items)
|
||
|
}
|