diff UnitTestsSources/MultiThreadingTests.cpp @ 2557:b4516a6f214b jobs

state machine
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 03 May 2018 13:45:31 +0200
parents 91e944c8389b
children 57f81b988713
line wrap: on
line diff
--- a/UnitTestsSources/MultiThreadingTests.cpp	Thu May 03 10:27:39 2018 +0200
+++ b/UnitTestsSources/MultiThreadingTests.cpp	Thu May 03 13:45:31 2018 +0200
@@ -272,7 +272,10 @@
 #  error The job engine cannot be used in sandboxed environments
 #endif
 
+#include "../Core/Logging.h"
+
 #include <boost/date_time/posix_time/posix_time.hpp>
+#include <queue>
 
 namespace Orthanc
 {
@@ -289,24 +292,24 @@
   enum JobStepStatus
   {
     JobStepStatus_Success,
-    JobStepStatus_Error,
+    JobStepStatus_Failure,
     JobStepStatus_Continue,
     JobStepStatus_Retry
   };
 
 
-  class IJobStepResult : public boost::noncopyable
+  class JobStepResult
   {
   private:
     JobStepStatus status_;
     
   public:
-    explicit IJobStepResult(JobStepStatus status) :
-      status_(status)
+    explicit JobStepResult(JobStepStatus status) :
+    status_(status)
     {
     }
 
-    virtual ~IJobStepResult()
+    virtual ~JobStepResult()
     {
     }
 
@@ -317,15 +320,15 @@
   };
 
 
-  class RetryResult : public IJobStepResult
+  class RetryResult : public JobStepResult
   {
   private:
     unsigned int  timeout_;   // Retry after "timeout_" milliseconds
 
   public:
     RetryResult(unsigned int timeout) :
-      IJobStepResult(JobStepStatus_Retry),
-      timeout_(timeout)
+    JobStepResult(JobStepStatus_Retry),
+    timeout_(timeout)
     {
     }
 
@@ -343,7 +346,7 @@
     {
     }
 
-    virtual IJobStepResult* ExecuteStep() = 0;
+    virtual JobStepResult* ExecuteStep() = 0;
 
     virtual void ReleaseResources() = 0;   // For pausing jobs
 
@@ -353,83 +356,690 @@
   };
 
 
+  class JobHandler : public boost::noncopyable
+  {
+  private:
+    std::string               id_;
+    JobState                  state_;
+    std::auto_ptr<IJob>       job_;
+    int                       priority_;  // "+inf()" means highest priority
+    boost::posix_time::ptime  creationTime_;
+    boost::posix_time::ptime  lastUpdateTime_;
+    boost::posix_time::ptime  retryTime_;
+    uint64_t                  runtime_;  // In milliseconds
+    bool                      pauseScheduled_;
+
+    void SetStateInternal(JobState state) 
+    {
+      const boost::posix_time::ptime now = boost::posix_time::microsec_clock::universal_time();
+
+      if (state_ == JobState_Running)
+      {
+        runtime_ += (now - lastUpdateTime_).total_milliseconds();
+      }
+
+      state_ = state;
+      lastUpdateTime_ = now;
+      pauseScheduled_ = false;
+    }
+
+  public:
+    JobHandler(IJob* job,
+               int priority) :
+      id_(Toolbox::GenerateUuid()),
+      state_(JobState_Pending),
+      job_(job),
+      priority_(priority),
+      creationTime_(boost::posix_time::microsec_clock::universal_time()),
+      lastUpdateTime_(creationTime_),
+      runtime_(0),
+      pauseScheduled_(false)
+    {
+      if (job == NULL)
+      {
+        throw OrthancException(ErrorCode_NullPointer);
+      }
+    }
+
+    const std::string& GetId() const
+    {
+      return id_;
+    }
+
+    IJob& GetJob() const
+    {
+      assert(job_.get() != NULL);
+      return *job_;
+    }
+
+    void SetPriority(int priority)
+    {
+      priority_ = priority;
+    }
+
+    int GetPriority() const
+    {
+      return priority_;
+    }
+
+    JobState GetState() const
+    {
+      return state_;
+    }
+
+    void SetState(JobState state) 
+    {
+      if (state == JobState_Retry)
+      {
+        // Use "SetRetryState()"
+        throw OrthancException(ErrorCode_BadSequenceOfCalls);
+      }
+      else
+      {
+        SetStateInternal(state);
+      }
+    }
+
+    void SetRetryState(unsigned int timeout)
+    {
+      if (state_ == JobState_Running)
+      {
+        SetStateInternal(JobState_Retry);
+        retryTime_ = (boost::posix_time::microsec_clock::universal_time() + 
+                      boost::posix_time::milliseconds(timeout));
+      }
+      else
+      {
+        // Only valid for running jobs
+        throw OrthancException(ErrorCode_BadSequenceOfCalls);
+      }
+    }
+
+    void SchedulePause()
+    {
+      if (state_ == JobState_Running)
+      {
+        pauseScheduled_ = true;
+      }
+      else
+      {
+        // Only valid for running jobs
+        throw OrthancException(ErrorCode_BadSequenceOfCalls);
+      }
+    }
+
+    bool IsPauseScheduled()
+    {
+      return pauseScheduled_;
+    }
+
+    bool IsRetryReady(const boost::posix_time::ptime& now) const
+    {
+      if (state_ != JobState_Retry)
+      {
+        throw OrthancException(ErrorCode_BadSequenceOfCalls);
+      }
+      else
+      {
+        return retryTime_ >= now;
+      }
+    }
+  };
+
+
   class JobsMonitor : public boost::noncopyable
   {
   private:
-    class JobHandler : public boost::noncopyable
+    struct PriorityComparator
     {
-    private:
-      std::string               id_;
-      JobState                  state_;
-      std::auto_ptr<IJob>       job_;
-      int                       priority_;  // "+inf()" means highest priority
-      boost::posix_time::ptime  creationTime_;
-      boost::posix_time::ptime  lastUpdateTime_;
-      uint64_t                  runtime_;  // In milliseconds
+      bool operator() (JobHandler*& a,
+                       JobHandler*& b) const
+      {
+        return a->GetPriority() < b->GetPriority();
+      }                       
+    };
+
+    typedef std::map<std::string, JobHandler*>              JobsIndex;
+    typedef std::list<const JobHandler*>                    CompletedJobs;
+    typedef std::set<JobHandler*>                           RetryJobs;
+    typedef std::priority_queue<JobHandler*, 
+                                std::vector<JobHandler*>,   // Could be a "std::deque"
+                                PriorityComparator>         PendingJobs;
+
+    boost::mutex               mutex_;
+    JobsIndex                  jobsIndex_;
+    PendingJobs                pendingJobs_;
+    CompletedJobs              completedJobs_;
+    RetryJobs                  retryJobs_;
+
+    boost::condition_variable  pendingJobAvailable_;
+    size_t                     maxCompletedJobs_;
+
+
+#ifndef NDEBUG
+    bool IsPendingJob(const JobHandler& job) const
+    {
+      PendingJobs copy = pendingJobs_;
+      while (!copy.empty())
+      {
+        if (copy.top() == &job)
+        {
+          return true;
+        }
 
-    public:
-      JobHandler(IJob* job,
-                 int priority) :
-        id_(Toolbox::GenerateUuid()),
-        state_(JobState_Pending),
-        job_(job),
-        priority_(priority),
-        creationTime_(boost::posix_time::microsec_clock::universal_time()),
-        lastUpdateTime_(creationTime_),
-        runtime_(0)
+        copy.pop();
+      }
+
+      return false;
+    }
+
+    bool IsCompletedJob(const JobHandler& job) const
+    {
+      for (CompletedJobs::const_iterator it = completedJobs_.begin();
+           it != completedJobs_.end(); ++it)
       {
-        if (job == NULL)
+        if (*it == &job)
         {
-          throw OrthancException(ErrorCode_NullPointer);
+          return true;
+        }
+      }
+
+      return false;
+    }
+
+    bool IsRetryJob(JobHandler& job) const
+    {
+      return retryJobs_.find(&job) != retryJobs_.end();
+    }
+#endif
+
+
+    void CheckInvariants()
+    {
+#ifndef NDEBUG
+      {
+        PendingJobs copy = pendingJobs_;
+        while (!copy.empty())
+        {
+          assert(copy.top()->GetState() == JobState_Pending);
+          copy.pop();
         }
       }
 
-      const std::string& GetId() const
+      assert(completedJobs_.size() <= maxCompletedJobs_);
+
+      for (CompletedJobs::const_iterator it = completedJobs_.begin();
+           it != completedJobs_.end(); ++it)
+      {
+        assert((*it)->GetState() == JobState_Success ||
+               (*it)->GetState() == JobState_Failure);
+      }
+
+      for (RetryJobs::const_iterator it = retryJobs_.begin();
+           it != retryJobs_.end(); ++it)
+      {
+        assert((*it)->GetState() == JobState_Retry);
+      }
+
+      for (JobsIndex::iterator it = jobsIndex_.begin();
+           it != jobsIndex_.end(); ++it)
+      {
+        JobHandler& job = *it->second;
+
+        assert(job.GetId() == it->first);
+
+        switch (job.GetState())
+        {
+          case JobState_Pending:
+            assert(!IsRetryJob(job) && IsPendingJob(job) && !IsCompletedJob(job));
+            break;
+            
+          case JobState_Success:
+          case JobState_Failure:
+            assert(!IsRetryJob(job) && !IsPendingJob(job) && IsCompletedJob(job));
+            break;
+            
+          case JobState_Retry:
+            assert(IsRetryJob(job) && !IsPendingJob(job) && !IsCompletedJob(job));
+            break;
+            
+          case JobState_Running:
+          case JobState_Paused:
+            assert(!IsRetryJob(job) && !IsPendingJob(job) && !IsCompletedJob(job));
+            break;
+
+          default:
+            throw OrthancException(ErrorCode_InternalError);
+        }
+      }
+#endif
+    }
+
+
+    void ForgetOldCompletedJobs()
+    {
+      if (maxCompletedJobs_ != 0)
       {
-        return id_;
+        while (completedJobs_.size() > maxCompletedJobs_)
+        {
+          assert(completedJobs_.front() != NULL);
+
+          std::string id = completedJobs_.front()->GetId();
+
+          assert(jobsIndex_.find(id) != jobsIndex_.end());
+
+          jobsIndex_.erase(id);
+          delete(completedJobs_.front());
+          completedJobs_.pop_front();
+        }
       }
-    };
+    }
+
+
+    void MarkRunningAsCompleted(JobHandler& job,
+                                bool success)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+      assert(job.GetState() == JobState_Running);
+
+      job.SetState(success ? JobState_Success : JobState_Failure);
+
+      completedJobs_.push_back(&job);
+      ForgetOldCompletedJobs();
+
+      CheckInvariants();
+    }
+
+
+    void MarkRunningAsRetry(JobHandler& job,
+                            unsigned int timeout)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      assert(job.GetState() == JobState_Running &&
+             retryJobs_.find(&job) == retryJobs_.end());
+
+      retryJobs_.insert(&job);
+      job.SetRetryState(timeout);
+
+      CheckInvariants();
+    }
+
+
+    void MarkRunningAsPaused(JobHandler& job)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+      assert(job.GetState() == JobState_Running);
+
+      job.SetState(JobState_Paused);
+
+      CheckInvariants();
+    }
+
+
+    JobHandler* WaitPendingJob(unsigned int timeout)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+
+      while (pendingJobs_.empty())
+      {
+        if (timeout == 0)
+        {
+          pendingJobAvailable_.wait(lock);
+        }
+        else
+        {
+          bool success = pendingJobAvailable_.timed_wait
+            (lock, boost::posix_time::milliseconds(timeout));
+          if (!success)
+          {
+            return NULL;
+          }
+        }
+      }
+
+      JobHandler* job = pendingJobs_.top();
+      pendingJobs_.pop();
+      
+      job->SetState(JobState_Running);
+      return job;
+    }
+
 
   public:
