Function TestServiceMap

lib/model/service_map_test.go:18–138

(t *testing.T)

)

func TestServiceMap(t *testing.T) {
	t.Parallel()
	ctx, cancel := context.WithCancel(context.Background())
	t.Cleanup(cancel)
	sup := suture.NewSimple("TestServiceMap")
	sup.ServeBackground(ctx)

	t.Run("SimpleAddRemove", func(t *testing.T) {
		t.Parallel()

		sm := newServiceMap[string, *dummyService](events.NoopLogger)
		sup.Add(sm)

		// Add two services. They should start.

		d1 := newDummyService()
		d2 := newDummyService()

		sm.Add("d1", d1)
		sm.Add("d2", d2)

		<-d1.started
		<-d2.started

		// Remove them. They should stop.

		if !sm.Remove("d1") {
			t.Errorf("Remove failed")
		}
		if !sm.Remove("d2") {
			t.Errorf("Remove failed")
		}

		<-d1.stopped
		<-d2.stopped
	})

	t.Run("OverwriteImpliesRemove", func(t *testing.T) {
		t.Parallel()

		sm := newServiceMap[string, *dummyService](events.NoopLogger)
		sup.Add(sm)

		d1 := newDummyService()
		d2 := newDummyService()

		// Add d1, it should start.

		sm.Add("k", d1)
		<-d1.started

		// Add d2, with the same key. The previous one should stop as we're
		// doing a replace.

		sm.Add("k", d2)
		<-d1.stopped
		<-d2.started