func TestGet(t *testing.T) {
db, mock, err := sqlmock.New()
assert.Nil(t, err)
rows := sqlmock.NewRows([]string{"hello", "High"})
mock.ExpectQuery("select taskId,task,priority from tasks;").WillReturnRows(rows)
contextObject := config.ContextObject{}
contextObject.ErrorLogFile, err = os.OpenFile(errorLogFilePath, os.O_APPEND|os.O_WRONLY, 0600)
contextObject.Db = db
Get(contextObject)
if err := mock.ExpectationsWereMet(); err != nil {
t.Errorf("there were unfulfilled expections: %s", err)
}
}
func TestDelete(t *testing.T) {
db, mock, err := sqlmock.New()
assert.Nil(t, err)
taskId := 1
mock.ExpectExec("delete from tasks").
WithArgs(taskId).
WillReturnResult(sqlmock.NewResult(1, 1))
contextObject := config.ContextObject{}
contextObject.ErrorLogFile, err = os.OpenFile(errorLogFilePath, os.O_APPEND|os.O_WRONLY, 0600)
contextObject.Db = db
Delete(contextObject, taskId)
if err := mock.ExpectationsWereMet(); err != nil {
t.Errorf("there were unfulfilled expections: %s", err)
}
}
func TestAdd(t *testing.T) {
db, mock, err := sqlmock.New()
assert.Nil(t, err)
task := "dring water"
priority := "High"
mock.ExpectExec("insert into tasks").
WithArgs(task, priority).
WillReturnResult(sqlmock.NewResult(1, 1))
contextObject := config.ContextObject{}
contextObject.ErrorLogFile, err = os.OpenFile(errorLogFilePath, os.O_APPEND|os.O_WRONLY, 0600)
contextObject.Db = db
Add(contextObject, task, priority)
if err := mock.ExpectationsWereMet(); err != nil {
t.Errorf("there were unfulfilled expections: %s", err)
}
}