+    JobsMonitor() :
+      maxCompletedJobs_(10)
+    {
+    }
+
+
+    ~JobsMonitor()
+    {
+      for (JobsIndex::iterator it = jobsIndex_.begin(); it != jobsIndex_.end(); ++it)
+      {
+        assert(it->second != NULL);
+        delete it->second;
+      }
+    }
+
+
+    void SetMaxCompletedJobs(size_t i)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      maxCompletedJobs_ = i;
+      ForgetOldCompletedJobs();
+
+      CheckInvariants();
+    }
+
+
+    void ListJobs(std::set<std::string>& target)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      for (JobsIndex::const_iterator it = jobsIndex_.begin();
+           it != jobsIndex_.end(); ++it)
+      {
+        target.insert(it->first);
+      }
+    }
+
+
     void Submit(std::string& id,
-                IJob* job,
+                IJob* job,        // Takes ownership
                 int priority)
     {
       std::auto_ptr<JobHandler>  handler(new JobHandler(job, priority));
+
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+      
       id = handler->GetId();
+      pendingJobs_.push(handler.get());
+      jobsIndex_.insert(std::make_pair(id, handler.release()));
+
+      pendingJobAvailable_.notify_one();
+
+      CheckInvariants();
     }
 
+
+    void Submit(IJob* job,        // Takes ownership
+                int priority)
+    {
+      std::string id;
+      Submit(id, job, priority);
+    }
+
+
     void SetPriority(const std::string& id,
                      int priority)
     {
-      // TODO
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      JobsIndex::iterator found = jobsIndex_.find(id);
+
+      if (found == jobsIndex_.end())
+      {
+        LOG(WARNING) << "Unknown job: " << id;
+      }
+      else
+      {
+        found->second->SetPriority(priority);
+
+        if (found->second->GetState() == JobState_Pending)
+        {
+          // If the job is pending, we need to reconstruct the
+          // priority queue, as the heap condition has changed
+
+          PendingJobs copy;
+          std::swap(copy, pendingJobs_);
+
+          assert(pendingJobs_.empty());
+          while (!copy.empty())
+          {
+            pendingJobs_.push(copy.top());
+            copy.pop();
+          }
+        }
+      }
+
+      CheckInvariants();
     }
 
