Kaynağa Gözat

Work in progress

Andrea Fazzi 6 yıl önce
ebeveyn
işleme
bfef0ab7a4
2 değiştirilmiş dosya ile 12 ekleme ve 2 silme
  1. 5 2
      orm/orm_test.go
  2. 7 0
      orm/teacher.go

+ 5 - 2
orm/orm_test.go

@@ -151,7 +151,7 @@ func (t *testSuite) TestSaveTeacher() {
 	}
 }
 
-func (t *testSuite) TestCreateTeacher() {
+func (t *testSuite) TestCreateDeleteTeacher() {
 	teacher := new(Teacher)
 	teacher.Name = "Giovanni"
 	teacher.Surname = "Giacobbe"
@@ -162,7 +162,10 @@ func (t *testSuite) TestCreateTeacher() {
 	if !t.Failed() {
 		teacher, _ = GetTeacher("10")
 		t.Equal("Giovanni", teacher.Name)
-
+		err := DeleteTeacher(teacher)
+		t.Nil(err)
+		teachers, _ := GetTeachers()
+		t.Equal(9, len(teachers))
 	}
 }
 

+ 7 - 0
orm/teacher.go

@@ -101,6 +101,13 @@ func CreateTeacher(teacher *Teacher) (*Teacher, error) {
 	return teacher, nil
 }
 
+func DeleteTeacher(teacher *Teacher) error {
+	if err := DB().Delete(teacher).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
 func (t *Teacher) GetClasses() ([]*Class, error) {
 	if err := DB().Raw(selectUniqueTeacherClasses, t.ID).Scan(&t.Classes).Error; err != nil {
 		return nil, err