コード例 #1
0
func commonTestStateDriverWatchAllStateModify(t *testing.T, d core.StateDriver) {
	state := &testState{IntField: 1234, StrField: "testString"}
	modState := &testState{IntField: 5678, StrField: "modString"}
	baseKey := "modify"
	key := baseKey + "/testKeyWatchAll"

	err := d.WriteState(key, state, json.Marshal)
	if err != nil {
		t.Fatalf("failed to write state. Error: %s", err)
	}
	defer func() {
		d.ClearState(key)
	}()

	recvErr := make(chan error, 1)
	stateCh := make(chan core.WatchState, 1)
	timer := time.After(waitTimeout)

	go func(rsps chan core.WatchState, retErr chan error) {
		err := d.WatchAllState(baseKey, state, json.Unmarshal, stateCh)
		if err != nil {
			retErr <- err
			return
		}
	}(stateCh, recvErr)

	// trigger modify after a slight pause to ensure that events are not missed
	time.Sleep(time.Second)
	err = d.WriteState(key, modState, json.Marshal)
	if err != nil {
		t.Fatalf("failed to write state. Error: %s", err)
	}

for_label:
	for {
		select {
		case watchState := <-stateCh:
			s := watchState.Curr.(*testState)
			if s.IntField != modState.IntField || s.StrField != modState.StrField {
				t.Fatalf("Watch state mismatch. Expctd: %+v, Rcvd: %+v", modState, s)
			}
			if watchState.Prev == nil {
				t.Fatalf("Received a modify event without previous state value set.")
			}
			s = watchState.Prev.(*testState)
			if s.IntField != state.IntField || s.StrField != state.StrField {
				t.Fatalf("Watch state mismatch. Expctd: %+v, Rcvd: %+v", state, s)
			}
			break for_label
		case err := <-recvErr:
			t.Fatalf("Watch failed. Error: %s", err)
			break for_label
		case <-timer:
			t.Fatalf("timed out waiting for events")
			break for_label
		}
	}
}
コード例 #2
0
func commonTestStateDriverClearState(t *testing.T, d core.StateDriver) {
	state := &testState{IntField: 1234, StrField: "testString"}
	key := "testKeyClear"

	err := d.WriteState(key, state, json.Marshal)
	if err != nil {
		t.Fatalf("failed to write state. Error: %s", err)
	}

	err = d.ClearState(key)
	if err != nil {
		t.Fatalf("failed to clear state. Error: %s", err)
	}
}
コード例 #3
0
func commonTestStateDriverReadStateAfterClear(t *testing.T, d core.StateDriver) {
	state := &testState{IntField: 1234, StrField: "testString"}
	key := "testKeyReadClear"

	err := d.WriteState(key, state, json.Marshal)
	if err != nil {
		t.Fatalf("failed to write state. Error: %s", err)
	}

	err = d.ClearState(key)
	if err != nil {
		t.Fatalf("failed to clear state. Error: %s", err)
	}

	readState := &testState{}
	err = d.ReadState(key, readState, json.Unmarshal)
	if err == nil {
		t.Fatalf("Able to read cleared state!. Key: %s, Value: %v",
			key, readState)
	}
}