+
     void Pause(const std::string& id)
     {
-      // TODO
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      JobsIndex::iterator found = jobsIndex_.find(id);
+
+      if (found == jobsIndex_.end())
+      {
+        LOG(WARNING) << "Unknown job: " << id;
+      }
+      else
+      {
+        switch (found->second->GetState())
+        {
+          case JobState_Pending:
+          {
+            // If the job is pending, we need to reconstruct the
+            // priority queue to remove it
+            PendingJobs copy;
+            std::swap(copy, pendingJobs_);
+
+            assert(pendingJobs_.empty());
+            while (!copy.empty())
+            {
+              if (copy.top()->GetId() != id)
+              {
+                pendingJobs_.push(copy.top());
+              }
+
+              copy.pop();
+            }
+
+            found->second->SetState(JobState_Paused);
+
+            break;
+          }
+
+          case JobState_Retry:
+          {
+            RetryJobs::iterator item = retryJobs_.find(found->second);
+            assert(item != retryJobs_.end());            
+            retryJobs_.erase(item);
+
+            found->second->SetState(JobState_Paused);
+
+            break;
+          }
+
+          case JobState_Paused:
+          case JobState_Success:
+          case JobState_Failure:
+            // Nothing to be done
+            break;
+
+          case JobState_Running:
+            found->second->SchedulePause();
+            break;
+
+          default:
+            throw OrthancException(ErrorCode_InternalError);
+        }
+      }
+
+      CheckInvariants();
     }
 
