Add term execution count and sleep tests

Signed-off-by: Ramil Mirhasanov <[email protected]>
Signed-off-by: Prasad Chandrasekaran <[email protected]>
Co-authored-by: Marek Siarkowicz <[email protected]>

runtime/http.go

@@ -20,6 +20,7 @@ import (
 	"net"
 	"net/http"
 	"sort"
+	"strconv"
 	"strings"
 )
 
@@ -87,12 +88,24 @@ func (*httpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 			sort.Strings(fps)
 			lines := make([]string, len(fps))
 			for i := range lines {
-				s, _ := status(fps[i])
+				s, _, _ := status(fps[i])
 				lines[i] = fps[i] + "=" + s
 			}
 			w.Write([]byte(strings.Join(lines, "\n") + "\n"))
+		} else if strings.HasSuffix(key, "/count") {
+			fp := key[:len(key)-len("/count")]
+			_, count, err := status(fp)
+			if err != nil {
+				if err == ErrNoExist {
+					http.Error(w, "failed to GET: "+err.Error(), http.StatusNotFound)
+				} else {
+					http.Error(w, "failed to GET: "+err.Error(), http.StatusInternalServerError)
+				}
+				return
+			}
+			w.Write([]byte(strconv.Itoa(count)))
 		} else {
-			status, err := status(key)
+			status, _, err := status(key)
 			if err != nil {
 				http.Error(w, "failed to GET: "+err.Error(), http.StatusNotFound)
 			}

runtime/runtime.go

@@ -112,25 +112,25 @@ func disable(name string) error {
 	return nil
 }
 
-// Status gives the current setting for the failpoint
-func Status(failpath string) (string, error) {
+// Status gives the current setting and termcount for the failpoint
+func Status(failpath string) (string, int, error) {
 	failpointsMu.Lock()
 	defer failpointsMu.Unlock()
 	return status(failpath)
 }
 
-func status(failpath string) (string, error) {
+func status(failpath string) (string, int, error) {
 	fp := failpoints[failpath]
 	if fp == nil {
-		return "", ErrNoExist
+		return "", 0, ErrNoExist
 	}
 
 	t := fp.t
 	if t == nil {
-		return "", ErrDisabled
+		return "", 0, ErrDisabled
 	}
 
-	return t.desc, nil
+	return t.desc, t.counter, nil
 }
 
 func List() []string {

runtime/terms.go

@@ -41,6 +41,8 @@ type terms struct {
 
 	// mu protects the state of the terms chain
 	mu sync.Mutex
+	// counts executions
+	counter int
 }
 
 // term is an executable unit of the failpoint terms chain
@@ -102,6 +104,7 @@ func (t *terms) eval() (interface{}, error) {
 	defer t.mu.Unlock()
 	for _, term := range t.chain {
 		if term.mods.allow() {
+			t.counter++
 			return term.do(), nil
 		}
 	}

runtime/termscounter_test.go

@@ -0,0 +1,151 @@
+package runtime_test
+
+import (
+	"go.etcd.io/gofail/runtime"
+	"testing"
+)
+
+// This variable mimics the code generated by gofail code package.
+// This works in tandem with exampleFunc function.
+var __fp_ExampleString *runtime.Failpoint //nolint:stylecheck
+
+// check if failpoint is initialized as gofail
+// tests can clear global variables of runtime packages
+func initFP() {
+	if __fp_ExampleString == nil { //nolint:stylecheck
+		__fp_ExampleString = runtime.NewFailpoint("ExampleString") //nolint:stylecheck
+	}
+}
+
+func TestTermsCounter(t *testing.T) {
+	testcases := []struct {
+		name              string
+		fp                string
+		failpointTerm     string
+		runBeforeEnabling int
+		runAfterEnabling  int
+		wantCount         int
+	}{
+		{
+			name:             "Terms limit Failpoint",
+			fp:               "ExampleString",
+			failpointTerm:    `10*sleep(10)->1*return("abc")`,
+			runAfterEnabling: 12,
+			// This example tests mods which allows users to restrict the
+			// number of failpoint actions as against their callsite executions.
+			// This is the reason why wantCount < runAfterEnabling
+			// In a real world example you can hit a code spot a million times but
+			// using mods restrict the associated fallpoint actions to run twice.
+			wantCount: 11,
+		},
+		{
+			name:              "Inbetween Enabling Failpoint",
+			fp:                "ExampleString",
+			failpointTerm:     `10*sleep(10)->1*return("abc")`,
+			runBeforeEnabling: 2,
+			runAfterEnabling:  3,
+			wantCount:         3,
+		},
+		{
+			name:              "Before Enabling Failpoint",
+			fp:                "ExampleString",
+			failpointTerm:     `10*sleep(10)->1*return("abc")`,
+			runBeforeEnabling: 2,
+			runAfterEnabling:  0,
+			wantCount:         0,
+		},
+	}
+
+	initFP()
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+
+			for i := 0; i < tc.runBeforeEnabling; i++ {
+				exampleFunc()
+			}
+
+			err := runtime.Enable(tc.fp, tc.failpointTerm)
+			if err != nil {
+				t.Fatal(err)
+			}
+			defer runtime.Disable(tc.fp)
+			for i := 0; i < tc.runAfterEnabling; i++ {
+				exampleFunc()
+			}
+			_, count, err := runtime.Status(tc.fp)
+			if err != nil {
+				t.Fatal(err)
+			}
+			if tc.wantCount != count {
+				t.Fatal("counter is not properly incremented")
+			}
+		})
+	}
+}
+
+func TestEnablingNewTermResetsCount(t *testing.T) {
+	testcases := []struct {
+		name       string
+		fp         string
+		oldTerm    string
+		newTerm    string
+		runOldTerm int
+		runNewTerm int
+		wantCount  int
+	}{
+		{
+			name:       "Change and Reset Counter",
+			fp:         "ExampleString",
+			oldTerm:    `10*sleep(10)->1*return("abc")`,
+			newTerm:    "sleep(10)",
+			runOldTerm: 2,
+			runNewTerm: 3,
+			wantCount:  3,
+		},
+	}
+
+	initFP()
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+			err := runtime.Enable(tc.fp, tc.oldTerm)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			for i := 0; i < tc.runOldTerm; i++ {
+				exampleFunc()
+			}
+			err = runtime.Enable(tc.fp, tc.newTerm)
+			if err != nil {
+				t.Fatal(err)
+			}
+			defer runtime.Disable(tc.fp)
+
+			for i := 0; i < tc.runNewTerm; i++ {
+				exampleFunc()
+			}
+			_, count, err := runtime.Status(tc.fp)
+			if err != nil {
+				t.Fatal(err)
+			}
+			if tc.wantCount != count {
+				t.Fatal("counter is not properly incremented")
+			}
+		})
+	}
+
+}
+
+// This function mimics a customized code that is generated by gofail code package.
+func exampleFunc() string {
+	if vExampleString, __fpErr := __fp_ExampleString.Acquire(); __fpErr == nil { //nolint:stylecheck
+		ExampleString, __fpTypeOK := vExampleString.(string) //nolint:stylecheck
+		if !__fpTypeOK {                                     //nolint:stylecheck
+			goto __badTypeExampleString //nolint:stylecheck
+		}
+		return ExampleString
+	__badTypeExampleString: //nolint:stylecheck
+		__fp_ExampleString.BadType(vExampleString, "string") //nolint:stylecheck
+	}
+	return "example"
+}