From ce552655e3f99d0c19cccea81a22826840143a32 Mon Sep 17 00:00:00 2001
From: akinross <akinross@cisco.com>
Date: Thu, 30 May 2024 19:22:16 +0200
Subject: [PATCH] [ignore] fix class include function and fix excluding
 unneeded testvars headers

---
 gen/generator.go                 | 2 --
 gen/templates/testvars.yaml.tmpl | 3 ++-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/gen/generator.go b/gen/generator.go
index c81f903c1..050a773f9 100644
--- a/gen/generator.go
+++ b/gen/generator.go
@@ -1141,8 +1141,6 @@ func (m *Model) SetClassInclude() {
 		for key, value := range classDetails.(map[interface{}]interface{}) {
 			if key.(string) == "include" {
 				m.Include = value.(bool)
-			} else {
-				m.Include = false
 			}
 		}
 	}
diff --git a/gen/templates/testvars.yaml.tmpl b/gen/templates/testvars.yaml.tmpl
index b6ba5ea8b..549b13f40 100644
--- a/gen/templates/testvars.yaml.tmpl
+++ b/gen/templates/testvars.yaml.tmpl
@@ -10,7 +10,7 @@ default:
   {{overwriteProperty .PkgName .SnakeCaseName $.Definitions}}: "{{overwritePropertyValue .PkgName $overwritePropertyName "" "default" 0 $.Definitions}}"
     {{- end}}
   {{- end}}
-
+{{ if .HasNaming }}
 datasource_non_existing:
   {{- range .Properties}}{{$testValue := createNonExistingValue .SnakeCaseName}}{{$overwritePropertyName := overwriteProperty .PkgName .SnakeCaseName $.Definitions}}
     {{- if .IsNaming }}
@@ -31,6 +31,7 @@ resource_required:
   {{overwriteProperty .PkgName .SnakeCaseName $.Definitions}}: "{{overwritePropertyValue .PkgName $overwritePropertyName $testValue "resource_required" 0 $.Definitions}}"
     {{- end}}
   {{- end}}
+{{- end }}
 
 all:
   {{- range .Properties}}{{$overwritePropertyName := overwriteProperty .PkgName .SnakeCaseName $.Definitions}}