calling setting up defer fn before checking error to prevent panic

[Uk1288]

Description

Some tests was setting up defer fn before checking error which caused panic when actual errors happended. This panics would mask the actual error. Checking for error before defer function ensures there is no attempt to dereference nil values.

Testing plan

Run Automated tests

External links

Output from tests

Including output from tests may require access to a TFE instance. Ignore this section if you have no environment to test against.

$ TFE_ADDRESS="https://example" TFE_TOKEN="example" go test ./... -v -run TestFunctionsAffectedByChange

...

Rollback Plan

Changes to Security Controls

[datadog-terraform-cloud-hashicorp]

✅ Tests

🎉 All green!

❄️ No new flaky tests detected
🧪 All tests passed

_{This comment will be updated automatically if new data arrives.
🔗 Commit SHA: 8f3f793 | Docs | Was this helpful? Give us feedback!}

[github-actions]

Reminder to the contributor that merged this PR: if your changes have added important functionality or fixed a relevant bug, open a follow-up PR to update CHANGELOG.md with a note on your changes.