Adds unit test for completeStreamSession (#1842)

CHANGELOG_PENDING.md

@@ -10,7 +10,7 @@
 
 #### Orchestrator
 
-- \#1840 Re-use remote transcoders for a stream sessions (@reubenr0d)
+- \#1842 Re-use remote transcoders for a stream sessions (@reubenr0d)
 
 #### Transcoder
 

core/orch_test.go

@@ -354,6 +354,28 @@ func TestSelectTranscoder(t *testing.T) {
 	assert.Equal(0, t1.load)
 }
 
+func TestCompleteStreamSession(t *testing.T) {
+	m := NewRemoteTranscoderManager()
+	strm := &StubTranscoderServer{manager: m, WithholdResults: false}
+	testSessionId := "testID"
+	assert := assert.New(t)
+
+	// register transcoders
+	go func() { m.Manage(strm, 1) }()
+	time.Sleep(1 * time.Millisecond) // allow time for first stream to register
+	t1 := m.liveTranscoders[strm]
+
+	// selectTranscoder and assert that session is added
+	m.selectTranscoder(testSessionId)
+	assert.Equal(t1, m.streamSessions[testSessionId])
+	assert.Equal(1, t1.load)
+
+	// complete session and assert that it is cleared
+	m.completeStreamSession(testSessionId)
+	assert.Nil(m.streamSessions[testSessionId])
+	assert.Equal(0, t1.load)
+}
+
 func TestTranscoderManagerTranscoding(t *testing.T) {
 	m := NewRemoteTranscoderManager()
 	s := &StubTranscoderServer{manager: m}