Added unit test on Closing BaseModule.

dipdup-io · Sep 11, 2023 · f49f741 · f49f741
1 parent b3594bd
commit f49f741
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/pkg/modules/baseModule_test.go b/pkg/modules/baseModule_test.go
@@ -89,3 +89,21 @@ func TestBaseModule_AttachToOnExistingChannel(t *testing.T) {
 	_, ok = <-input.Listen() // TODO-DISCUSS
 	assert.False(t, ok)
 }
+
+func TestBaseModule_CloseSuccessfullyCloseInputChannels(t *testing.T) {
+	//bmSrc := &BaseModule{Outputs: sync.NewMap[string, *Output]()}
+	bmDst := &BaseModule{Inputs: sync.NewMap[string, *Input]()}
+	channelName := "data"
+
+	//bmSrc.Outputs.Set(channelName, NewOutput(channelName))
+	bmDst.Inputs.Set(channelName, NewInput(channelName))
+
+	input, err := bmDst.Input(channelName)
+	assert.NoError(t, err)
+
+	err = bmDst.Close()
+	assert.NoError(t, err)
+
+	_, ok := <-input.Listen() // TODO-DISCUSS
+	assert.False(t, ok)
+}