opensource/cocos
1
0
Fork 0
mirror of https://github.com/ultravioletrs/cocos.git synced 2026-06-23 04:10:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

COCOS-242-Agent tests fail ocassionally due to a missing mock (#251)

Browse Source 
* Clean up files after algo run

Signed-off-by: Jilks Smith <smithjilks@gmail.com>

* Add test cleanup

Signed-off-by: Jilks Smith <smithjilks@gmail.com>

---------

Signed-off-by: Jilks Smith <smithjilks@gmail.com>
This commit is contained in:
Smith Jilks
2024-09-24 15:44:27 +03:00
committed by GitHub
parent 5d5ae35e2b
commit af3817d3b7
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agent/service_test.go
+15 -5
View File
@@ -126,6 +126,11 @@ func TestAlgo(t *testing.T) {
err = svc.Algo(ctx, tc.algo)
assert.True(t, errors.Contains(err, tc.err), "expected %v, got %v", tc.err, err)
t.Cleanup(func() {
err = os.RemoveAll("venv")
err = os.RemoveAll("algo")
err = os.RemoveAll("datasets")
})
})
}
}
@@ -219,9 +224,13 @@ func TestData(t *testing.T) {
time.Sleep(300 * time.Millisecond)
}
err = svc.Data(ctx, tc.data)
_ = os.RemoveAll("datasets")
_ = os.RemoveAll("results")
assert.True(t, errors.Contains(err, tc.err), "expected %v, got %v", tc.err, err)
t.Cleanup(func() {
_ = os.RemoveAll("datasets")
_ = os.RemoveAll("results")
err = os.RemoveAll("venv")
err = os.RemoveAll("algo")
})
})
}
}
@@ -302,9 +311,10 @@ func TestResult(t *testing.T) {
go svc.sm.Start(ctx)
tc.setup(svc)
_, err := svc.Result(ctx)
_ = os.RemoveAll("datasets")
_ = os.RemoveAll("results")
t.Cleanup(func() {
_ = os.RemoveAll("datasets")
_ = os.RemoveAll("results")
})
assert.ErrorIs(t, err, tc.err, "expected %v, got %v", tc.err, err)
})
}