Просмотр исходного кода

Omit Tutor when saving student

Andrea Fazzi 6 лет назад
Родитель
Сommit
35651e114b
2 измененных файлов с 8 добавлено и 2 удалено
  1. 3 1
      orm/orm_test.go
  2. 5 1
      orm/student.go

+ 3 - 1
orm/orm_test.go

@@ -289,11 +289,13 @@ func (t *testSuite) TestSaveStudent() {
 	t.Equal(uint(1), student.(*Student).ClassID)
 
 	student.(*Student).ClassID = 2
+	student.(*Student).Handicap = true
 	_, err = SaveStudent(student)
 	t.Nil(err)
 
 	if !t.Failed() {
 		student, _ = GetStudent(map[string]string{"id": "1"})
-		t.Equal(uint(2), student.(*Student).ClassID)
+		t.Equal(uint(1), student.(*Student).ClassID)
+		t.Equal(true, student.(*Student).Handicap)
 	}
 }

+ 5 - 1
orm/student.go

@@ -2,6 +2,7 @@ package orm
 
 import (
 	"fmt"
+	"log"
 	"net/http"
 
 	"github.com/jinzhu/gorm"
@@ -103,7 +104,7 @@ func GetStudentsAll(args map[string]string) (interface{}, error) {
 }
 
 func SaveStudent(student interface{}) (interface{}, error) {
-	if err := DB().Omit("Class").Save(student).Error; err != nil {
+	if err := DB().Omit("Class").Omit("Tutor").Save(student).Error; err != nil {
 		return nil, err
 	}
 	return student, nil
@@ -124,6 +125,9 @@ func UpdateStudent(args map[string]string, r *http.Request) (IDer, error) {
 	if err != nil {
 		return nil, err
 	}
+
+	log.Println(student)
+
 	_, err = SaveStudent(student)
 	if err != nil {
 		return nil, err