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