From 436fc20fdb8c6884678136d045f99cf4553281fc Mon Sep 17 00:00:00 2001
From: David Stainton <dstainton@elixxir.io>
Date: Tue, 23 Aug 2022 16:41:18 -0400
Subject: [PATCH] WIP

---
 ud/channelIDTracking.go      | 7 ++++---
 ud/channelIDTracking_test.go | 9 ++++++---
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/ud/channelIDTracking.go b/ud/channelIDTracking.go
index 113c4a149..a1546acc2 100644
--- a/ud/channelIDTracking.go
+++ b/ud/channelIDTracking.go
@@ -203,10 +203,12 @@ func (c *clientIDTracker) Start() (stoppable.Stoppable, error) {
 }
 
 func (c *clientIDTracker) registrationWorker(stopper *stoppable.Single) {
-
 	for {
 		if time.Now().After(c.registrationDisk.GetLease().Add(-graceDuration)) {
-			c.register()
+			err := c.register()
+			if err != nil {
+				jww.FATAL.Panic(err)
+			}
 		}
 
 		select {
@@ -221,7 +223,6 @@ func (c *clientIDTracker) registrationWorker(stopper *stoppable.Single) {
 			return
 		case <-time.After(time.Second):
 		}
-
 	}
 }
 
diff --git a/ud/channelIDTracking_test.go b/ud/channelIDTracking_test.go
index 5e8c5e11d..c8b516e00 100644
--- a/ud/channelIDTracking_test.go
+++ b/ud/channelIDTracking_test.go
@@ -71,7 +71,10 @@ func TestChannelIDTracking(t *testing.T) {
 	myTestClientIDTracker := newclientIDTracker(comms, host, username,
 		kv, m.user.GetReceptionIdentity(), ed25519.PublicKey(udPubKeyBytes), rngGen)
 
-	stopper, err := myTestClientIDTracker.Start()
+	//stopper, err := myTestClientIDTracker.Start()
+	//require.NoError(t, err)
+
+	err = myTestClientIDTracker.register()
 	require.NoError(t, err)
 
 	require.Equal(t, myTestClientIDTracker.GetUsername(), username)
@@ -90,6 +93,6 @@ func TestChannelIDTracking(t *testing.T) {
 
 	//_ = myTestClientIDTracker.ValidateChannelMessage(username, lease, pubKey, authorIDSignature)
 
-	err = stopper.Close()
-	require.NoError(t, err)
+	//err = stopper.Close()
+	//require.NoError(t, err)
 }
-- 
GitLab