[AzDatalake] Set Expiry (#22170)

* corrected release date in changelog * setExpiry for files * lint error * recordings
Azure · Dec 28, 2023 · 20b1992 · 20b1992
1 parent 396c97c
commit 20b1992
Show file tree

Hide file tree

Showing 2 changed files with 80 additions and 1 deletion.
diff --git a/sdk/storage/azdatalake/assets.json b/sdk/storage/azdatalake/assets.json
@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "go",
   "TagPrefix": "go/storage/azdatalake",
-  "Tag": "go/storage/azdatalake_36b1978c0d"
+  "Tag": "go/storage/azdatalake_c883370a69"
 }
diff --git a/sdk/storage/azdatalake/file/client_test.go b/sdk/storage/azdatalake/file/client_test.go
@@ -924,6 +924,85 @@ func (s *RecordedTestSuite) TestDeleteFileIfETagMatchFalse() {
 	testcommon.ValidateErrorCode(_require, err, datalakeerror.ConditionNotMet)
 }
 
+func (s *RecordedTestSuite) TestFileSetExpiry() {
+	_require := require.New(s.T())
+	testName := s.T().Name()
+
+	filesystemName := testcommon.GenerateFileSystemName(testName)
+	fsClient, err := testcommon.GetFileSystemClient(filesystemName, s.T(), testcommon.TestAccountDatalake, nil)
+	_require.NoError(err)
+	defer testcommon.DeleteFileSystem(context.Background(), _require, fsClient)
+
+	_, err = fsClient.Create(context.Background(), nil)
+	_require.NoError(err)
+
+	fileName := testcommon.GenerateFileName(testName)
+	fClient, err := testcommon.GetFileClient(filesystemName, fileName, s.T(), testcommon.TestAccountDatalake, nil)
+	_require.NoError(err)
+
+	resp, err := fClient.Create(context.Background(), nil)
+	_require.NoError(err)
+	_require.NotNil(resp)
+
+	_, err = fClient.SetExpiry(context.Background(), file.SetExpiryValues{ExpiryType: file.SetExpiryTypeNeverExpire}, nil)
+	_require.NoError(err)
+
+	res, err := fClient.GetProperties(context.Background(), nil)
+	_require.Nil(res.ExpiresOn)
+	_require.NoError(err)
+
+	_, err = fClient.SetExpiry(
+		context.Background(),
+		file.SetExpiryValues{
+			ExpiryType: file.SetExpiryTypeRelativeToCreation,
+			ExpiresOn:  strconv.Itoa(10),
+		},
+		nil,
+	)
+	_require.NoError(err)
+
+	time.Sleep(time.Second * 12)
+
+	_, err = fClient.GetProperties(context.Background(), nil)
+	testcommon.ValidateErrorCode(_require, err, datalakeerror.PathNotFound)
+}
+
+func (s *UnrecordedTestSuite) TestFileSetExpiryTypeAbsoluteTime() {
+	_require := require.New(s.T())
+	testName := s.T().Name()
+
+	filesystemName := testcommon.GenerateFileSystemName(testName)
+	fsClient, err := testcommon.GetFileSystemClient(filesystemName, s.T(), testcommon.TestAccountDatalake, nil)
+	_require.NoError(err)
+	defer testcommon.DeleteFileSystem(context.Background(), _require, fsClient)
+
+	_, err = fsClient.Create(context.Background(), nil)
+	_require.NoError(err)
+
+	fileName := testcommon.GenerateFileName(testName)
+	fClient, err := testcommon.GetFileClient(filesystemName, fileName, s.T(), testcommon.TestAccountDatalake, nil)
+	_require.NoError(err)
+
+	resp, err := fClient.Create(context.Background(), nil)
+	_require.NoError(err)
+	_require.NotNil(resp)
+
+	_, err = fClient.SetExpiry(
+		context.Background(),
+		file.SetExpiryValues{
+			ExpiryType: file.SetExpiryTypeAbsolute,
+			ExpiresOn:  time.Now().Add(5 * time.Second).UTC().Format(http.TimeFormat),
+		},
+		nil)
+	_require.NoError(err)
+
+	time.Sleep(time.Second * 7)
+
+	_, err = fClient.GetProperties(context.Background(), nil)
+	testcommon.ValidateErrorCode(_require, err, datalakeerror.PathNotFound)
+
+}
+
 func (s *RecordedTestSuite) TestFileSetAccessControlNil() {
 	_require := require.New(s.T())
 	testName := s.T().Name()