gf/g/os/gfsnotify/gfsnotify_z_unit_test.go

// Copyright 2019 gf Author(https://github.com/gogf/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.

// go test *.go -bench=".*" -benchmem

package gfsnotify_test

import (
    "github.com/gogf/gf/g/container/gtype"
    "github.com/gogf/gf/g/os/gfile"
    "github.com/gogf/gf/g/os/gfsnotify"
    "github.com/gogf/gf/g/os/gtime"
    "github.com/gogf/gf/g/test/gtest"
    "github.com/gogf/gf/g/util/gconv"
    "testing"
    "time"
)

func TestWatcher_AddRemove(t *testing.T) {
    gtest.Case(t, func() {
        path1 := gconv.String(gtime.Nanosecond())
        path2 := gconv.String(gtime.Nanosecond()) + "2"
        gfile.PutContents(path1, "1")
        defer func() {
            gfile.Remove(path1)
            gfile.Remove(path2)
        }()
        v := gtype.NewInt(1)
        callback, err := gfsnotify.Add(path1, func(event *gfsnotify.Event) {
            if event.IsWrite() {
                v.Set(2)
                return
            }
            if event.IsRename() {
                v.Set(3)
                gfsnotify.Exit()
                return
            }
        })
        gtest.Assert(err,        nil)
        gtest.AssertNE(callback, nil)

        gfile.PutContents(path1, "2")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 2)

        gfile.Rename(path1, path2)
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 3)
    })

    gtest.Case(t, func() {
        path1 := gconv.String(gtime.Nanosecond())
        gfile.PutContents(path1, "1")
        defer func() {
            gfile.Remove(path1)
        }()
        v := gtype.NewInt(1)
        callback, err := gfsnotify.Add(path1, func(event *gfsnotify.Event) {
            if event.IsWrite() {
                v.Set(2)
                return
            }
            if event.IsRemove() {
                v.Set(4)
                return
            }
        })
        gtest.Assert(err,        nil)
        gtest.AssertNE(callback, nil)

        gfile.PutContents(path1, "2")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 2)

        gfile.Remove(path1)
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 4)

        gfile.PutContents(path1, "1")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 4)
    })
}

func TestWatcher_Callback(t *testing.T) {
    gtest.Case(t, func() {
        path1 := gconv.String(gtime.Nanosecond())
        gfile.PutContents(path1, "1")
        defer func() {
            gfile.Remove(path1)
        }()
        v := gtype.NewInt(1)
        callback, err := gfsnotify.Add(path1, func(event *gfsnotify.Event) {
            if event.IsWrite() {
                v.Set(2)
                return
            }
        })
        gtest.Assert(err,        nil)
        gtest.AssertNE(callback, nil)

        gfile.PutContents(path1, "2")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 2)

        v.Set(3)
        gfsnotify.RemoveCallback(callback.Id)
        gfile.PutContents(path1, "3")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v.Val(), 3)
    })
    // multiple callbacks
    gtest.Case(t, func() {
        path1 := gconv.String(gtime.Nanosecond())
        gfile.PutContents(path1, "1")
        defer func() {
            gfile.Remove(path1)
        }()
        v1 := gtype.NewInt(1)
        v2 := gtype.NewInt(1)
        callback1, err1 := gfsnotify.Add(path1, func(event *gfsnotify.Event) {
            if event.IsWrite() {
                v1.Set(2)
                return
            }
        })
        callback2, err2 := gfsnotify.Add(path1, func(event *gfsnotify.Event) {
            if event.IsWrite() {
                v2.Set(2)
                return
            }
        })
        gtest.Assert(err1,        nil)
        gtest.Assert(err2,        nil)
        gtest.AssertNE(callback1, nil)
        gtest.AssertNE(callback2, nil)

        gfile.PutContents(path1, "2")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v1.Val(), 2)
        gtest.Assert(v2.Val(), 2)

        v1.Set(3)
        v2.Set(3)
        gfsnotify.RemoveCallback(callback1.Id)
        gfile.PutContents(path1, "3")
        time.Sleep(100*time.Millisecond)
        gtest.Assert(v1.Val(), 3)
        gtest.Assert(v2.Val(), 2)
    })
}