add UT

projecteru2 · Mar 8, 2024 · 41258e2 · 41258e2
1 parent d955791
commit 41258e2
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/cluster/calcium/raw_engine_test.go b/cluster/calcium/raw_engine_test.go
@@ -32,3 +32,19 @@ func TestRawEngine(t *testing.T) {
 	_, err := c.RawEngine(ctx, &types.RawEngineOptions{ID: "id1", Op: "xxxx"})
 	assert.NoError(t, err)
 }
+
+func TestRawEngineIgnoreLock(t *testing.T) {
+	c := NewTestCluster()
+	ctx := context.Background()
+	store := c.store.(*storemocks.Store)
+	workload := &types.Workload{
+		ID:         "id1",
+		Privileged: true,
+	}
+	engine := &enginemocks.API{}
+	workload.Engine = engine
+	store.On("GetWorkload", mock.Anything, mock.Anything).Return(workload, nil)
+	engine.On("RawEngine", mock.Anything, mock.Anything).Return(&enginetypes.RawEngineResult{}, nil).Once()
+	_, err := c.RawEngine(ctx, &types.RawEngineOptions{ID: "id1", Op: "xxxx", IgnoreLock: true})
+	assert.NoError(t, err)
+}