dird: add comparison operators to SchedulerJobItem

bareos · Nov 21, 2019 · 485791e · 485791e
1 parent b8748e2
commit 485791e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/core/src/dird/scheduler_job_item_queue.h b/core/src/dird/scheduler_job_item_queue.h
@@ -43,6 +43,13 @@ struct SchedulerJobItem {
   {
     is_valid = job && runtime;
   };
+  bool operator==(const SchedulerJobItem& rhs) const
+  {
+    return runtime == rhs.runtime && job == rhs.job &&
+           priority == rhs.priority && run == rhs.run;
+  }
+  bool operator!=(const SchedulerJobItem& rhs) const { return !(*this == rhs); }
+
   JobResource* job{nullptr};
   RunResource* run{nullptr};
   time_t runtime{0};

diff --git a/core/src/tests/scheduler_job_item_queue.cc b/core/src/tests/scheduler_job_item_queue.cc
@@ -42,6 +42,19 @@ TEST(scheduler_job_item_queue, job_item)
   EXPECT_TRUE(item_unitialised.is_valid);
 }
 
+TEST(scheduler_job_item_queue, compare_job_items)
+{
+  JobResource job[2];
+  RunResource run[2];
+
+  SchedulerJobItem item1(&job[0], &run[0], time(nullptr), 10);
+  SchedulerJobItem item2 = item1;
+  SchedulerJobItem item3(&job[1], &run[1], time(nullptr) + 3600, 11);
+
+  EXPECT_EQ(item1, item2);
+  EXPECT_NE(item1, item3);
+}
+
 TEST(scheduler_job_item_queue, priority_and_time)
 {
   time_t now = time(nullptr);