Add tests to diode

2021-02-04 15:42:08 -05:00
parent c7af3b8256
commit 26f838167e
3 changed files with 60 additions and 1 deletions
@@ -20,6 +20,14 @@ func (d *diode) set(data message) {
 	d.d.Set(diodes.GenericDataType(&data))
 }

+func (d *diode) tryNext() (*message, bool) {
+	data, ok := d.d.TryNext()
+	if !ok {
+		return nil, ok
+	}
+	return (*message)(data), true
+}
+
 func (d *diode) next() *message {
 	data := d.d.Next()
 	return (*message)(data)