cursor.one test

Author: CMogilko

cursor_test.go

@@ -0,0 +1,34 @@
+package rethinkdb
+
+import (
+	test "gopkg.in/check.v1"
+	"gopkg.in/rethinkdb/rethinkdb-go.v5/internal/integration/tests"
+)
+
+type CursorSuite struct{}
+
+var _ = test.Suite(&CursorSuite{})
+
+func (s *CursorSuite) TestCursor_One_Ok(c *test.C) {
+	data := map[string]interface{}{
+		"A": 1,
+		"B": true,
+	}
+
+	mock := NewMock()
+	ch := make(chan []interface{})
+	mock.On(DB("test").Table("test")).Return(ch, nil)
+	go func() {
+		ch <- []interface{}{data}
+		close(ch)
+	}()
+	res, err := DB("test").Table("test").Run(mock)
+	c.Assert(err, test.IsNil)
+
+	var response interface{}
+	err = res.One(&response)
+
+	c.Assert(err, test.IsNil)
+	c.Assert(response, tests.JsonEquals, data)
+	mock.AssertExpectations(c)
+}

mock.go

@@ -372,6 +372,15 @@ func (m *Mock) Query(ctx context.Context, q Query) (*Cursor, error) {
 	} else if conn != nil {
 		conn.cursors[query.Query.Token] = c
 		c.finished = false
+		c.releaseConn = func() error {
+			return conn.Close()
+		}
+		c.mu.Lock()
+		err := c.fetchMore()
+		c.mu.Unlock()
+		if err != nil {
+			return nil, err
+		}
 	} else {
 		c.buffer = append(c.buffer, query.Response)
 	}
@@ -499,7 +508,7 @@ func (c *mockConn) Write(b []byte) (n int, err error) {
 	c.tokens <- token
 	return len(b), nil
 }
-func (c *mockConn) Close() error                       { panic("not implemented") }
+func (c *mockConn) Close() error                       { return nil }
 func (c *mockConn) LocalAddr() net.Addr                { panic("not implemented") }
 func (c *mockConn) RemoteAddr() net.Addr               { panic("not implemented") }
 func (c *mockConn) SetDeadline(t time.Time) error      { panic("not implemented") }