//  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 (
	"fmt"
	"io/ioutil"
	"os"
	"sync"
	"testing"
	"time"

	"github.com/edsrzf/mmap-go"
)

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

	f, err := os.Create(tmpDir + string(os.PathSeparator) + "test.file")
	if err != nil {
		t.Errorf("expected open file to work, err: %v", err)
	}

	offset := 1024 * 1024 * 1024 // 1 GB.

	f.WriteAt([]byte("hello"), int64(offset))

	var mms []mmap.MMap

	for i := 0; i < 100; i++ { // Re-mmap the file.
		mm, err := mmap.Map(f, mmap.RDONLY, 0)
		if err != nil {
			t.Errorf("expected mmap to work, err: %v", err)
		}

		if string(mm[offset:offset+5]) != "hello" {
			t.Errorf("expected hello")
		}

		mms = append(mms, mm)
	}

	for _, mm := range mms {
		if string(mm[offset:offset+5]) != "hello" {
			t.Errorf("expected hello")
		}

		for j := 0; j < offset; j += 1024 * 1024 {
			if mm[j] != 0 {
				t.Errorf("expected 0")
			}
		}
	}

	for _, mm := range mms {
		mm.Unmap()
	}

	f.Close()
}

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

	var mu sync.Mutex
	counts := map[EventKind]int{}
	eventWaiters := map[EventKind]chan bool{}

	co := CollectionOptions{
		MergeOperator: &MergeOperatorStringAppend{Sep: ":"},
		OnEvent: func(event Event) {
			mu.Lock()
			counts[event.Kind]++
			eventWaiter := eventWaiters[event.Kind]
			mu.Unlock()
			if eventWaiter != nil {
				eventWaiter <- true
			}
		},
	}

	store, m, err := OpenStoreCollection(tmpDir, StoreOptions{
		CollectionOptions: co,
	}, StorePersistOptions{})
	if err != nil || m == nil || store == nil {
		t.Errorf("expected open empty store collection to work")
	}

	b, _ := m.NewBatch(0, 0)
	for i := 0; i < 1000; i++ {
		xs := fmt.Sprintf("%d", i)
		x := []byte(xs)
		b.Set(x, x)
	}
	err = m.ExecuteBatch(b, WriteOptions{})
	if err != nil {
		t.Errorf("expected exec batch to work")
	}
	b.Close()

	waitUntilClean := func() error {
		for {
			stats, err := m.Stats()
			if err != nil {
				return err
			}

			if stats.CurDirtyOps <= 0 &&
				stats.CurDirtyBytes <= 0 &&
				stats.CurDirtySegments <= 0 {
				break
			}

			time.Sleep(200 * time.Millisecond)
		}

		return nil
	}

	waitUntilClean()

	store.m.Lock()

	if store.refs != 1 {
		t.Errorf("expected 1 store ref, got: %d", store.refs)
	}

	footer := store.footer
	if footer == nil {
		t.Errorf("expected footer")
	}

	footer.m.Lock()
	if footer.refs != 2 {
		t.Errorf("expected 2 footer ref, : got: %d", footer.refs)
	}

	mref := footer.SegmentLocs[0].mref
	if mref == nil {
		t.Errorf("expected mref")
	}

	mrefsCheck := func(expected int) {
		mref.m.Lock()
		if mref.refs != expected {
			t.Errorf("expected mref.refs to be %d, got: %d", expected, mref.refs)
		}
		mref.m.Unlock()
	}

	mrefsCheck(1)

	rv := mref.AddRef()

	mrefsCheck(2)

	if rv != mref {
		t.Errorf("expected rv == mref")
	}

	if mref.DecRef() != nil {
		t.Errorf("expected mref.DecRef to be nil")
	}

	mrefsCheck(1)

	footer.m.Unlock()

	store.m.Unlock()

	m.Close()

	store.Close()

	footer.m.Lock()
	if footer.refs != 0 {
		t.Errorf("expected footer refs to be 0, got: %d", footer.refs)
	}
	footer.m.Unlock()

	store.m.Lock()
	if store.refs != 0 {
		t.Errorf("expected store refs to be 0, got: %d", footer.refs)
	}
	store.m.Unlock()

	mrefsCheck(0)
}

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

	var mu sync.Mutex
	counts := map[EventKind]int{}
	eventWaiters := map[EventKind]chan bool{}

	co := CollectionOptions{
		OnEvent: func(event Event) {
			mu.Lock()
			counts[event.Kind]++
			eventWaiter := eventWaiters[event.Kind]
			mu.Unlock()
			if eventWaiter != nil {
				eventWaiter <- true
			}
		},
	}

	store, m, err := OpenStoreCollection(tmpDir, StoreOptions{
		CollectionOptions:    co,
		CompactionPercentage: 10000.0,
	}, StorePersistOptions{
		CompactionConcern: CompactionAllow,
	})
	if err != nil || m == nil || store == nil {
		t.Errorf("expected open empty store collection to work")
	}

	// ---------------------------------------------

	checkRefs := func(f *Footer, frefs, mrefs int, cb func(), msg string) {
		f.m.Lock()

		if f.refs != frefs {
			t.Errorf("%s - expected footer.refs to be %d, got: %d",
				msg, frefs, f.refs)
		}

		n := len(f.SegmentLocs)
		if n > 0 &&
			f.SegmentLocs[n-1].mref != nil {
			f.SegmentLocs[n-1].mref.m.Lock()

			if f.SegmentLocs[n-1].mref.refs != mrefs {
				t.Errorf("%s - expected mrefs to be %d, got: %d",
					msg, mrefs, f.SegmentLocs[n-1].mref.refs)
			}

			if cb != nil {
				cb()
			}

			f.SegmentLocs[n-1].mref.m.Unlock()
		} else if mrefs > 0 {
			t.Errorf("%s - expected footer.mref to be %d, but nil mref",
				msg, mrefs)
		}

		f.m.Unlock()
	}

	writeHello := func() {
		b, _ := m.NewBatch(0, 0)
		b.Set([]byte("hello"), []byte("world"))
		err = m.ExecuteBatch(b, WriteOptions{})
		if err != nil {
			t.Errorf("expected exec batch to work")
		}
		b.Close()
	}

	waitUntilClean := func() error {
		for {
			stats, err := m.Stats()
			if err != nil {
				return err
			}

			if stats.CurDirtyOps <= 0 &&
				stats.CurDirtyBytes <= 0 &&
				stats.CurDirtySegments <= 0 {
				break
			}

			time.Sleep(time.Millisecond)
		}

		return nil
	}

	// ---------------------------------------------

	store.m.Lock()

	checkRefs(store.footer, 2, 0, nil, "new, empty store")

	store.m.Unlock()

	// ---------------------------------------------

	writeHello()

	waitUntilClean()

	// ---------------------------------------------

	store.m.Lock()

	checkRefs(store.footer, 2, 1, nil, "after 1st batch persisted")

	store.m.Unlock()

	// ---------------------------------------------

	ss0, err := store.Snapshot()
	if err != nil {
		t.Errorf("expected no err on snapshot")
	}

	f0, ok := ss0.(*Footer)
	if !ok {
		t.Errorf("expected Footer")
	}

	store.m.Lock()

	checkRefs(store.footer, 3, 1, nil, "after 1st batch persisted")

	checkRefs(f0, 3, 1, nil, "after 1st batch persisted, against f0")

	store.m.Unlock()

	for i := 0; i < 10; i++ {
		writeHello()
		waitUntilClean()
	}

	store.m.Lock()

	if f0 == store.footer {
		t.Errorf("expected curr footer to be != f0 after many mutations")
	}

	checkRefs(store.footer, 2, 1, nil, "after nth batch persisted")

	store.m.Unlock()

	var mref *mmapRef

	checkRefs(f0, 1, 2, func() {
		mref = f0.SegmentLocs[0].mref
	}, "oldest footer check")

	// ----------------------------------------

	ss0.Close() // Close the first, oldest snapshot.

	store.m.Lock()

	checkRefs(store.footer, 2, 1, nil, "after oldest snapshot closed")

	store.m.Unlock()

	checkRefs(f0, 0, 0, nil, "oldest footer after ss0.Close()")

	mref.m.Lock()
	if mref.refs != 1 {
		t.Errorf("expected mref.refs 1 after oldest ss closed, got: %d", mref.refs)
	}
	mref.m.Unlock()

	// ----------------------------------------

	m.Close()

	var fLast *Footer

	store.m.Lock()
	fLast = store.footer
	checkRefs(store.footer, 1, 1, nil, "after collection Close()'ed")
	store.m.Unlock()

	mref.m.Lock()
	if mref.refs != 1 {
		t.Errorf("expected mref.refs 1 after coll closed, got: %d", mref.refs)
	}
	mref.m.Unlock()

	// ----------------------------------------

	store.Close()

	mref.m.Lock()
	if mref.refs != 0 {
		t.Errorf("expected 0 mref.refs after everything closed")
	}
	mref.m.Unlock()

	checkRefs(fLast, 0, 0, nil, "last footer after store.Close()")

	checkRefs(f0, 0, 0, nil, "oldest footer after store.Close()")
}