+
     void Resume(const std::string& id)
     {
-      // TODO
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      JobsIndex::iterator found = jobsIndex_.find(id);
+
+      if (found == jobsIndex_.end())
+      {
+        LOG(WARNING) << "Unknown job: " << id;
+      }
+      else if (found->second->GetState() != JobState_Paused)
+      {
+        LOG(WARNING) << "Cannot resume a job that is not paused: " << id;
+      }
+      else
+      {
+        found->second->SetState(JobState_Pending);
+        pendingJobs_.push(found->second);
+        pendingJobAvailable_.notify_one();
+      }
+
+      CheckInvariants();
     }
 
+
     void Resubmit(const std::string& id)
     {
-      // TODO
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      JobsIndex::iterator found = jobsIndex_.find(id);
+
+      if (found == jobsIndex_.end())
+      {
+        LOG(WARNING) << "Unknown job: " << id;
+      }
+      else if (found->second->GetState() != JobState_Failure)
+      {
+        LOG(WARNING) << "Cannot resubmit a job that has not failed: " << id;
+      }
+      else
+      {
+        bool ok = false;
+        for (CompletedJobs::iterator it = completedJobs_.begin(); 
+             it != completedJobs_.end(); ++it)
+        {
+          if (*it == found->second)
+          {
+            ok = true;
+            completedJobs_.erase(it);
+            break;
+          }
+        }
+
+        assert(ok);
+
+        found->second->SetState(JobState_Pending);
+        pendingJobs_.push(found->second);
+        pendingJobAvailable_.notify_one();
+      }
+
+      CheckInvariants();
     }
 
