Function TestFileRef

file_test.go:19–106

(t *testing.T)

)

func TestFileRef(t *testing.T) {
	tmpDir, _ := ioutil.TempDir("", "mossStore")
	defer os.RemoveAll(tmpDir)

	file, _ := os.OpenFile(path.Join(tmpDir, "test.mmap"),
		os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600)

	fref := &FileRef{file: file, refs: 1}

	var m sync.Mutex
	x := 0
	after := 0
	fref.OnBeforeClose(func() {
		m.Lock()
		x++
		m.Unlock()
	})
	fref.OnAfterClose(func() {
		m.Lock()
		after++
		m.Unlock()
	})

	file2 := fref.AddRef()

	fref.m.Lock()
	if fref.refs != 2 {
		t.Errorf("expected 2 refs")
	}
	if fref.file != file2 {
		t.Errorf("expected file == file2")
	}
	fref.m.Unlock()

	fref.DecRef()

	m.Lock()
	if x != 0 {
		t.Errorf("expected x 0")
	}
	if after != 0 {
		t.Errorf("expected after 0")
	}
	m.Unlock()

	fref.m.Lock()
	if fref.refs != 1 {
		t.Errorf("expected 1 refs")
	}
	if fref.file != file2 {
		t.Errorf("expected file == file2")
	}
	fref.m.Unlock()

	fref.DecRef()

	m.Lock()
	if x != 1 {