diskqueue/diskqueue_test.go at master · kkdai/diskqueue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package diskqueue_test

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"strconv"
	"testing"
	"time"

	. "github.com/kkdai/diskqueue"
)

func TestEmptyQueue(t *testing.T) {
	dqName := "test_disk_queue" + strconv.Itoa(int(time.Now().Unix()))
	tmpDir, err := ioutil.TempDir("", fmt.Sprintf("nsq-test-%d", time.Now().UnixNano()))
	if err != nil {
		panic(err)
	}

	defer os.RemoveAll(tmpDir)
	dq := NewDiskqueue(dqName, tmpDir)

	if dq == nil {
		t.Error("empty test error")
	}
	dq.Close()
}

func TestBasicQueue(t *testing.T) {
	dqName := "test_disk_queue" + strconv.Itoa(int(time.Now().Unix()))
	tmpDir, err := ioutil.TempDir("", fmt.Sprintf("nsq-test-%d", time.Now().UnixNano()))
	if err != nil {
		panic(err)
	}

	defer os.RemoveAll(tmpDir)
	dq := NewDiskqueue(dqName, tmpDir)
	err = dq.Put([]byte("00"))
	if err != nil {
		t.Error("err on test1 :", err)
	}

	log.Println("First put done")
	err = dq.Put([]byte("111"))
	if err != nil {
		t.Error("err on test2 :", err)
	}

	data := <-dq.ReadChan()
	log.Println("READ: ", string(data))
	dq.Close()
}

func TestEmpty(t *testing.T) {
	dqName := "test_disk_queue" + strconv.Itoa(int(time.Now().Unix()))
	tmpDir, err := ioutil.TempDir("", fmt.Sprintf("nsq-test-%d", time.Now().UnixNano()))
	if err != nil {
		panic(err)
	}

	dq := NewDiskqueue(dqName, tmpDir)
	err = dq.Empty()
	if err != nil {
		t.Error("Empty err:", err)
	}
}