test: Add tests for courses repo

internal/repository/courses_test.go

@@ -1,6 +1,7 @@
 package repository_test
 
 import (
+	"courses-microservice/config"
 	"courses-microservice/internal/errors"
 	"courses-microservice/internal/models"
 	"courses-microservice/internal/repository"
@@ -11,6 +12,7 @@ import (
 	"github.com/jackc/pgx/v5"
 	pgxmock "github.com/pashagolub/pgxmock/v2"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestAddCourse(t *testing.T) {
@@ -387,6 +389,291 @@ func TestGetEnrollmentsCount(t *testing.T) {
 	assert.NoError(t, mock.ExpectationsWereMet())
 }
 
+func TestAddAuxProfessor_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID := uuid.New()
+
+	mock.ExpectExec("INSERT INTO aux_professors").
+		WithArgs(courseID, userID).
+		WillReturnResult(pgxmock.NewResult("INSERT", 1))
+
+	err := repo.AddAuxProfessor(courseID, userID)
+	require.NoError(t, err)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestRemoveAuxProfessor_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID := uuid.New()
+
+	mock.ExpectExec("DELETE FROM aux_professors").
+		WithArgs(courseID, userID).
+		WillReturnResult(pgxmock.NewResult("DELETE", 1))
+
+	err := repo.RemoveAuxProfessor(courseID, userID)
+	require.NoError(t, err)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetAuxProfessorLogs_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID := uuid.New()
+	page := 1
+	now := time.Now()
+
+	rows := pgxmock.NewRows([]string{
+		"course_id", "user_id", "action", "created_at",
+	}).AddRow(courseID, userID, "added", now)
+
+	mock.ExpectQuery("SELECT course_id, user_id, action, created_at FROM aux_professor_logs").
+		WithArgs(courseID, config.CoursesPerPage, config.CoursesPerPage*(page-1)).
+		WillReturnRows(rows)
+
+	logs, err := repo.GetAuxProfessorLogs(courseID, page)
+	require.NoError(t, err)
+	require.Len(t, logs, 1)
+	assert.Equal(t, "added", logs[0].Action)
+	assert.Equal(t, courseID, logs[0].CourseID)
+	assert.Equal(t, userID, logs[0].ProfessorID)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestLogAuxProfessorAction_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID := uuid.New()
+	action := "added"
+
+	mock.ExpectExec(`INSERT INTO aux_professor_logs`).
+		WithArgs(courseID, userID, action).
+		WillReturnResult(pgxmock.NewResult("INSERT", 1))
+
+	err := repo.LogAuxProfessorAction(courseID, userID, action)
+	require.NoError(t, err)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestIsAuxProfessor_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID := uuid.New()
+
+	mock.ExpectQuery(`SELECT EXISTS \(.*FROM aux_professors`).
+		WithArgs(courseID, userID).
+		WillReturnRows(pgxmock.NewRows([]string{"exists"}).AddRow(true))
+
+	isAux := repo.IsAuxProfessor(courseID, userID)
+
+	assert.True(t, isAux)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetAuxProfessorsByCourse_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	courseID := uuid.New()
+	userID1 := uuid.New()
+	userID2 := uuid.New()
+
+	rows := pgxmock.NewRows([]string{"user_id"}).
+		AddRow(userID1).
+		AddRow(userID2)
+
+	mock.ExpectQuery(`SELECT user_id FROM aux_professors WHERE course_id = \$1`).
+		WithArgs(courseID).
+		WillReturnRows(rows)
+
+	result, err := repo.GetAuxProfessorsByCourse(courseID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, []uuid.UUID{userID1, userID2}, result)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetStudentInfo_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	studentID := uuid.New()
+	expectedCredits := 120
+
+	rows := pgxmock.NewRows([]string{"credits"}).
+		AddRow(expectedCredits)
+
+	mock.ExpectQuery(`SELECT credits FROM students_info WHERE student_id = \$1`).
+		WithArgs(studentID).
+		WillReturnRows(rows)
+
+	result, err := repo.GetStudentInfo(studentID)
+
+	assert.NoError(t, err)
+	assert.NotNil(t, result)
+	assert.Equal(t, studentID, result.StudentId)
+	assert.Equal(t, expectedCredits, result.Credits)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestUpdateStudentInfo_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+	studentID := uuid.New()
+	info := &models.StudentInfo{
+		Credits: 150,
+	}
+
+	mock.ExpectExec(`INSERT INTO students_info \(student_id, credits\)`).
+		WithArgs(studentID, info.Credits).
+		WillReturnResult(pgxmock.NewResult("INSERT", 1))
+
+	err := repo.UpdateStudentInfo(studentID, info)
+
+	assert.NoError(t, err)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestSetStudentFinalGrade_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+	studentID := uuid.New()
+	finalGrade := 87.5
+
+	mock.ExpectExec(`UPDATE enrollments SET final_grade = \$1`).
+		WithArgs(finalGrade, courseID, studentID).
+		WillReturnResult(pgxmock.NewResult("UPDATE", 1))
+
+	err := repo.SetStudentFinalGrade(courseID, studentID, finalGrade)
+
+	assert.NoError(t, err)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetStudentFinalGrade_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+	studentID := uuid.New()
+	expectedGrade := 92.3
+
+	rows := pgxmock.NewRows([]string{"final_grade"}).AddRow(expectedGrade)
+	mock.ExpectQuery(`SELECT final_grade FROM enrollments WHERE course_id = \$1 AND student_id = \$2`).
+		WithArgs(courseID, studentID).
+		WillReturnRows(rows)
+
+	finalGrade, err := repo.GetStudentFinalGrade(courseID, studentID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, expectedGrade, finalGrade)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestIsCourseProfessor_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+	userID := uuid.New()
+
+	// Simulate that the user is professor (exists = true)
+	rows := pgxmock.NewRows([]string{"exists"}).AddRow(true)
+	mock.ExpectQuery(`SELECT EXISTS \(.+\) OR EXISTS \(.+\)`).
+		WithArgs(courseID, userID).
+		WillReturnRows(rows)
+
+	isProfessor, err := repo.IsCourseProfessor(courseID, userID)
+
+	assert.NoError(t, err)
+	assert.True(t, isProfessor)
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetExamsStatsByCourse_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+
+	// Prepare mocked row with totalExams, completedExams, averageScore
+	rows := pgxmock.NewRows([]string{"total_exams", "completed_exams", "avg"}).
+		AddRow(5, 3, 87.5)
+
+	mock.ExpectQuery(`SELECT DISTINCT COUNT\(e\.id\) AS total_exams,.*`).
+		WithArgs(courseID).
+		WillReturnRows(rows)
+
+	total, completed, avgScore, err := repo.GetExamsStatsByCourse(courseID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 5, total)
+	assert.Equal(t, 3, completed)
+	assert.Equal(t, 87.5, avgScore)
+
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetAssignmentsStatsByCourse_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+
+	rows := pgxmock.NewRows([]string{"total_assignments", "completed_assignments", "avg"}).
+		AddRow(8, 5, 92.3)
+
+	mock.ExpectQuery(`SELECT COUNT\(a\.id\) AS total_assignments,.*`).
+		WithArgs(courseID).
+		WillReturnRows(rows)
+
+	total, completed, avgScore, err := repo.GetAssignmentsStatsByCourse(courseID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 8, total)
+	assert.Equal(t, 5, completed)
+	assert.InDelta(t, 92.3, avgScore, 0.0001)
+
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetExamsStatsByStudent_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+	studentID := uuid.New()
+
+	rows := pgxmock.NewRows([]string{"count", "avg"}).
+		AddRow(7, 87.5)
+
+	mock.ExpectQuery(`SELECT COUNT\(es\.id\),\s+COALESCE\(AVG\(es\.score\), 0\)`).
+		WithArgs(courseID, studentID).
+		WillReturnRows(rows)
+
+	total, avgScore, err := repo.GetExamsStatsByStudent(courseID, studentID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 7, total)
+	assert.InDelta(t, 87.5, avgScore, 0.0001)
+
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
+func TestGetAssignmentsStatsByStudent_Success(t *testing.T) {
+	mock, repo := setupMockRepoForCourses(t)
+
+	courseID := uuid.New()
+	studentID := uuid.New()
+
+	rows := pgxmock.NewRows([]string{"count", "avg"}).
+		AddRow(5, 92.3)
+
+	mock.ExpectQuery(`SELECT COUNT\(asub\.id\),\s+COALESCE\(AVG\(asub\.score\), 0\)`).
+		WithArgs(courseID, studentID).
+		WillReturnRows(rows)
+
+	total, avgScore, err := repo.GetAssignmentsStatsByStudent(courseID, studentID)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 5, total)
+	assert.InDelta(t, 92.3, avgScore, 0.0001)
+
+	assert.NoError(t, mock.ExpectationsWereMet())
+}
+
 /* HELPER FUNCTIONS */
 
 func setupMockRepoForCourses(t *testing.T) (pgxmock.PgxConnIface, repository.CoursesRepository) {