func (s *S) TestIsResidentTwoPages(c *C) {
testPath := path.Join(c.MkDir(), "test.txt")
file, err := os.Create(testPath)
c.Assert(err, IsNil)
defer file.Close()
file.Seek(int64(os.Getpagesize()*2-1), 0)
file.Write([]byte{'x'})
mmap, err := gommap.Map(file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_PRIVATE)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
// Not entirely a stable test, but should usually work.
mmap[len(mmap)-1] = 'x'
mapped, err := mmap.IsResident()
c.Assert(err, IsNil)
c.Assert(mapped, DeepEquals, []bool{false, true})
mmap[0] = 'x'
mapped, err = mmap.IsResident()
c.Assert(err, IsNil)
c.Assert(mapped, DeepEquals, []bool{true, true})
}
func (s *S) TestIsResidentUnderOnePage(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_PRIVATE)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
mapped, err := mmap.IsResident()
c.Assert(err, IsNil)
c.Assert(mapped, DeepEquals, []bool{true})
}
func (s *S) TestProtFlagsAndErr(c *C) {
testPath := s.file.Name()
s.file.Close()
file, err := os.Open(testPath)
c.Assert(err, IsNil)
s.file = file
_, err = gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_SHARED)
// For this to happen, both the error and the protection flag must work.
c.Assert(err, Equals, syscall.EACCES)
}
func (s *S) TestAdvise(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_PRIVATE)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
// A bit tricky to blackbox-test these.
err = mmap.Advise(gommap.MADV_RANDOM)
c.Assert(err, IsNil)
err = mmap.Advise(9999)
c.Assert(err, ErrorMatches, "invalid argument")
}
func (s *S) TestProtect(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ, gommap.MAP_SHARED)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
c.Assert([]byte(mmap), DeepEquals, testData)
err = mmap.Protect(gommap.PROT_READ | gommap.PROT_WRITE)
c.Assert(err, IsNil)
// If this operation doesn't blow up tests, the call above worked.
mmap[9] = 'X'
}
func (s *S) TestFlags(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_PRIVATE)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
mmap[9] = 'X'
mmap.Sync(gommap.MS_SYNC)
fileData, err := ioutil.ReadFile(s.file.Name())
c.Assert(err, IsNil)
// Shouldn't have written, since the map is private.
c.Assert(fileData, DeepEquals, []byte("0123456789ABCDEF"))
}
func (s *S) TestSliceMethods(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_SHARED)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
c.Assert([]byte(mmap), DeepEquals, testData)
mmap[9] = 'X'
mmap[7:10].Sync(gommap.MS_SYNC)
fileData, err := ioutil.ReadFile(s.file.Name())
c.Assert(err, IsNil)
c.Assert(fileData, DeepEquals, []byte("012345678XABCDEF"))
}
func (s *S) TestLock(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_PRIVATE)
c.Assert(err, IsNil)
defer mmap.UnsafeUnmap()
// A bit tricky to blackbox-test these.
err = mmap.Lock()
c.Assert(err, IsNil)
err = mmap.Lock()
c.Assert(err, IsNil)
err = mmap.Unlock()
c.Assert(err, IsNil)
err = mmap.Unlock()
c.Assert(err, IsNil)
}
func (s *S) TestUnsafeUnmap(c *C) {
mmap, err := gommap.Map(s.file.Fd(), gommap.PROT_READ|gommap.PROT_WRITE, gommap.MAP_SHARED)
c.Assert(err, IsNil)
c.Assert(mmap.UnsafeUnmap(), IsNil)
}