f Add test coverage for insert_or_update

Author: tnull

src/data_store.rs

@@ -270,5 +270,18 @@ mod tests {
 		let bogus_id = TestObjectId { id: [84u8; 4] };
 		let update = TestObjectUpdate { id: bogus_id, data: [12u8; 3] };
 		assert_eq!(Ok(DataStoreUpdateResult::NotFound), data_store.update(&update));
+
+		// Check `insert_or_update` inserts unknown objects
+		let iou_id = TestObjectId { id: [55u8; 4] };
+		let iou_object = TestObject { id: iou_id, data: [34u8; 3] };
+		assert_eq!(Ok(true), data_store.insert_or_update(&iou_object));
+
+		// Check `insert_or_update` doesn't update the same object
+		assert_eq!(Ok(false), data_store.insert_or_update(&iou_object));
+
+		// Check `insert_or_update` updates if object changed
+		let mut new_iou_object = iou_object.clone();
+		new_iou_object.data[0] += 1;
+		assert_eq!(Ok(true), data_store.insert_or_update(&new_iou_object));
 	}
 }