fix tests

Signed-off-by: Ivan Milchev <ivan@mondoo.com>
mondoohq · Apr 19, 2024 · c1fd676 · c1fd676
1 parent 5c4243b
commit c1fd676
Showing 1 changed file with 76 additions and 0 deletions.
diff --git a/policy/evidence_test.go b/policy/evidence_test.go
@@ -53,6 +53,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  },
  },
  }
@@ -68,6 +72,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  }
  require.Equal(t, expected, cm)
  })
@@ -88,6 +96,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Mrn: "query1"},
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Mrn: "control1"},
+ {Mrn: "control2"},
+ },
  },
  },
  }
@@ -103,6 +115,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Mrn: "query1"},
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Mrn: "control1"},
+ {Mrn: "control2"},
+ },
  }
  require.Equal(t, expected, cm)
  })
@@ -123,6 +139,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  },
  {
  Uid: "evidence-uid-2",
@@ -136,6 +156,10 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Uid: "query3"},
  {Uid: "query4"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control3"},
+ {Uid: "control4"},
+ },
  },
  },
  }
@@ -155,6 +179,12 @@ func TestGenerateEvidenceControlMap(t *testing.T) {
  {Uid: "query3"},
  {Uid: "query4"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ {Uid: "control3"},
+ {Uid: "control4"},
+ },
  }
  require.Equal(t, expected, cm)
  })
@@ -197,6 +227,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  },
  },
  }
@@ -228,6 +262,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  },
  },
  PolicyDependencies: []*explorer.ObjectRef{{Uid: "policy-uid"}},
@@ -249,6 +287,9 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  Queries: []*explorer.Mquery{
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Mrn: "control2"},
+ },
  },
  },
  }
@@ -278,6 +319,9 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  Queries: []*ControlRef{
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Mrn: "control2"},
+ },
  },
  },
  }
@@ -300,6 +344,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query1"},
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Mrn: "control2"},
+ },
  },
  },
  }
@@ -331,6 +379,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query1"},
  {Mrn: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Mrn: "control2"},
+ },
  },
  },
  PolicyDependencies: []*explorer.ObjectRef{{Uid: "policy-uid"}},
@@ -354,6 +406,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ },
  },
  {
  Uid: "evidence-uid-2",
@@ -367,6 +423,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query3"},
  {Uid: "query4"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control3"},
+ {Uid: "control4"},
+ },
  },
  },
  }
@@ -385,6 +445,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query5"},
  {Uid: "query6"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control5"},
+ {Uid: "control6"},
+ },
  },
  },
  }
@@ -418,6 +482,12 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query3"},
  {Uid: "query4"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control1"},
+ {Uid: "control2"},
+ {Uid: "control3"},
+ {Uid: "control4"},
+ },
  },
  {
  Uid: "control-uid-2",
@@ -429,6 +499,10 @@ func TestGenerateEvidenceFrameworkMap(t *testing.T) {
  {Uid: "query5"},
  {Uid: "query6"},
  },
+ Controls: []*ControlRef{
+ {Uid: "control5"},
+ {Uid: "control6"},
+ },
  },
  },
  PolicyDependencies: []*explorer.ObjectRef{{Uid: "policy-uid"}},
@@ -775,6 +849,7 @@ func TestGenerateEvidenceObjects(t *testing.T) {
  {Uid: "query1"},
  {Uid: "query2"},
  },
+ Controls: []*ControlRef{},
  },
  },
  PolicyDependencies: []*explorer.ObjectRef{{Uid: "framework-uid-evidence-policy"}},
@@ -830,6 +905,7 @@ func TestGenerateEvidenceObjects(t *testing.T) {
  Queries: []*ControlRef{
  {Mrn: "query1"},
  },
+ Controls: []*ControlRef{},
  },
  },
  }