Browse Source

Omit Class field when updating student

Andrea Fazzi 6 years ago
parent
commit
85befb28d0
2 changed files with 17 additions and 1 deletions
  1. 16 0
      orm/orm_test.go
  2. 1 1
      orm/student.go

+ 16 - 0
orm/orm_test.go

@@ -252,3 +252,19 @@ func (t *testSuite) TestGetSubjectForAdd() {
 	}
 
 }
+
+func (t *testSuite) TestSaveStudent() {
+	student, err := GetStudentAll(map[string]string{"id": "1"})
+	t.Nil(err)
+
+	t.Equal(uint(1), student.(*Student).ClassID)
+
+	student.(*Student).ClassID = 2
+	_, err = SaveStudent(student)
+	t.Nil(err)
+
+	if !t.Failed() {
+		student, _ = GetStudent(map[string]string{"id": "1"})
+		t.Equal(uint(2), student.(*Student).ClassID)
+	}
+}

+ 1 - 1
orm/student.go

@@ -90,7 +90,7 @@ func GetStudentsAll(args map[string]string) (interface{}, error) {
 }
 
 func SaveStudent(student interface{}) (interface{}, error) {
-	if err := DB().Omit("Classes", "Subjects", "Activities").Save(student).Error; err != nil {
+	if err := DB().Omit("Class").Save(student).Error; err != nil {
 		return nil, err
 	}
 	return student, nil