| /* |
| * 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 cache |
| |
| import ( |
| "context" |
| "fmt" |
| "testing" |
| "time" |
| ) |
| |
| type level1 struct { |
| } |
| |
| func (l *level1) Name(ctx context.Context, _ *Node) string { |
| return ctx.Value("key1").(string) |
| } |
| |
| func (l *level1) Init(ctx context.Context, parent *Node) (node *Node, err error) { |
| p := l.Name(ctx, parent) |
| if p == "err" { |
| return nil, fmt.Errorf("wrong logic") |
| } |
| |
| if p == "null" { |
| return |
| } |
| |
| node = NewNode() |
| node.Cache.Set("a", "a") |
| node.Cache.Set("b", "b") |
| node.Cache.Set("c", "c") |
| return |
| } |
| |
| type level2 struct { |
| changed string |
| } |
| |
| func (l *level2) Name(ctx context.Context, _ *Node) string { |
| return ctx.Value("key2").(string) |
| } |
| |
| func (l *level2) Init(ctx context.Context, parent *Node) (node *Node, err error) { |
| if parent == nil { |
| return |
| } |
| |
| p := l.Name(ctx, parent) |
| if p == "err" { |
| return nil, fmt.Errorf("wrong logic") |
| } |
| |
| if p == "null" { |
| return |
| } |
| |
| node = NewNode() |
| if len(l.changed) == 0 { |
| i := parent.Cache.Get(p) |
| node.Cache.Set("aa", i) |
| } else { |
| node.Cache.Set("aa", l.changed) |
| } |
| |
| return |
| } |
| |
| func TestTree_Get(t *testing.T) { |
| tree := NewTree(Configure().WithMaxSize(1).WithTTL(time.Second)) |
| node, err := tree.Get(context.Background()) |
| if node != nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| tree.Remove(context.Background()) |
| |
| l2 := &level2{} |
| tree.AddFilter(&level1{}, l2) |
| |
| ctx := context.WithValue(context.WithValue(context.Background(), "key1", "err"), "key2", "") |
| node, err = tree.Get(ctx) |
| if node != nil || err == nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "null"), "key2", "")) |
| if node != nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "err")) |
| node, err = tree.Get(ctx) |
| if node != nil || err == nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "null")) |
| if node != nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "err")) |
| if node != nil || err == nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "b")) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("aa") != "b" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "a")) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("aa") != "a" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "b")) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("aa") != "b" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| tree.Remove(context.WithValue(ctx, "key1", "1")) |
| |
| l2.changed = "changed" |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "b")) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("aa") != "changed" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| l2.changed = "" |
| node, err = tree.Get(context.WithValue(context.WithValue(ctx, "key1", "1"), "key2", "b"), |
| Options().Temporary(true)) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("aa") != "b" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| node, err = tree.Get(context.WithValue(ctx, "key1", "1"), Options().ToLevel(1)) |
| if node == nil || err != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| if node.Cache.Get("a") != "a" || node.Cache.Get("aa") != nil { |
| t.Fatalf("TestTree_Get failed") |
| } |
| |
| childs := node.ChildNodes() |
| if len(childs) != 1 || childs[0].Cache.Get("aa") != "changed" { |
| t.Fatalf("TestTree_Get failed") |
| } |
| } |