Add ceoverrides to unit test objects

knative · May 29, 2020 · e8843ae · e8843ae
1 parent cdc98fc
commit e8843ae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/pkg/sources/v1alpha2/apiserver_client_test.go b/pkg/sources/v1alpha2/apiserver_client_test.go
@@ -130,13 +130,16 @@ func TestListAPIServerSource(t *testing.T) {
 }
 
 func newAPIServerSource(name, resource string) *v1alpha2.ApiServerSource {
-	b := NewAPIServerSourceBuilder(name).ServiceAccount("testsa").EventMode("Reference")
-	b.Sink(duckv1.Destination{
-		Ref: &duckv1.KReference{
-			Kind:      "Service",
-			Name:      "foosvc",
-			Namespace: "default",
-		}})
+	b := NewAPIServerSourceBuilder(name).
+		ServiceAccount("testsa").
+		EventMode("Reference").
+		CloudEventOverrides(map[string]string{"type": "foo"}, []string{}).
+		Sink(duckv1.Destination{
+			Ref: &duckv1.KReference{
+				Kind:      "Service",
+				Name:      "foosvc",
+				Namespace: "default",
+			}})
 
 	if resource != "" {
 		b.Resources([]v1alpha2.APIVersionKindSelector{{

diff --git a/pkg/sources/v1alpha2/ping_client_test.go b/pkg/sources/v1alpha2/ping_client_test.go
@@ -135,7 +135,8 @@ func TestListPingSource(t *testing.T) {
 func newPingSource(name string, sink string) *v1alpha2.PingSource {
 	b := NewPingSourceBuilder(name).
 		Schedule("* * * * *").
-		JsonData("mydata")
+		JsonData("mydata").
+		CloudEventOverrides(map[string]string{"type": "foo"}, []string{})
 
 	if sink != "" {
 		b.Sink(