From 1858ccdd3579e909adea8b182b982b34467af9cb Mon Sep 17 00:00:00 2001
From: Steven Allen <steven@stebalien.com>
Date: Mon, 23 Sep 2019 17:52:17 -0700
Subject: [PATCH] namesys(test): test TTL on publish

No fix is complete without a test.

fixes #6670
---
 namesys/namesys_test.go | 49 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/namesys/namesys_test.go b/namesys/namesys_test.go
index 4da3b17cb48..524037dcff2 100644
--- a/namesys/namesys_test.go
+++ b/namesys/namesys_test.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"testing"
+	"time"
 
 	ds "github.com/ipfs/go-datastore"
 	dssync "github.com/ipfs/go-datastore/sync"
@@ -117,3 +118,51 @@ func TestPublishWithCache0(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestPublishWithTTL(t *testing.T) {
+	dst := dssync.MutexWrap(ds.NewMapDatastore())
+	priv, _, err := ci.GenerateKeyPair(ci.RSA, 2048)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ps := pstoremem.NewPeerstore()
+	pid, err := peer.IDFromPrivateKey(priv)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = ps.AddPrivKey(pid, priv)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	routing := offroute.NewOfflineRouter(dst, record.NamespacedValidator{
+		"ipns": ipns.Validator{KeyBook: ps},
+		"pk":   record.PublicKeyValidator{},
+	})
+
+	nsys := NewNameSystem(routing, dst, 128)
+	p, err := path.ParsePath(unixfs.EmptyDirNode().Cid().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ttl := 1 * time.Second
+	eol := time.Now().Add(2 * time.Second)
+
+	ctx := context.WithValue(context.Background(), "ipns-publish-ttl", ttl)
+	err = nsys.Publish(ctx, priv, p)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ientry, ok := nsys.(*mpns).cache.Get(pid.Pretty())
+	if !ok {
+		t.Fatal("cache get failed")
+	}
+	entry, ok := ientry.(cacheEntry)
+	if !ok {
+		t.Fatal("bad cache item returned")
+	}
+	if entry.eol.Sub(eol) > 10*time.Millisecond {
+		t.Fatalf("bad cache ttl: expected %s, got %s", eol, entry.eol)
+	}
+}