Function TestNode_GarbageCollection

onecache/onecache_test.go:156–198

(t *testing.T)

}

func TestNode_GarbageCollection(t *testing.T) {
	n, err := defaultTestNode()
	if err != nil {
		t.Fatalf("defaultTestNode() failed: %v", err)
	}

	// Inject a key that should be GC'd, shouldn't be and those that are
	// replicating.
	keys := []string{"foo", "bar", "baz"}
	for _, k := range keys {
		if err := n.data.Set(k, []byte(k), 0, 0); err != nil {
			t.Fatalf("Set() failed: %v", err)
		}
	}

	// Mock the replicator such that the first key is being replicated.
	n.replicator = newMockReplicator(func(key string) bool {
		if key == keys[0] {
			return true
		}
		return false
	})

	// Mock the ring such that the second key shouldn't be replicated.
	n.ring = &mockRing{
		replicaPeersImpl: func(key string) []string {
			if key == keys[1] {
				return nil
			}

			return []string{n.name}
		},
	}

	gc, replicating := n.getGCKeys()
	if !replicating {
		t.Fatal("getGCKeys() should indicate that a follow up gc is needed")
	}

	if len(gc) != 1 || gc[0] != keys[2] {
		t.Fatalf("getGCKeys() returned wrong key")
	}
}