htwkmooc/poseidon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Introduce context for the codeOceanOutputWriter

Browse Source 
that represents its lifespan.
This commit is contained in:
Maximilian Paß
2023-04-11 19:29:12 +01:00
parent 0c8fa9ccfa
commit 2aa10a130f
3 changed files with 53 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/api/ws/codeocean_writer_test.go
View File

@ -16,7 +16,7 @@ func TestRawToCodeOceanWriter(t *testing.T) {
connectionMock, message := buildConnectionMock(t)
proxyCtx, cancel := context.WithCancel(context.Background())
defer cancel()
output := NewCodeOceanOutputWriter(connectionMock, proxyCtx)
output := NewCodeOceanOutputWriter(connectionMock, proxyCtx, cancel)
<-message // start message
t.Run("StdOut", func(t *testing.T) {
@ -69,10 +69,10 @@ func TestCodeOceanOutputWriter_SendExitInfo(t *testing.T) {
connectionMock, message := buildConnectionMock(t)
proxyCtx, cancel := context.WithCancel(context.Background())
defer cancel()
output := NewCodeOceanOutputWriter(connectionMock, proxyCtx)
output := NewCodeOceanOutputWriter(connectionMock, proxyCtx, cancel)
<-message // start message
output.SendExitInfo(test.info)
output.Close(test.info)
expected, err := json.Marshal(test.message)
require.NoError(t, err)