From d0c74653356826e8eccf847df40a768620370691 Mon Sep 17 00:00:00 2001
From: Pieter Noordhuis <pieter.noordhuis@databricks.com>
Date: Wed, 24 May 2023 11:10:27 +0200
Subject: [PATCH] Use constants instead of hardcoding strings

---
 internal/workspace_test.go | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/internal/workspace_test.go b/internal/workspace_test.go
index 5c5cc6e32..7bf8002ee 100644
--- a/internal/workspace_test.go
+++ b/internal/workspace_test.go
@@ -37,8 +37,8 @@ func TestAccWorkspaceIntegration(t *testing.T) {
 	// Import the test notebook
 	err := w.Workspace.Import(ctx, workspace.Import{
 		Path:      notebook,
-		Format:    "SOURCE",
-		Language:  "PYTHON",
+		Format:    workspace.ExportFormatSource,
+		Language:  workspace.LanguagePython,
 		Content:   base64.StdEncoding.EncodeToString([]byte("# Databricks notebook source\nprint('hello from job')")),
 		Overwrite: true,
 	})
@@ -47,14 +47,14 @@ func TestAccWorkspaceIntegration(t *testing.T) {
 	// Get test notebook status
 	getStatusResponse, err := w.Workspace.GetStatusByPath(ctx, notebook)
 	require.NoError(t, err)
-	assert.True(t, getStatusResponse.Language == "PYTHON")
+	assert.True(t, getStatusResponse.Language == workspace.LanguagePython)
 	assert.True(t, getStatusResponse.Path == notebook)
-	assert.True(t, string(getStatusResponse.ObjectType) == "NOTEBOOK")
+	assert.True(t, getStatusResponse.ObjectType == workspace.ObjectTypeNotebook)
 
 	// Export the notebook and assert the contents
 	exportResponse, err := w.Workspace.Export(ctx, workspace.ExportRequest{
 		DirectDownload: false,
-		Format:         "SOURCE",
+		Format:         workspace.ExportFormatSource,
 		Path:           notebook,
 	})
 	require.NoError(t, err)
@@ -81,8 +81,8 @@ func TestAccWorkspaceRecursiveListNoTranspile(t *testing.T) {
 	// Import the test notebook
 	err := w.Workspace.Import(ctx, workspace.Import{
 		Path:      notebook,
-		Format:    "SOURCE",
-		Language:  "PYTHON",
+		Format:    workspace.ExportFormatSource,
+		Language:  workspace.LanguagePython,
 		Content:   base64.StdEncoding.EncodeToString([]byte("# Databricks notebook source\nprint('hello from job')")),
 		Overwrite: true,
 	})