test dual kernel

fogfish · Sep 28, 2024 · ed446a3 · ed446a3
1 parent ce1e22e
commit ed446a3
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 7 deletions.
diff --git a/kernel/cathode_test.go b/kernel/cathode_test.go
@@ -23,6 +23,15 @@ func TestDequeuer(t *testing.T) {
 		[]swarm.Bag{{Ctx: &swarm.Context{Category: "test", Digest: "1"}, Object: []byte(`"1"`)}},
 	)
 
+	t.Run("Kernel", func(t *testing.T) {
+		k := New(nil, NewDequeuer(mockCathode(nil, nil), swarm.Config{}))
+		go func() {
+			time.Sleep(yield_before_close)
+			k.Close()
+		}()
+		k.Await()
+	})
+
 	t.Run("None", func(t *testing.T) {
 		k := NewDequeuer(none, swarm.Config{PollFrequency: 1 * time.Second})
 		go k.Await()

diff --git a/kernel/emitter_test.go b/kernel/emitter_test.go
@@ -19,9 +19,6 @@ import (
 )
 
 func TestEnqueuer(t *testing.T) {
-	// none := mockEmitter(0, nil)
-	// pass := mockEmitter(10, make(chan []byte))
-
 	mockit := func(config swarm.Config) (*Enqueuer, *emitter) {
 		mock := mockEmitter(10)
 		k := NewEnqueuer(mock, config)
@@ -34,6 +31,15 @@ func TestEnqueuer(t *testing.T) {
 		return k, mock
 	}
 
+	t.Run("Kernel", func(t *testing.T) {
+		k := New(NewEnqueuer(mockEmitter(10), swarm.Config{}), nil)
+		go func() {
+			time.Sleep(yield_before_close)
+			k.Close()
+		}()
+		k.Await()
+	})
+
 	t.Run("None", func(t *testing.T) {
 		k, _ := mockit(swarm.Config{})
 		Enqueue(k, "test", swarm.NewCodecJson[string]())

diff --git a/kernel/kernel.go b/kernel/kernel.go
@@ -21,11 +21,21 @@ func New(enqueuer *Enqueuer, dequeuer *Dequeuer) *Kernel {
 }
 
 func (k *Kernel) Close() {
-	k.Dequeuer.Close()
-	k.Enqueuer.Close()
+	if k.Dequeuer != nil {
+		k.Dequeuer.Close()
+	}
+
+	if k.Enqueuer != nil {
+		k.Enqueuer.Close()
+	}
 }
 
 func (k *Kernel) Await() {
-	k.Dequeuer.Await()
-	k.Enqueuer.Await()
+	if k.Dequeuer != nil {
+		k.Dequeuer.Await()
+	}
+
+	if k.Enqueuer != nil {
+		k.Enqueuer.Await()
+	}
 }