-    class JobToRun : public boost::noncopyable
+
+    void ScheduleRetries()
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      RetryJobs copy;
+      std::swap(copy, retryJobs_);
+
+      const boost::posix_time::ptime now = boost::posix_time::microsec_clock::universal_time();
+
+      assert(retryJobs_.empty());
+      for (RetryJobs::iterator it = copy.begin(); it != copy.end(); ++it)
+      {
+        if ((*it)->IsRetryReady(now))
+        {
+          (*it)->SetState(JobState_Pending);
+        }
+        else
+        {
+          retryJobs_.insert(*it);
+        }
+      }
+
+      CheckInvariants();
+    }
+
+
+    bool GetState(JobState& state,
+                  const std::string& id)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+      CheckInvariants();
+
+      JobsIndex::const_iterator it = jobsIndex_.find(id);
+      if (it == jobsIndex_.end())
+      {
+        return false;
+      }
+      else
+      {
+        state = it->second->GetState();
+        return true;
+      }
+    }
+
+
+    class RunningJob : public boost::noncopyable
     {
     private:
-      JobHandler*  handler_;
+      JobsMonitor&  that_;
+      JobHandler*   handler_;
+      JobState      targetState_;
+      unsigned int  retryTimeout_;
       
     public:
-      JobToRun(JobsMonitor& that,
-               unsigned int timeout) :
-        handler_(NULL)
+      RunningJob(JobsMonitor& that,
+                 unsigned int timeout) :
+        that_(that),
+        handler_(NULL),
+        targetState_(JobState_Failure),
+        retryTimeout_(0)
+      {
+        handler_ = that_.WaitPendingJob(timeout);
+      }
+
+      ~RunningJob()
       {
+        if (IsValid())
+        {
+          switch (targetState_)
+          {
+            case JobState_Failure:
+              that_.MarkRunningAsCompleted(*handler_, false);
+              break;
+
+            case JobState_Success:
+              that_.MarkRunningAsCompleted(*handler_, true);
+              break;
+
+            case JobState_Paused:
+              that_.MarkRunningAsPaused(*handler_);
+              break;            
+
+            case JobState_Retry:
+              that_.MarkRunningAsRetry(*handler_, retryTimeout_);
+              break;
+            
+            default:
+              assert(0);
+          }
+        }
       }
 
       bool IsValid() const
@@ -437,7 +1047,271 @@
         return handler_ != NULL;
       }
 
-      
+      const std::string& GetId() const
+      {
+        if (IsValid())
+        {
+          return handler_->GetId();
+        }
+        else
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }      
+      }
+
+      int GetPriority() const
+      {
+        if (IsValid())
+        {
+          return handler_->GetPriority();
+        }
+        else
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }      
+      }
+
+      bool IsPauseScheduled()
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        boost::mutex::scoped_lock lock(that_.mutex_);
+        that_.CheckInvariants();
+        assert(handler_->GetState() == JobState_Running);
+
+        return handler_->IsPauseScheduled();
+      }
+
+      IJob& GetJob()
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        boost::mutex::scoped_lock lock(that_.mutex_);
+        that_.CheckInvariants();
+        assert(handler_->GetState() == JobState_Running);
+
+        return handler_->GetJob();
+      }
+
+      void MarkSuccess()
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        targetState_ = JobState_Success;
+      }
+
+      void MarkFailure()
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        targetState_ = JobState_Failure;
+      }
+
+      void MarkPause()
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        targetState_ = JobState_Paused;
+      }
+
+      void MarkRetry(unsigned int timeout)
+      {
+        if (!IsValid())
+        {
+          throw OrthancException(ErrorCode_BadSequenceOfCalls);
+        }
+
+        targetState_ = JobState_Retry;
+        retryTimeout_ = timeout;
+      }
     };
   };
 }
