orm_test.go 9.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425
  1. package orm
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/jinzhu/gorm"
  6. "github.com/remogatto/prettytest"
  7. )
  8. type testSuite struct {
  9. prettytest.Suite
  10. }
  11. func TestRunner(t *testing.T) {
  12. prettytest.Run(
  13. t,
  14. new(testSuite),
  15. )
  16. }
  17. func (t *testSuite) BeforeAll() {
  18. var (
  19. db *gorm.DB
  20. err error
  21. )
  22. connected := false
  23. for !connected {
  24. time.Sleep(10 * time.Second)
  25. db, err = New("karmen:karmen@/karmen_test?charset=utf8&parseTime=True&loc=Local")
  26. if err != nil {
  27. time.Sleep(5 * time.Second)
  28. continue
  29. }
  30. connected = true
  31. }
  32. Use(db)
  33. AutoMigrate()
  34. }
  35. func (t *testSuite) TestDB() {
  36. t.Not(t.Nil(DB()))
  37. }
  38. func (t *testSuite) TestGetTeachers() {
  39. teachers, err := GetTeachers()
  40. t.Nil(err)
  41. t.Equal(9, len(teachers))
  42. teacher := teachers[0]
  43. classes, err := teacher.GetClasses()
  44. t.Nil(err)
  45. t.Equal(1, len(classes))
  46. t.Equal(1, len(teacher.Classes))
  47. class := classes[0]
  48. t.Equal("1 A", class.Name)
  49. }
  50. // func (t *testSuite) Before() {
  51. // var activities []Activity
  52. // if err := DB().Find(&activities).Error; err != nil {
  53. // panic(err)
  54. // }
  55. // for _, a := range activities {
  56. // DB().Unscoped().Delete(&a)
  57. // }
  58. // if err := DB().Exec("ALTER TABLE activities AUTO_INCREMENT = 1").Error; err != nil {
  59. // panic(err)
  60. // }
  61. // }
  62. // func (t *testSuite) TestCredential() {
  63. // var teacher Teacher
  64. // teacher.Name = "John"
  65. // teacher.Surname = "Doe"
  66. // if err := DB().Create(&teacher).Error; err != nil {
  67. // panic(err)
  68. // }
  69. // if err := DB().Find(&teacher).Error; err != nil {
  70. // panic(err)
  71. // }
  72. // t.Equal(teacher.Username, "john.doe")
  73. // t.Equal(teacher.Email, "john.doe@foo.org")
  74. // t.Equal(teacher.Password, "RKDb/GzQw8jzfAFwiAjclA81oBtzYWx0")
  75. // }
  76. // func (t *testSuite) TestCreateUpdateActivity() {
  77. // var (
  78. // class Class
  79. // subject Subject
  80. // )
  81. // activity := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  82. // // Activity 1
  83. // if err := DB().Create(&activity).Error; err != nil {
  84. // panic(err)
  85. // }
  86. // // Marco Rossi
  87. // teacher := Teacher{}
  88. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  89. // panic(err)
  90. // }
  91. // // 1A
  92. // if err := DB().Find(&class, 1).Error; err != nil {
  93. // panic(err)
  94. // }
  95. // // Italiano
  96. // if err := DB().Find(&subject, 1).Error; err != nil {
  97. // panic(err)
  98. // }
  99. // t.Equal(5, teacher.CurrHours)
  100. // t.Equal("1A", teacher.Classes[0].Name)
  101. // t.Equal("Italiano", teacher.Subjects[0].Name)
  102. // // Add another activity and check for teacher's hours, subjects and classes
  103. // // Subject: "Latino"
  104. // // Class: "1B"
  105. // activity = Activity{Hours: 3, TeacherID: 1, ClassID: 2, SubjectID: 2}
  106. // // Activity 2
  107. // if err := DB().Create(&activity).Error; err != nil {
  108. // panic(err)
  109. // }
  110. // // Marco Rossi
  111. // teacher = Teacher{}
  112. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  113. // panic(err)
  114. // }
  115. // t.Equal(8, teacher.CurrHours)
  116. // t.Equal("1A", teacher.Classes[0].Name)
  117. // t.Equal("1B", teacher.Classes[1].Name)
  118. // t.Equal("Italiano", teacher.Subjects[0].Name)
  119. // t.Equal("Latino", teacher.Subjects[1].Name)
  120. // // Add an activity without teacher
  121. // // Subject: "Latino"
  122. // // Class: "2A"
  123. // activity = Activity{Hours: 3, TeacherID: 0, ClassID: 3, SubjectID: 2}
  124. // // Activity 3
  125. // if err := DB().Create(&activity).Error; err != nil {
  126. // panic(err)
  127. // }
  128. // // Test update activty
  129. // activity = Activity{}
  130. // if err := DB().Find(&activity, 1).Error; err != nil {
  131. // panic(err)
  132. // }
  133. // activity.Hours = 4
  134. // if err := DB().Save(&activity).Error; err != nil {
  135. // panic(err)
  136. // }
  137. // // Marco Rossi
  138. // teacher = Teacher{}
  139. // if err := DB().Find(&teacher, 1).Error; err != nil {
  140. // panic(err)
  141. // }
  142. // t.Equal(7, teacher.CurrHours)
  143. // // Delete an activity and check for the right number of hours and associations
  144. // if err := DB().Delete(&activity).Error; err != nil {
  145. // panic(err)
  146. // }
  147. // // Marco Rossi
  148. // teacher = Teacher{}
  149. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  150. // panic(err)
  151. // }
  152. // t.Equal(3, teacher.CurrHours)
  153. // t.Equal(1, len(teacher.Classes))
  154. // t.Equal(1, len(teacher.Subjects))
  155. // t.Equal("1B", teacher.Classes[0].Name)
  156. // t.Equal("Latino", teacher.Subjects[0].Name)
  157. // // Remove teacher from activity and check for consistency
  158. // activity = Activity{}
  159. // teacher = Teacher{}
  160. // if err := DB().Find(&activity, 2).Error; err != nil {
  161. // panic(err)
  162. // }
  163. // activity.TeacherID = 0
  164. // if err := DB().Save(&activity).Error; err != nil {
  165. // panic(err)
  166. // }
  167. // // Marco Rossi
  168. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  169. // panic(err)
  170. // }
  171. // t.Equal(0, teacher.CurrHours)
  172. // t.Equal(0, len(teacher.Classes))
  173. // t.Equal(0, len(teacher.Subjects))
  174. // // Change teacher in activity and check for consistency
  175. // // Subject: "Latino"
  176. // // Class: "1B"
  177. // activity = Activity{Hours: 3, TeacherID: 1, ClassID: 2, SubjectID: 2}
  178. // // Activity 3
  179. // if err := DB().Create(&activity).Error; err != nil {
  180. // panic(err)
  181. // }
  182. // activity.TeacherID = 2
  183. // if err := DB().Save(&activity).Error; err != nil {
  184. // panic(err)
  185. // }
  186. // // Marco Rossi
  187. // teacher = Teacher{}
  188. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  189. // panic(err)
  190. // }
  191. // t.Equal(0, teacher.CurrHours)
  192. // t.Equal(0, len(teacher.Classes))
  193. // t.Equal(0, len(teacher.Subjects))
  194. // }
  195. // func (t *testSuite) TestClassTeachers() {
  196. // var class Class
  197. // activity_1 := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  198. // if err := DB().Create(&activity_1).Error; err != nil {
  199. // panic(err)
  200. // }
  201. // activity_2 := Activity{Hours: 3, TeacherID: 1, ClassID: 2, SubjectID: 2}
  202. // if err := DB().Create(&activity_2).Error; err != nil {
  203. // panic(err)
  204. // }
  205. // if err := DB().Preload("Teachers").First(&class, 1).Error; err != nil {
  206. // panic(err)
  207. // }
  208. // subjects := class.Teachers[0].GetSubjectsByClassID(class.ID)
  209. // t.Equal(1, len(subjects))
  210. // t.Equal("Italiano", subjects[0].Name)
  211. // }
  212. // func (t *testSuite) TestCoordinatorMinuter() {
  213. // var (
  214. // class Class
  215. // teacher_1 Teacher
  216. // teacher_2 Teacher
  217. // )
  218. // activity_1 := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  219. // if err := DB().Create(&activity_1).Error; err != nil {
  220. // panic(err)
  221. // }
  222. // activity_2 := Activity{Hours: 5, TeacherID: 2, ClassID: 1, SubjectID: 1}
  223. // if err := DB().Create(&activity_2).Error; err != nil {
  224. // panic(err)
  225. // }
  226. // if err := DB().First(&class, 1).Error; err != nil {
  227. // panic(err)
  228. // }
  229. // class.CoordinatorID = 1
  230. // class.MinuterID = 2
  231. // if err := DB().Save(&class).Error; err != nil {
  232. // panic(err)
  233. // }
  234. // if err := DB().Find(&teacher_1, 1).Error; err != nil {
  235. // panic(err)
  236. // }
  237. // if err := DB().Find(&teacher_2, 2).Error; err != nil {
  238. // panic(err)
  239. // }
  240. // classes_1 := teacher_1.IsCoordinator()
  241. // t.Equal(1, len(classes_1))
  242. // if len(classes_1) > 0 {
  243. // t.Equal("1A", classes_1[0].Name)
  244. // }
  245. // classes_2 := teacher_2.IsMinuter()
  246. // t.Equal(1, len(classes_2))
  247. // if len(classes_2) > 0 {
  248. // t.Equal("1A", classes_2[0].Name)
  249. // }
  250. // }
  251. // func (t *testSuite) TestTeacherHasClass() {
  252. // activity := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  253. // // Activity 1
  254. // if err := DB().Create(&activity).Error; err != nil {
  255. // panic(err)
  256. // }
  257. // // Marco Rossi
  258. // teacher := Teacher{}
  259. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  260. // panic(err)
  261. // }
  262. // t.True(teacher.HasClass("1A"))
  263. // }
  264. // func (t *testSuite) TestTeacherHasSubject() {
  265. // activity := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  266. // // Activity 1
  267. // if err := DB().Create(&activity).Error; err != nil {
  268. // panic(err)
  269. // }
  270. // // Marco Rossi
  271. // teacher := Teacher{}
  272. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  273. // panic(err)
  274. // }
  275. // t.True(teacher.HasSubject("Italiano"))
  276. // }
  277. // func (t *testSuite) TestTeacherHasSubjectInClass() {
  278. // activity := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  279. // // Activity 1
  280. // if err := DB().Create(&activity).Error; err != nil {
  281. // panic(err)
  282. // }
  283. // // Marco Rossi
  284. // teacher := Teacher{}
  285. // if err := DB().Preload("Classes").Preload("Subjects").Find(&teacher, 1).Error; err != nil {
  286. // panic(err)
  287. // }
  288. // t.True(teacher.HasSubjectInClass("Italiano", "1A"))
  289. // }
  290. // func (t *testSuite) TestTeacherIsCoordinatorInClass() {
  291. // activity := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  292. // // Activity 1
  293. // if err := DB().Create(&activity).Error; err != nil {
  294. // panic(err)
  295. // }
  296. // var class Class
  297. // if err := DB().First(&class, 1).Error; err != nil {
  298. // panic(err)
  299. // }
  300. // class.CoordinatorID = 1
  301. // // Marco Rossi
  302. // teacher := Teacher{}
  303. // if err := DB().Preload("Classes").Find(&teacher, 1).Error; err != nil {
  304. // panic(err)
  305. // }
  306. // t.True(teacher.IsCoordinatorInClass("1A"))
  307. // }
  308. // func (t *testSuite) TestTeacherIssues() {
  309. // // activity_1 := Activity{Hours: 5, TeacherID: 1, ClassID: 1, SubjectID: 1}
  310. // // activity_2 := Activity{Hours: 10, TeacherID: 1, ClassID: 2, SubjectID: 1}
  311. // // if err := DB().Create(&activity_1).Error; err != nil {
  312. // // panic(err)
  313. // // }
  314. // // if err := DB().Create(&activity_2).Error; err != nil {
  315. // // panic(err)
  316. // // }
  317. // // // Marco Rossi
  318. // // teacher := Teacher{}
  319. // // if err := DB().Preload("Issues").Find(&teacher, 1).Error; err != nil {
  320. // // panic(err)
  321. // // }
  322. // // t.Equal(1, len(teacher.Issues))
  323. // // t.Equal(IssueTeacherHours, teacher.Issues[0].Type)
  324. // }