1 files changed, 9 insertions, 3 deletions

diff --git a/src/main/java/com/stileeducation/markr/service/TestService.java b/src/main/java/com/stileeducation/markr/service/TestService.java
index 0e109ac..f42da1a 100644
--- a/src/main/java/com/stileeducation/markr/service/TestService.java
+++ b/src/main/java/com/stileeducation/markr/service/TestService.java
@@ -2,6 +2,7 @@ package com.stileeducation.markr.service;
 
 import com.stileeducation.markr.entity.Test;
 import com.stileeducation.markr.repository.TestRepository;
+import jakarta.transaction.Transactional;
 import org.springframework.stereotype.Service;
 
 import java.util.Optional;
@@ -16,22 +17,27 @@ public class TestService {
     this.testRepository = testRepository;
   }
 
+  @Transactional
   public Optional<Test> findTest(String testId) {
     return testRepository.findByTestId(testId);
   }
 
+  @Transactional
   public Test findOrCreateTest(String testId, Integer marksAvailable) {
     Optional<Test> optionalTest = testRepository.findByTestId(testId);
+
     if (optionalTest.isPresent()) {
       Test test = optionalTest.get();
-      if (test.getMarksAvailable() < marksAvailable) {
+      if (marksAvailable > test.getMarksAvailable()) {
         test.setMarksAvailable(marksAvailable);
+        test.setCreated(false);
         test.setUpdated(true);
-        testRepository.save(test);
+        return testRepository.save(test);
       } else {
+        test.setCreated(false);
         test.setUpdated(false);
+        return test;
       }
-      return test;
     } else {
       Test test = new Test();
       test.setTestId(testId);