RAI-8289 add suspend/resume functionality to julia sdk (#94)

* Add suspend/resume functionality.

* add integration test to suspend/resume functionality

Author: janrous-rai

src/RAI.jl

@@ -43,7 +43,9 @@ export
     create_engine,
     delete_engine,
     get_engine,
-    list_engines
+    list_engines,
+    suspend_engine,
+    resume_engine
 
 export
     delete_models,

src/api.jl

@@ -207,7 +207,11 @@ function _request(ctx::Context, method, path; query = nothing, body = UInt8[], k
     # _print_request(method, path, query, body);
     try
         rsp = @mock request(ctx, method, _mkurl(ctx, path); query = query, body = body, kw...)
-        return JSON3.read(rsp.body)
+        if length(rsp.body) == 0
+            return Dict()
+        else
+            return JSON3.read(rsp.body)
+        end
     catch e
         if e isa HTTP.ExceptionRequest.StatusError
             throw(HTTPError(e.status, String(e.response.body)))
@@ -238,6 +242,16 @@ function create_engine(ctx::Context, engine::AbstractString; size = nothing, kw.
     return _put(ctx, PATH_ENGINE; body = JSON3.write(data), kw...)
 end
 
+function suspend_engine(ctx::Context, engine::AbstractString; kw...)
+    payload=Dict("suspend" => true)
+    return _patch(ctx, "$PATH_ENGINE/$engine"; body=JSON3.write(payload), kw...)
+end
+
+function resume_engine(ctx::Context, engine::AbstractString; kw...)
+    payload=Dict("suspend" => false)
+    return _patch(ctx, "$PATH_ENGINE/$engine"; body=JSON3.write(payload), kw...)
+end
+
 function create_oauth_client(ctx::Context, name::AbstractString, permissions; kv...)
     isnothing(permissions) && (permissions = [])
     data = Dict("name" => name, "permissions" => permissions)

test/integration.jl

@@ -114,6 +114,25 @@ const CTX = test_context()
 # engine
 @testset "engine" begin end
 
+# -----------------------------------
+# suspend and resume
+with_engine(CTX) do engine_name
+    @testset "suspend" begin
+        suspend_engine(CTX, engine_name)
+        start_time = time()
+        _poll_with_specified_overhead(; POLLING_KWARGS..., start_time) do
+            eng = get_engine(CTX, engine_name)
+            return eng[:state] == "SUSPENDED" && eng[:suspend] == true
+        end
+        resume_engine(CTX, engine_name)
+        start_time = time()
+        _poll_with_specified_overhead(; POLLING_KWARGS..., start_time) do
+            eng = get_engine(CTX, engine_name)
+            return eng[:state] == "PROVISIONED" && eng[:suspend] == false
+        end
+    end
+end
+
 with_engine(CTX) do engine_name
     # -----------------------------------
     # database