// ---------------------------------------------

// SKIPPED because segfault isn't caught by recover()
func SKIPPED_TestAccessAfterUnmap(t *testing.T) {
	tmpDir, _ := ioutil.TempDir("", "mossMMap")
	defer os.RemoveAll(tmpDir)

	f, err := os.Create(tmpDir + string(os.PathSeparator) + "test.file")
	if err != nil {
		t.Errorf("expected open file to work, err: %v", err)
	}

	defer f.Close()

	offset := 1024 * 1024 * 1024 // 1 GB.

	f.WriteAt([]byte("hello"), int64(offset))

	var mm mmap.MMap

	mm, err = mmap.Map(f, mmap.RDONLY, 0)
	if err != nil {
		t.Errorf("expected mmap to work, err: %v", err)
	}

	x := mm[offset : offset+5]

	if string(x) != "hello" {
		t.Errorf("expected hello")
	}

	defer func() {
		if r := recover(); r != nil {
			fmt.Println("Recovered in f", r)
		} else {
			t.Errorf("expected recover from panic")
		}
	}()

	mm.Unmap()

	/*
			The following access of x results in a segfault, like...

				unexpected fault address 0x4060c000
				fatal error: fault
				[signal 0xb code=0x1 addr=0x4060c000 pc=0xb193f]

		    The recover() machinery doesn't handle this situation, however,
		    as it's not a normal kind of panic()
	*/
	if x[0] != 'h' {
		t.Errorf("expected h, but actually expected a segfault")
	}

	t.Errorf("expected segfault, but instead unmmapped mem access worked")
}
