remove redundant return statement and skip over nil messages

Author: stephybun

internal/plugintest/working_dir.go

@@ -538,9 +538,14 @@ func (wd *WorkingDir) Query(ctx context.Context) ([]tfjson.LogMsg, error) {
 	}
 
 	for msg := range logs {
+		if msg.Msg == nil {
+			continue
+		}
+
 		if msg.Err != nil {
 			return nil, fmt.Errorf("retrieving next message: %w", msg.Err)
 		}
+
 		if msg.Msg.Level() == tfjson.Error {
 			// TODO reimplement missing .tf config error
 			diags = append(diags, msg.Msg)

querycheck/expect_identity.go

@@ -79,8 +79,6 @@ func (e expectIdentity) CheckQuery(_ context.Context, req CheckQueryRequest, res
 	}
 	errCollection = append(errCollection, fmt.Errorf("address: %s\n", e.listResourceAddress))
 	resp.Error = errors.Join(errCollection...)
-
-	return
 }
 
 // ExpectIdentity returns a query check that asserts that the identity at the given resource matches a known object, where each