featurebase/bench/import_test.go

package bench_test

import (
	"bytes"
	"testing"

	"io/ioutil"

	"os"

	"github.com/pilosa/pilosa/bench"
)

func TestImportInit(t *testing.T) {
	imp := bench.NewImport(os.Stdin, os.Stdout, os.Stderr)
	imp.BaseBitmapID = 0
	imp.MaxBitmapID = 10
	imp.BaseProfileID = 0
	imp.MaxProfileID = 10
	imp.RandomBitmapOrder = false
	imp.MinBitsPerMap = 2
	imp.MaxBitsPerMap = 3
	imp.AgentControls = "width"
	imp.Seed = 0

	imp.Init([]string{"blah"}, 2)
	f, err := os.Open(imp.Paths[0])
	if err != nil {
		t.Fatalf("Couldn't open file: %v, err: %v", imp.Paths[0], err)
	}
	bytes, err := ioutil.ReadAll(f)
	if err != nil {
		t.Fatalf("error reading file: %v", err)
	}

	expected := `
0,21
0,22
1,20
1,22
2,22
2,26
3,21
3,23
4,21
4,22
5,20
5,28
6,23
6,27
7,20
7,20
8,23
8,29
9,23
9,29
`[1:]

	if string(bytes) != expected {
		t.Fatalf("unexpected result: %v", string(bytes))
	}
}

func TestGenerateImportCSVNonRand(t *testing.T) {
	b := bytes.NewBuffer(make([]byte, 0))

	bench.GenerateImportCSV(b, 0, 10, 20, 30, 2, 3, 2, false)

	bytes, err := ioutil.ReadAll(b)
	if err != nil {
		t.Fatalf("Error reading buffer: %v", err)
	}

	expected := `
0,21
0,22
1,20
1,22
2,22
2,26
3,21
3,23
4,21
4,22
5,20
5,28
6,23
6,27
7,20
7,20
8,23
8,29
9,23
9,29
`[1:]

	if string(bytes) != expected {
		t.Fatalf("unexpected value for generated csv: \n%v", string(bytes))
	}
}

func TestGenerateImportCSVRand(t *testing.T) {
	b := bytes.NewBuffer(make([]byte, 0))

	bench.GenerateImportCSV(b, 0, 10, 21, 29, 1, 4, 0, true)

	bytes, err := ioutil.ReadAll(b)
	if err != nil {
		t.Fatalf("Error reading buffer: %v", err)
	}

	expected := `
8,25
2,23
3,22
3,28
3,28
0,25
0,23
5,26
5,23
7,21
7,23
1,25
6,23
6,27
6,25
9,26
4,22
4,24
`[1:]

	if string(bytes) != expected {
		t.Fatalf("unexpected value for generated csv: \n%v", string(bytes))
	}
}