95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you 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 casbin
|
|
|
|
import "testing"
|
|
|
|
type SampleWatcher struct {
|
|
callback func(string)
|
|
}
|
|
|
|
func (w *SampleWatcher) Close() {
|
|
}
|
|
|
|
func (w *SampleWatcher) SetUpdateCallback(callback func(string)) error {
|
|
w.callback = callback
|
|
return nil
|
|
}
|
|
|
|
func (w *SampleWatcher) Update() error {
|
|
if w.callback != nil {
|
|
w.callback("")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func TestSetWatcher(t *testing.T) {
|
|
e, err := NewEnforcer("examples/rbac_model.conf", "examples/rbac_policy.csv")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sampleWatcher := &SampleWatcher{}
|
|
err = e.SetWatcher(sampleWatcher)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = e.SavePolicy() // calls watcher.Update()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestSelfModify(t *testing.T) {
|
|
e, err := NewEnforcer("examples/rbac_model.conf", "examples/rbac_policy.csv")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
sampleWatcher := &SampleWatcher{}
|
|
err = e.SetWatcher(sampleWatcher)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var called int
|
|
|
|
called = -1
|
|
_ = e.watcher.SetUpdateCallback(func(s string) {
|
|
called = 1
|
|
})
|
|
_, err = e.AddPolicy("eva", "data", "read") // calls watcher.Update()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if called != 1 {
|
|
t.Fatal("callback should be called")
|
|
}
|
|
|
|
called = -1
|
|
_ = e.watcher.SetUpdateCallback(func(s string) {
|
|
called = 1
|
|
})
|
|
_, err = e.SelfAddPolicy("p", "p", []string{"eva", "data", "write"}) // calls watcher.Update()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if called != -1 {
|
|
t.Fatal("callback should not be called")
|
|
}
|
|
}
|