gopkg.in/rethinkdb/rethinkdb-go.v6@v6.2.2/cursor_test.go (about)

     1  package rethinkdb
     2  
     3  import (
     4  	test "gopkg.in/check.v1"
     5  	"gopkg.in/rethinkdb/rethinkdb-go.v6/internal/integration/tests"
     6  )
     7  
     8  type CursorSuite struct{}
     9  
    10  var _ = test.Suite(&CursorSuite{})
    11  
    12  func (s *CursorSuite) TestCursor_One_Ok(c *test.C) {
    13  	data := map[string]interface{}{
    14  		"A": 1,
    15  		"B": true,
    16  	}
    17  
    18  	mock := NewMock()
    19  	ch := make(chan []interface{})
    20  	mock.On(DB("test").Table("test")).Return(ch, nil)
    21  	go func() {
    22  		ch <- []interface{}{data}
    23  		close(ch)
    24  	}()
    25  	res, err := DB("test").Table("test").Run(mock)
    26  	c.Assert(err, test.IsNil)
    27  
    28  	var response interface{}
    29  	err = res.One(&response)
    30  
    31  	c.Assert(err, test.IsNil)
    32  	c.Assert(response, tests.JsonEquals, data)
    33  	mock.AssertExpectations(c)
    34  }