+
+
+
+class DummyJob : public Orthanc::IJob
+{
+private:
+  JobStepResult  result_;
+
+public:
+  DummyJob() :
+    result_(Orthanc::JobStepStatus_Success)
+  {
+  }
+
+  explicit DummyJob(JobStepResult result) :
+  result_(result)
+  {
+  }
+
+  virtual JobStepResult* ExecuteStep()
+  {
+    return new JobStepResult(result_);
+  }
+
+  virtual void ReleaseResources()
+  {
+  }
+
+  virtual float GetProgress()
+  {
+    return 0;
+  }
+
+  virtual void FormatStatus(Json::Value& value)
+  {
+  }
+};
+
+
+static bool CheckState(Orthanc::JobsMonitor& monitor,
+                       const std::string& id,
+                       Orthanc::JobState state)
+{
+  Orthanc::JobState s;
+  if (monitor.GetState(s, id))
+  {
+    return state == s;
+  }
+  else
+  {
+    return false;
+  }
+}
+
+
+TEST(JobsMonitor, Priority)
+{
+  JobsMonitor monitor;
+
+  std::string i1, i2, i3, i4;
+  monitor.Submit(i1, new DummyJob(), 10);
+  monitor.Submit(i2, new DummyJob(), 30);
+  monitor.Submit(i3, new DummyJob(), 20);
+  monitor.Submit(i4, new DummyJob(), 5);  
+
+  monitor.SetMaxCompletedJobs(2);
+
+  std::set<std::string> id;
+  monitor.ListJobs(id);
+
+  ASSERT_EQ(4u, id.size());
+  ASSERT_TRUE(id.find(i1) != id.end());
+  ASSERT_TRUE(id.find(i2) != id.end());
+  ASSERT_TRUE(id.find(i3) != id.end());
+  ASSERT_TRUE(id.find(i4) != id.end());
+
+  ASSERT_TRUE(CheckState(monitor, i2, Orthanc::JobState_Pending));
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    ASSERT_EQ(30, job.GetPriority());
+    ASSERT_EQ(i2, job.GetId());
+
+    ASSERT_TRUE(CheckState(monitor, i2, Orthanc::JobState_Running));
+  }
+
+  ASSERT_TRUE(CheckState(monitor, i2, Orthanc::JobState_Failure));
+  ASSERT_TRUE(CheckState(monitor, i3, Orthanc::JobState_Pending));
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    ASSERT_EQ(20, job.GetPriority());
+    ASSERT_EQ(i3, job.GetId());
+
+    job.MarkSuccess();
+
+    ASSERT_TRUE(CheckState(monitor, i3, Orthanc::JobState_Running));
+  }
+
+  ASSERT_TRUE(CheckState(monitor, i3, Orthanc::JobState_Success));
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    ASSERT_EQ(10, job.GetPriority());
+    ASSERT_EQ(i1, job.GetId());
+  }
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    ASSERT_EQ(5, job.GetPriority());
+    ASSERT_EQ(i4, job.GetId());
+  }
+
+  {
+    JobsMonitor::RunningJob job(monitor, 1);
+    ASSERT_FALSE(job.IsValid());
+  }
+
+  Orthanc::JobState s;
+  ASSERT_TRUE(monitor.GetState(s, i1));
+  ASSERT_FALSE(monitor.GetState(s, i2));  // Removed because oldest
+  ASSERT_FALSE(monitor.GetState(s, i3));  // Removed because second oldest
+  ASSERT_TRUE(monitor.GetState(s, i4));
+
+  monitor.SetMaxCompletedJobs(1);  // (*)
+  ASSERT_FALSE(monitor.GetState(s, i1));  // Just discarded by (*)
+  ASSERT_TRUE(monitor.GetState(s, i4));
+}
+
+
+TEST(JobsMonitor, Resubmit)
+{
+  JobsMonitor monitor;
+
+  std::string id;
+  monitor.Submit(id, new DummyJob(), 10);
+
+  ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Pending));
+
+  monitor.Resubmit(id);
+  ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Pending));
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    job.MarkFailure();
+
+    ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Running));
+
+    monitor.Resubmit(id);
+    ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Running));
+  }
+
+  ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Failure));
+
+  monitor.Resubmit(id);
+  ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Pending));
+
+  {
+    JobsMonitor::RunningJob job(monitor, 0);
+    ASSERT_TRUE(job.IsValid());
+    ASSERT_EQ(id, job.GetId());
+
+    job.MarkSuccess();
+    ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Running));
+  }
+
+  ASSERT_TRUE(CheckState(monitor, id, Orthanc::JobState_Success));
+}