File: file_test.go

package info (click to toggle)
golang-github-couchbase-moss 0.0~git20170914.0.07c86e8-4
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid
  • size: 664 kB
  • sloc: python: 230; makefile: 37
file content (109 lines) | stat: -rw-r--r-- 2,074 bytes parent folder | download
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
//  Copyright (c) 2016 Couchbase, 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 moss

import (
	"io/ioutil"
	"os"
	"path"
	"sync"
	"testing"
)

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 {
		t.Errorf("expected x 1")
	}
	if after != 1 {
		t.Errorf("expected after 1")
	}
	m.Unlock()

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

	err := file.Close()
	if err == nil {
		t.Errorf("expected re-close to err")
	}

	fref = nil
	if fref.DecRef() != nil {
		t.Errorf("expected DecRef on nil to nil")
	}

	if ToOsFile(file) != file {
		t.Errorf("expected ToOsFile(file) == file")
	}
}