Method checkType

websocket_message.go:212–261

(opChain *chain, types ...int)

}

func (wm *WebsocketMessage) checkType(opChain *chain, types ...int) {
	if opChain.failed() {
		return
	}

	if len(types) == 0 {
		opChain.fail(AssertionFailure{
			Type: AssertUsage,
			Errors: []error{
				errors.New("missing type argument"),
			},
		})
		return
	}

	found := false
	for _, t := range types {
		if t == wm.typ {
			found = true
			break
		}
	}

	if !found {
		if len(types) == 1 {
			opChain.fail(AssertionFailure{
				Type:     AssertEqual,
				Actual:   &AssertionValue{wsMessageType(wm.typ)},
				Expected: &AssertionValue{wsMessageType(types[0])},
				Errors: []error{
					errors.New("expected: message types are equal"),
				},
			})
		} else {
			typeList := make([]interface{}, 0, len(types))
			for _, t := range types {
				typeList = append(typeList, wsMessageType(t))
			}

			opChain.fail(AssertionFailure{
				Type:     AssertBelongs,
				Actual:   &AssertionValue{wsMessageType(wm.typ)},
				Expected: &AssertionValue{AssertionList(typeList)},
				Errors: []error{
					errors.New("expected: message type belongs to given list"),
				},
			})
		}
	}
}

func (wm *WebsocketMessage) checkNotType(opChain *chain, types ...int) {
	if opChain.failed() {