adding Schema to test attestors (part of interface)

Signed-off-by: chaosinthecrd <[email protected]>
ChaosInTheCRD · May 10, 2024 · 9f3096e · 9f3096e
1 parent 009796d
commit 9f3096e
Show file tree

Hide file tree

Showing 9 changed files with 57 additions and 24 deletions.
diff --git a/internal/attestors/commandrun.go b/internal/attestors/commandrun.go
@@ -18,11 +18,10 @@ import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/commandrun"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ commandrun.CommandRunAttestor = &TestCommandRunAttestor{}
-)
+var _ commandrun.CommandRunAttestor = &TestCommandRunAttestor{}
 
 type TestCommandRunAttestor struct {
 	comAtt commandrun.CommandRun
@@ -45,6 +44,10 @@ func (t *TestCommandRunAttestor) RunType() attestation.RunType {
 	return t.comAtt.RunType()
 }
 
+func (t *TestCommandRunAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestCommandRunAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/environment.go b/internal/attestors/environment.go
@@ -17,11 +17,10 @@ package attestors
 import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/environment"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ environment.EnvironmentAttestor = &TestEnvironmentAttestor{}
-)
+var _ environment.EnvironmentAttestor = &TestEnvironmentAttestor{}
 
 type TestEnvironmentAttestor struct {
 	environmentAtt environment.Attestor
@@ -44,6 +43,10 @@ func (t *TestEnvironmentAttestor) RunType() attestation.RunType {
 	return t.environmentAtt.RunType()
 }
 
+func (t *TestEnvironmentAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestEnvironmentAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/git.go b/internal/attestors/git.go
@@ -18,11 +18,10 @@ import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/git"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ git.GitAttestor = &TestGitAttestor{}
-)
+var _ git.GitAttestor = &TestGitAttestor{}
 
 type TestGitAttestor struct {
 	gitAtt git.Attestor
@@ -45,6 +44,10 @@ func (t *TestGitAttestor) RunType() attestation.RunType {
 	return t.gitAtt.RunType()
 }
 
+func (t *TestGitAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestGitAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/github.go b/internal/attestors/github.go
@@ -19,11 +19,10 @@ import (
 	"github.com/in-toto/go-witness/attestation/github"
 	"github.com/in-toto/go-witness/attestation/jwt"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ github.GitHubAttestor = &TestGitHubAttestor{}
-)
+var _ github.GitHubAttestor = &TestGitHubAttestor{}
 
 type TestGitHubAttestor struct {
 	githubAtt github.Attestor
@@ -47,6 +46,10 @@ func (t *TestGitHubAttestor) RunType() attestation.RunType {
 	return t.githubAtt.RunType()
 }
 
+func (t *TestGitHubAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestGitHubAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/gitlab.go b/internal/attestors/gitlab.go
@@ -19,11 +19,10 @@ import (
 	"github.com/in-toto/go-witness/attestation/gitlab"
 	"github.com/in-toto/go-witness/attestation/jwt"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ gitlab.GitLabAttestor = &TestGitLabAttestor{}
-)
+var _ gitlab.GitLabAttestor = &TestGitLabAttestor{}
 
 type TestGitLabAttestor struct {
 	gitlabAtt gitlab.Attestor
@@ -47,6 +46,10 @@ func (t *TestGitLabAttestor) RunType() attestation.RunType {
 	return t.gitlabAtt.RunType()
 }
 
+func (t *TestGitLabAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestGitLabAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/material.go b/internal/attestors/material.go
@@ -18,11 +18,10 @@ import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/material"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ material.MaterialAttestor = &TestMaterialAttestor{}
-)
+var _ material.MaterialAttestor = &TestMaterialAttestor{}
 
 type TestMaterialAttestor struct {
 	matAtt    *material.Attestor
@@ -47,6 +46,10 @@ func (t *TestMaterialAttestor) RunType() attestation.RunType {
 	return t.matAtt.RunType()
 }
 
+func (t *TestMaterialAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestMaterialAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/oci.go b/internal/attestors/oci.go
@@ -18,11 +18,10 @@ import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/oci"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ oci.OCIAttestor = &TestOCIAttestor{}
-)
+var _ oci.OCIAttestor = &TestOCIAttestor{}
 
 type TestOCIAttestor struct {
 	ociAtt oci.Attestor
@@ -45,6 +44,10 @@ func (t *TestOCIAttestor) RunType() attestation.RunType {
 	return t.ociAtt.RunType()
 }
 
+func (t *TestOCIAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestOCIAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/internal/attestors/product.go b/internal/attestors/product.go
@@ -18,11 +18,10 @@ import (
 	"github.com/in-toto/go-witness/attestation"
 	"github.com/in-toto/go-witness/attestation/product"
 	"github.com/in-toto/go-witness/cryptoutil"
+	"github.com/invopop/jsonschema"
 )
 
-var (
-	_ product.ProductAttestor = &TestProductAttestor{}
-)
+var _ product.ProductAttestor = &TestProductAttestor{}
 
 type TestProductAttestor struct {
 	prodAtt product.ProductAttestor
@@ -45,6 +44,10 @@ func (t *TestProductAttestor) RunType() attestation.RunType {
 	return t.prodAtt.RunType()
 }
 
+func (t *TestProductAttestor) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(&t)
+}
+
 func (t *TestProductAttestor) Attest(ctx *attestation.AttestationContext) error {
 	return nil
 }

diff --git a/policy/policy_test.go b/policy/policy_test.go
@@ -32,6 +32,7 @@ import (
 	"github.com/in-toto/go-witness/cryptoutil"
 	"github.com/in-toto/go-witness/intoto"
 	"github.com/in-toto/go-witness/source"
+	"github.com/invopop/jsonschema"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
@@ -367,6 +368,10 @@ func (DummyMaterialer) RunType() attestation.RunType {
 	return attestation.PreMaterialRunType
 }
 
+func (DummyMaterialer) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(DummyMaterialer{})
+}
+
 func (DummyMaterialer) Attest(*attestation.AttestationContext) error {
 	return nil
 }
@@ -391,6 +396,10 @@ func (DummyProducer) RunType() attestation.RunType {
 	return attestation.PostProductRunType
 }
 
+func (DummyProducer) Schema() *jsonschema.Schema {
+	return jsonschema.Reflect(DummyProducer{})
+}
+
 func (DummyProducer) Attest(*attestation.AttestationContext) error {
 	return nil
 }