lifo_disk_queue_test.go

package queue

import (
	"context"
	"errors"
	"io/ioutil"
	"os"
	"reflect"
	"testing"
)

func TestNewLifoDiskQueue1(t *testing.T) {
	file, err := ioutil.TempFile("", "")
	if err != nil {
		panic(err)
	}
	defer os.RemoveAll(file.Name())
	file.Close()
	queue, err := NewLifoDiskQueue(file.Name())
	if err != nil {
		panic(err)
	}
	name := "TestNewLifoDiskQueue"
	{
		if length := queue.Len(); length != 0 {
			t.Error(name, "空队列-获取长度为0", length)
		}
		if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueEmpty) {
			t.Error(name, "空队列-Get数据返回ErrQueueEmpty", data, err)
		}
	}
	{
		if err := queue.Put(nil, []byte{}); err != nil {
			t.Error(name, "空队列-Put数据返回nil", err)
		}
		if data, err := queue.Get(nil); !reflect.DeepEqual(data, []byte{}) || err != nil {
			t.Error(name, "非空队列-Get数据返回nil", data, err)
		}
	}
	{
		if err := queue.Close(); err != nil {
			t.Error(name, "队列关闭返回nil", err)
		}
	}
	{
		if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueClosed) {
			t.Error(name, "关闭队列-Get数据返回ErrQueueClosed", data, err)
		}
		if data, err := queue.Get(context.Background()); data != nil || !errors.Is(err, ErrQueueClosed) {
			t.Error(name, "关闭队列-Get数据返回ErrQueueClosed", data, err)
		}
		if err := queue.Put(nil, []byte{}); !errors.Is(err, ErrQueueClosed) {
			t.Error(name, "关闭队列-推送数据返回ErrQueueClosed", err)
		}
		if err := queue.Put(context.Background(), []byte{}); !errors.Is(err, ErrQueueClosed) {
			t.Error(name, "关闭队列-推送数据返回ErrQueueClosed", err)
		}
	}
	{
		if err := queue.Close(); err != nil {
			t.Error(name, "队列关闭返回nil", err)
		}
	}
}

func TestNewLifoDiskQueue2(t *testing.T) {
	file, err := ioutil.TempFile("", "")
	if err != nil {
		panic(err)
	}
	defer os.RemoveAll(file.Name())
	file.Close()
	queue, err := NewLifoDiskQueue(file.Name())
	if err != nil {
		panic(err)
	}
	name := "TestNewLifoDiskQueue2"
	if err := queue.Put(nil, []byte{}); err != nil {
		t.Error(name, "空队列-Put数据返回nil", err)
	}
	if err := queue.Close(); err != nil {
		t.Error(name, "队列关闭返回nil", err)
	}
	queue, err = NewLifoDiskQueue(file.Name())
	if err != nil {
		panic(err)
	}
	if data, err := queue.Get(nil); !reflect.DeepEqual(data, []byte{}) || err != nil {
		t.Error(name, "非空队列-Get数据返回nil", data, err)
	}
}