Merge pull request #646 from upper/add-test-for-canceled-context

Add test for canceled context and `Save`

internal/testsuite/record_suite.go

 package testsuite
 
 import (
+	"context"
 	"database/sql"
 	"fmt"
 	"time"
@@ -401,3 +402,27 @@ func (s *RecordTestSuite) TestUnknownCollection() {
 	_, err = sess.Collection("users").Insert(&User{Username: "Foo"})
 	s.NoError(err)
 }
+
+func (s *RecordTestSuite) TestContextCanceled() {
+	var err error
+
+	sess := s.Session()
+
+	err = sess.Collection("users").Truncate()
+	s.NoError(err)
+
+	{
+		ctx, cancelFn := context.WithTimeout(context.Background(), time.Minute)
+		canceledSess := sess.WithContext(ctx)
+
+		cancelFn()
+
+		user := User{Username: "foo"}
+		err = canceledSess.Save(&user)
+		s.Error(err)
+
+		c, err := sess.Collection("users").Count()
+		s.NoError(err)
+		s.Equal(uint64(0), c)
+	}
+}