A few more MustXxx functions

Author: pascallouisperez

dbstore_test.go

@@ -21,11 +21,8 @@ import (
 )
 
 func (s *DbZuite) TestExample() {
-	ws, err := s.store.defs.NewWorksheet("simple")
-	require.NoError(s.T(), err)
-
-	err = ws.Set("name", NewText("Alice"))
-	require.NoError(s.T(), err)
+	ws := s.store.defs.MustNewWorksheet("simple")
+	ws.MustSet("name", NewText("Alice"))
 
 	s.MustRunTransaction(func(tx *runner.Tx) error {
 		session := s.store.Open(tx)

worksheets.go

@@ -65,6 +65,14 @@ func NewDefinitions(reader io.Reader) (*Definitions, error) {
 	}, nil
 }
 
+func (defs *Definitions) MustNewWorksheet(name string) *Worksheet {
+	ws, err := defs.NewWorksheet(name)
+	if err != nil {
+		panic(err)
+	}
+	return ws
+}
+
 func (defs *Definitions) NewWorksheet(name string) (*Worksheet, error) {
 	ws, err := defs.newUninitializedWorksheet(name)
 	if err != nil {
@@ -141,6 +149,12 @@ func (ws *Worksheet) Name() string {
 	return ws.def.name
 }
 
+func (ws *Worksheet) MustSet(name string, value Value) {
+	if err := ws.Set(name, value); err != nil {
+		panic(err)
+	}
+}
+
 func (ws *Worksheet) Set(name string, value Value) error {
 	// TODO(pascal): create a 'change', and then commit that change, garantee
 	// that commits are atomic, and either win or lose the race by using

worksheets_test.go

@@ -22,8 +22,7 @@ func (s *Zuite) TestExample() {
 	defs, err := NewDefinitions(strings.NewReader(`worksheet simple {1:name text}`))
 	require.NoError(s.T(), err)
 
-	ws, err := defs.NewWorksheet("simple")
-	require.NoError(s.T(), err)
+	ws := defs.MustNewWorksheet("simple")
 
 	isSet, err := ws.IsSet("name")
 	require.NoError(s.T(), err)