Skip to content

Commit 7289b79

Browse files
shreyas-goenkashreyas-goenka
committed
add acceptance test
1 parent 12b9f34 commit 7289b79

File tree

7 files changed

+163
-0
lines changed

7 files changed

+163
-0
lines changed

‎acceptance/bundle/deploy/models/basic/out.test.toml‎

Lines changed: 5 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎acceptance/bundle/deploy/models/basic/output.txt‎

Lines changed: 88 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,88 @@
1+
2+
=== deply a model
3+
>>> [CLI] bundle plan
4+
create models.my_model
5+
6+
Plan: 1 to add, 0 to change, 0 to delete, 0 unchanged
7+
8+
>>> [CLI] bundle deploy
9+
Uploading bundle files to /Workspace/Users/[USERNAME]/.bundle/deploy-models-basic-[UNIQUE_NAME]/default/files...
10+
Deploying resources...
11+
Updating deployment state...
12+
Deployment complete!
13+
14+
=== GET Model
15+
>>> [CLI] model-registry get-model my-model-[UNIQUE_NAME]
16+
{
17+
"name": "my-model-[UNIQUE_NAME]",
18+
"description": "original description",
19+
"tags": [
20+
{
21+
"key": "key1",
22+
"value": "value1"
23+
}
24+
]
25+
}
26+
27+
=== add a new tag, this should be a no-op
28+
>>> [CLI] bundle plan
29+
update models.my_model
30+
31+
Plan: 0 to add, 1 to change, 0 to delete, 0 unchanged
32+
33+
>>> [CLI] bundle deploy
34+
Uploading bundle files to /Workspace/Users/[USERNAME]/.bundle/deploy-models-basic-[UNIQUE_NAME]/default/files...
35+
Deploying resources...
36+
Updating deployment state...
37+
Deployment complete!
38+
39+
=== GET Model
40+
>>> [CLI] model-registry get-model my-model-[UNIQUE_NAME]
41+
{
42+
"name": "my-model-[UNIQUE_NAME]",
43+
"description": "original description",
44+
"tags": [
45+
{
46+
"key": "key1",
47+
"value": "value1"
48+
}
49+
]
50+
}
51+
52+
=== update the description, this should update the description
53+
>>> [CLI] bundle plan
54+
update models.my_model
55+
56+
Plan: 0 to add, 1 to change, 0 to delete, 0 unchanged
57+
58+
>>> [CLI] bundle deploy
59+
Uploading bundle files to /Workspace/Users/[USERNAME]/.bundle/deploy-models-basic-[UNIQUE_NAME]/default/files...
60+
Deploying resources...
61+
Updating deployment state...
62+
Deployment complete!
63+
64+
=== GET Model
65+
>>> [CLI] model-registry get-model my-model-[UNIQUE_NAME]
66+
{
67+
"name": "my-model-[UNIQUE_NAME]",
68+
"description": "new description",
69+
"tags": [
70+
{
71+
"key": "key1",
72+
"value": "value1"
73+
}
74+
]
75+
}
76+
77+
=== destroy the model
78+
>>> [CLI] bundle destroy --auto-approve
79+
The following resources will be deleted:
80+
delete model my_model
81+
82+
All files and directories at the following location will be deleted: /Workspace/Users/[USERNAME]/.bundle/deploy-models-basic-[UNIQUE_NAME]/default
83+
84+
Deleting files...
85+
Destroy complete!
86+
87+
>>> [CLI] bundle destroy --auto-approve
88+
No active deployment found to destroy!

‎acceptance/bundle/deploy/models/basic/script‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
title "deply a model"
2+
envsubst < templates/original.tmpl > databricks.yml
3+
4+
trace $CLI bundle plan
5+
6+
cleanup() {
7+
trace $CLI bundle destroy --auto-approve
8+
}
9+
trap cleanup EXIT
10+
11+
trace $CLI bundle deploy
12+
13+
title "GET Model"
14+
trace $CLI model-registry get-model my-model-$UNIQUE_NAME | jq '.registered_model_databricks | {name, description, tags}'
15+
16+
title "add a new tag, this should be a no-op"
17+
envsubst < templates/additional_tag.tmpl > databricks.yml
18+
trace $CLI bundle plan
19+
trace $CLI bundle deploy
20+
21+
title "GET Model"
22+
trace $CLI model-registry get-model my-model-$UNIQUE_NAME | jq '.registered_model_databricks | {name, description, tags}'
23+
24+
title "update the description, this should update the description"
25+
envsubst < templates/new_description.tmpl > databricks.yml
26+
trace $CLI bundle plan
27+
trace $CLI bundle deploy
28+
29+
title "GET Model"
30+
trace $CLI model-registry get-model my-model-$UNIQUE_NAME | jq '.registered_model_databricks | {name, description, tags}'
31+
32+
title "destroy the model"
33+
trace $CLI bundle destroy --auto-approve

‎acceptance/bundle/deploy/models/basic/templates/additional_tag.tmpl‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
bundle:
2+
name: deploy-models-basic-$UNIQUE_NAME
3+
4+
resources:
5+
models:
6+
my_model:
7+
name: my-model-$UNIQUE_NAME
8+
description: "original description"
9+
tags:
10+
- key: "key1"
11+
value: "value1"
12+
- key: "key2"
13+
value: "value2"

‎acceptance/bundle/deploy/models/basic/templates/new_description.tmpl‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
bundle:
2+
name: deploy-models-basic-$UNIQUE_NAME
3+
4+
resources:
5+
models:
6+
my_model:
7+
name: my-model-$UNIQUE_NAME
8+
description: "new description"
9+
tags:
10+
- key: "key1"
11+
value: "value1"

‎acceptance/bundle/deploy/models/basic/templates/original.tmpl‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
bundle:
2+
name: deploy-models-basic-$UNIQUE_NAME
3+
4+
resources:
5+
models:
6+
my_model:
7+
name: my-model-$UNIQUE_NAME
8+
description: "original description"
9+
tags:
10+
- key: "key1"
11+
value: "value1"

‎acceptance/bundle/deploy/models/basic/test.toml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
Cloud = true
2+
Local = false

0 commit comments

Comments
 (0)