chainscripttest: withDegree (#28)

* chainscripttest: withDegree * outDegree: add error for append failures
stratumn · Sep 18, 2018 · 6cacc3d · 6cacc3d
1 parent 88d0029
commit 6cacc3d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/chainscripttest/builder.go b/chainscripttest/builder.go
@@ -79,6 +79,12 @@ func (lb *LinkBuilder) WithData(t *testing.T, data interface{}) *LinkBuilder {
 	return lb
 }
 
+// WithDegree sets the link's out degree.
+func (lb *LinkBuilder) WithDegree(d int) *LinkBuilder {
+	lb.Link.Meta.OutDegree = int32(d)
+	return lb
+}
+
 // WithInvalidFields makes the link invalid by setting some fields to invalid
 // values.
 func (lb *LinkBuilder) WithInvalidFields() *LinkBuilder {

diff --git a/link.go b/link.go
@@ -36,6 +36,7 @@ const (
 // Link errors.
 var (
 	ErrMissingVersion     = errors.New("version is missing")
+	ErrOutDegree          = errors.New("cannot create link: parent has too many children")
 	ErrUnknownLinkVersion = errors.New("unknown link version")
 	ErrUnknownClientID    = errors.New("link was created with a unknown client: can't deserialize it")
 )