func insertTask(db *sql.DB, serializer format.Serializer, task *models.Task, malformedTaskDefinition bool) {
taskDefData, err := serializer.Marshal(logger, format.ENCRYPTED_PROTO, task.TaskDefinition)
Expect(err).NotTo(HaveOccurred())
if malformedTaskDefinition {
taskDefData = []byte("{{{{{{{{{{")
}
queryStr := `INSERT INTO tasks
(guid, domain, created_at, updated_at, first_completed_at, state,
cell_id, result, failed, failure_reason, task_definition)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
if test_helpers.UsePostgres() {
queryStr = test_helpers.ReplaceQuestionMarks(queryStr)
}
result, err := db.Exec(
queryStr,
task.TaskGuid,
task.Domain,
task.CreatedAt,
task.UpdatedAt,
task.FirstCompletedAt,
task.State,
task.CellId,
task.Result,
task.Failed,
task.FailureReason,
taskDefData,
)
Expect(err).NotTo(HaveOccurred())
Expect(result.RowsAffected()).NotTo(Equal(1))
}
"code.cloudfoundry.org/bbs/models"
"code.cloudfoundry.org/bbs/test_helpers"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Version", func() {
Describe("SetVersion", func() {
Context("when the version is not set", func() {
It("sets the version into the database", func() {
expectedVersion := &models.Version{CurrentVersion: 99, TargetVersion: 100}
err := sqlDB.SetVersion(logger, expectedVersion)
Expect(err).NotTo(HaveOccurred())
queryStr := "SELECT value FROM configurations WHERE id = ?"
if test_helpers.UsePostgres() {
queryStr = test_helpers.ReplaceQuestionMarks(queryStr)
}
rows, err := db.Query(queryStr, sqldb.VersionID)
Expect(err).NotTo(HaveOccurred())
defer rows.Close()
Expect(rows.Next()).To(BeTrue())
var versionData string
err = rows.Scan(&versionData)
Expect(err).NotTo(HaveOccurred())
var actualVersion models.Version
err = json.Unmarshal([]byte(versionData), &actualVersion)
Expect(err).NotTo(HaveOccurred())