github.com/cayleygraph/cayley@v0.7.7/internal/gephi/stream_test.go (about)

     1  package gephi
     2  
     3  import (
     4  	"bytes"
     5  	"testing"
     6  
     7  	"github.com/cayleygraph/quad"
     8  	"github.com/stretchr/testify/require"
     9  )
    10  
    11  func TestStreamEncoder(t *testing.T) {
    12  	buf := bytes.NewBuffer(nil)
    13  	gs := NewGraphStream(buf)
    14  	p := map[quad.Value]quad.Value{iriPosX: quad.Float(0), iriPosY: quad.Float(0)}
    15  	gs.AddNode(quad.String("aaa"), p)
    16  	gs.AddNode(quad.String("bbb"), p)
    17  	gs.Flush()
    18  	const expect = "{\"an\":{\"0\":{\"label\":\"aaa\",\"size\":20,\"x\":0,\"y\":0}}}\r\n{\"an\":{\"1\":{\"label\":\"bbb\",\"size\":20,\"x\":0,\"y\":0}}}\r\n"
    19  	require.Equal(t, expect, buf.String())
    20  }