Simplify distributor unit tests.

2 files changed, 7 insertions, 0 deletions

diff --git a/vdslib/src/vespa/vdslib/state/cluster_state_bundle.cpp b/vdslib/src/vespa/vdslib/state/cluster_state_bundle.cpp
index c55f1aadd06..a2bbba5e52c 100644
--- a/vdslib/src/vespa/vdslib/state/cluster_state_bundle.cpp
+++ b/vdslib/src/vespa/vdslib/state/cluster_state_bundle.cpp
@@ -31,4 +31,10 @@ ClusterStateBundle::getVersion() const
     return _baselineClusterState->getVersion();
 }
 
+bool
+ClusterStateBundle::operator==(const ClusterStateBundle &rhs) const
+{
+    return *_baselineClusterState == *rhs._baselineClusterState;
+}
+
 }
diff --git a/vdslib/src/vespa/vdslib/state/cluster_state_bundle.h b/vdslib/src/vespa/vdslib/state/cluster_state_bundle.h
index c54df1d1952..77d26092f4e 100644
--- a/vdslib/src/vespa/vdslib/state/cluster_state_bundle.h
+++ b/vdslib/src/vespa/vdslib/state/cluster_state_bundle.h
@@ -21,6 +21,7 @@ public:
     const std::shared_ptr<const ClusterState> &getBaselineClusterState() const;
     const std::shared_ptr<const ClusterState> &getDerivedClusterState(document::BucketSpace bucketSpace) const;
     uint32_t getVersion() const;
+    bool operator==(const ClusterStateBundle &rhs) const;
 };
 
 }