add more integration test for model webhook(set custom createdAt) (#467)

Author: googs1025

test/integration/webhook/model_test.go

@@ -17,6 +17,8 @@ limitations under the License.
 package webhook
 
 import (
+	"time"
+
 	"github.com/google/go-cmp/cmp/cmpopts"
 	"github.com/onsi/ginkgo/v2"
 	"github.com/onsi/gomega"
@@ -26,6 +28,10 @@ import (
 	"github.com/inftyai/llmaz/test/util/wrapper"
 )
 
+var (
+	testTime = metav1.Date(2025, 6, 20, 10, 0, 0, 0, time.UTC)
+)
+
 var _ = ginkgo.Describe("model default and validation", func() {
 
 	// Delete all the Models for each case.
@@ -75,6 +81,14 @@ var _ = ginkgo.Describe("model default and validation", func() {
 				return wrapper.MakeModel("llama3-8b").ModelSourceWithModelID("LLM-Research/Meta-Llama-3-8B", "", "main", nil, nil).ModelSourceWithModelHub("ModelScope").FamilyName("llama3").Label(coreapi.ModelFamilyNameLabelKey, "llama3").OwnedBy("custom-owner").Obj()
 			},
 		}),
+		ginkgo.Entry("set custom createdAt", &testDefaultingCase{
+			model: func() *coreapi.OpenModel {
+				return wrapper.MakeModel("llama3-8b").FamilyName("llama3").ModelSourceWithModelHub("ModelScope").ModelSourceWithModelID("LLM-Research/Meta-Llama-3-8B", "", "", nil, nil).CreatedAt(testTime).Obj()
+			},
+			wantModel: func() *coreapi.OpenModel {
+				return wrapper.MakeModel("llama3-8b").ModelSourceWithModelID("LLM-Research/Meta-Llama-3-8B", "", "main", nil, nil).ModelSourceWithModelHub("ModelScope").FamilyName("llama3").Label(coreapi.ModelFamilyNameLabelKey, "llama3").OwnedBy(coreapi.DefaultOwnedBy).CreatedAt(testTime).Obj()
+			},
+		}),
 	)
 
 	type testValidatingCase struct {

test/util/wrapper/model.go

@@ -112,6 +112,11 @@ func (w *ModelWrapper) OwnedBy(ownedBy string) *ModelWrapper {
 	return w
 }
 
+func (w *ModelWrapper) CreatedAt(createdAt metav1.Time) *ModelWrapper {
+	w.Spec.CreatedAt = &createdAt
+	return w
+}
+
 func MakeFlavor(name string) *FlavorWrapper {
 	return &FlavorWrapper{
 		coreapi.Flavor{