requests_test.go

// Copyright (c) DataStax, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package proxycore

import (
	"io"
	"testing"

	"github.com/datastax/go-cassandra-native-protocol/frame"
	"github.com/stretchr/testify/assert"
)

func TestPendingRequests(t *testing.T) {
	const max = 10

	p := newPendingRequests(max)

	errs := make([]error, 0)

	for i := int16(0); i < max; i++ {
		assert.Equal(t, i, p.store(&testPendingRequest{stream: i, errs: &errs}))
	}
	assert.Equal(t, int16(-1), p.store(&testPendingRequest{}))

	r := p.loadAndDelete(0).(*testPendingRequest)
	assert.Equal(t, int16(0), r.stream)

	r = p.loadAndDelete(9).(*testPendingRequest)
	assert.Equal(t, int16(9), r.stream)

	assert.Equal(t, int16(0), p.store(&testPendingRequest{stream: 0, errs: &errs}))
	assert.Equal(t, int16(9), p.store(&testPendingRequest{stream: 9, errs: &errs}))
	assert.Equal(t, int16(-1), p.store(&testPendingRequest{}))

	p.closing(io.EOF)

	assert.Equal(t, 10, len(errs))

	for _, err := range errs {
		assert.ErrorAs(t, err, &io.EOF)
	}
}

type testPendingRequest struct {
	stream int16
	errs   *[]error
}

func (t *testPendingRequest) Execute(_ bool) {
	panic("not implemented")
}

func (t *testPendingRequest) IsPrepareRequest() bool {
	panic("not implemented")
}

func (t *testPendingRequest) Frame() interface{} {
	panic("not implemented")
}

func (t *testPendingRequest) OnClose(err error) {
	*t.errs = append(*t.errs, err)
}

func (t testPendingRequest) OnResult(_ *frame.RawFrame) {
	panic("not implemented")
}