fix: Fix tests for snapshot does not exist error

Follow-up to #2925 Fix test matching and add additional wrapping for EBS
kanisterio · Jun 7, 2024 · d26d339 · d26d339
1 parent 3727ec4
commit d26d339
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/blockstorage/awsebs/awsebs.go b/pkg/blockstorage/awsebs/awsebs.go
@@ -530,7 +530,7 @@ func getSnapshots(ctx context.Context, ec2Cli *EC2, snapIDs []*string) ([]*ec2.S
 	dsi := &ec2.DescribeSnapshotsInput{SnapshotIds: snapIDs}
 	dso, err := ec2Cli.DescribeSnapshotsWithContext(ctx, dsi)
 	if err != nil {
-		return nil, errors.Wrapf(err, "Failed to get snapshot, snapshot_ids: %p", snapIDs)
+		return nil, errors.Wrapf(err, blockstorage.SnapshotDoesNotExistError+", snapshot_ids: %p", snapIDs)
 	}
 	// TODO: handle paging and continuation
 	if len(dso.Snapshots) != len(snapIDs) {

diff --git a/pkg/blockstorage/blockstorage_test.go b/pkg/blockstorage/blockstorage_test.go
@@ -143,7 +143,7 @@ func (s *BlockStorageProviderSuite) TestCreateSnapshot(c *C) {
 		s.snapshots = nil
 		_, err = s.provider.SnapshotGet(context.Background(), snapshot.ID)
 		c.Assert(err, NotNil)
-		c.Assert(err.Error(), Matches, blockstorage.SnapshotDoesNotExistError)
+		c.Assert(strings.Contains(err.Error(), blockstorage.SnapshotDoesNotExistError), Equals, true)
 	}
 }