27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
import unittest
|
|
from cosmosmesh_privacy_preserving_federated_ import PlanDeltaCRDT
|
|
|
|
|
|
class TestPlanDeltaCRDT(unittest.TestCase):
|
|
def test_merge_numeric_max(self):
|
|
a = PlanDeltaCRDT(delta_id="d1", changes={"x": 1}, ts=1, actor="A", version_vector={"A": 1})
|
|
b = PlanDeltaCRDT(delta_id="d2", changes={"x": 3}, ts=2, actor="B", version_vector={"B": 2})
|
|
merged = a.merge(b)
|
|
self.assertEqual(merged.changes["x"], 3)
|
|
self.assertEqual(merged.ts, 2)
|
|
self.assertEqual(merged.version_vector.get("A"), 1)
|
|
self.assertEqual(merged.version_vector.get("B"), 2)
|
|
|
|
def test_merge_non_numeric_prefers_later_timestamp(self):
|
|
a = PlanDeltaCRDT(delta_id="d1", changes={"flag": True}, ts=1, actor="A", version_vector={"A": 1})
|
|
b = PlanDeltaCRDT(delta_id="d2", changes={"flag": False}, ts=2, actor="B", version_vector={"B": 2})
|
|
merged = a.merge(b)
|
|
self.assertIs(merged.changes["flag"], False)
|
|
self.assertEqual(merged.ts, 2)
|
|
self.assertEqual(merged.version_vector.get("A"), 1)
|
|
self.assertEqual(merged.version_vector.get("B"), 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|