Adding authentication to docker file.

Author: rkube

docker-compose.test.yaml

@@ -24,7 +24,7 @@ services:
       "
 
   mlflow:
-    image: ghcr.io/mlflow/mlflow:v2.14.1
+    image: ghcr.io/mlflow/mlflow:v3.6.0
     depends_on:
       - create-buckets
     ports:
@@ -33,11 +33,14 @@ services:
       - AWS_ACCESS_KEY_ID=minioadmin
       - AWS_SECRET_ACCESS_KEY=minioadmin
       - MLFLOW_S3_ENDPOINT_URL=http://minio:9000
+      - MLFLOW_FLASK_SERVER_SECRET_KEY='mlflowclient.jl'
     command: >
       /bin/sh -c "
       pip install boto3 &&
+      pip install mlflow[auth] && 
       mlflow server
       --host 0.0.0.0
       --port 5050
+      --app-name basic-auth
       --default-artifact-root s3://mlflow
       "

test/runtests.jl

@@ -38,4 +38,4 @@ include("services/artifact.jl")
 include("services/experiment.jl")
 include("services/registered_model.jl")
 include("services/model_version.jl")
-#include("services/user.jl")
+include("services/user.jl")

test/services/experiment.jl

@@ -217,54 +217,54 @@ end
 
 # Disabling since /api/2.0/mlflow/users/create is no longer the right api to call.
 # This one is used by createuser
-#@testset verbose = true "create experiment permission" begin
-#    @ensuremlf
-#
-#    experiment_id = createexperiment(mlf, UUIDs.uuid4() |> string)
-#    permission = Permission.parse("READ")
-#
-#    @testset "with string experiment id" begin
-#        user = createuser(mlf, "missy", "gala12345678")
-#        experiment_permission = 
-#            createexperimentpermission(mlf, experiment_id, user.username, permission)
-#
-#        @test experiment_permission isa ExperimentPermission
-#        @test experiment_permission.experiment_id == experiment_id
-#        @test experiment_permission.user_id == user.id
-#        @test experiment_permission.permission == permission
-#        deleteexperimentpermission(mlf, experiment_id, user.username)
-#        deleteuser(mlf, user.username)
-#    end
-#
-#    @testset "with integer experiment id" begin
-#        user = createuser(mlf, "missy", "gala12345678")
-#        experiment_permission = 
-#            createexperimentpermission(mlf, parse(Int, experiment_id), user.username, permission)
-#
-#        @test experiment_permission isa ExperimentPermission
-#        @test experiment_permission.experiment_id == experiment_id
-#        @test experiment_permission.user_id == user.id
-#        @test experiment_permission.permission == permission
-#        deleteexperimentpermission(mlf, experiment_id, user.username)
-#        deleteuser(mlf, user.username)
-#    end
-#
-#    @testset "with Experiment" begin
-#        experiment = getexperiment(mlf, experiment_id)
-#        user = createuser(mlf, "missy", "gala12345678")
-#        experiment_permission = 
-#            createexperimentpermission(mlf, experiment, user.username, permission)
-#
-#        @test experiment_permission isa ExperimentPermission
-#        @test experiment_permission.experiment_id == experiment_id
-#        @test experiment_permission.user_id == user.id
-#        @test experiment_permission.permission == permission
-#        deleteexperimentpermission(mlf, experiment_id, user.username)
-#        deleteuser(mlf, user.username)
-#    end
-#
-#    deleteexperiment(mlf, experiment_id)
-#end
+@testset verbose = true "create experiment permission" begin
+    @ensuremlf
+
+    experiment_id = createexperiment(mlf, UUIDs.uuid4() |> string)
+    permission = Permission.parse("READ")
+
+    @testset "with string experiment id" begin
+        user = createuser(mlf, "missy", "gala12345678")
+        experiment_permission = 
+            createexperimentpermission(mlf, experiment_id, user.username, permission)
+
+        @test experiment_permission isa ExperimentPermission
+        @test experiment_permission.experiment_id == experiment_id
+        @test experiment_permission.user_id == user.id
+        @test experiment_permission.permission == permission
+        deleteexperimentpermission(mlf, experiment_id, user.username)
+        deleteuser(mlf, user.username)
+    end
+
+    @testset "with integer experiment id" begin
+        user = createuser(mlf, "missy", "gala12345678")
+        experiment_permission = 
+            createexperimentpermission(mlf, parse(Int, experiment_id), user.username, permission)
+
+        @test experiment_permission isa ExperimentPermission
+        @test experiment_permission.experiment_id == experiment_id
+        @test experiment_permission.user_id == user.id
+        @test experiment_permission.permission == permission
+        deleteexperimentpermission(mlf, experiment_id, user.username)
+        deleteuser(mlf, user.username)
+    end
+
+    @testset "with Experiment" begin
+        experiment = getexperiment(mlf, experiment_id)
+        user = createuser(mlf, "missy", "gala12345678")
+        experiment_permission = 
+            createexperimentpermission(mlf, experiment, user.username, permission)
+
+        @test experiment_permission isa ExperimentPermission
+        @test experiment_permission.experiment_id == experiment_id
+        @test experiment_permission.user_id == user.id
+        @test experiment_permission.permission == permission
+        deleteexperimentpermission(mlf, experiment_id, user.username)
+        deleteuser(mlf, user.username)
+    end
+
+    deleteexperiment(mlf, experiment_id)
+end
 #
 #@testset verbose = true "get experiment permission" begin
 #    @ensuremlf