Skip to content
This repository has been archived by the owner on Aug 12, 2022. It is now read-only.

Commit

Permalink
add small test
Browse files Browse the repository at this point in the history
  • Loading branch information
Michel Vocks committed Aug 18, 2021
1 parent 1acff62 commit 2831d0c
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions provider/schema/execution_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -316,6 +316,26 @@ func TestExecutionData_ResolveTable(t *testing.T) {
assert.Equal(t, []string{"name", "name_no_prefix", "prefix_name", "cq_id", "meta", "injected_field"}, expectedResource.columns)
assert.Equal(t, []interface{}{"other", "name_no_prefix", "prefix_name", expectedResource.cqId, expectedResource.Get("meta"), 1}, values)
})

t.Run("test partial fetch", func(t *testing.T) {
mockDb := new(databaseMock)
execDefault := NewExecutionData(mockDb, logger, testDefaultsTable, false, nil, true)
mockDb.On("CopyFrom", mock.Anything, mock.Anything, false, mock.Anything).Return(nil)
testDefaultsTable.Resolver = func(ctx context.Context, meta ClientMeta, parent *Resource, res chan interface{}) error {
res <- testDefaultsTableData{Name: nil}
return nil
}
var expectedResource *Resource
testDefaultsTable.PostResourceResolver = func(ctx context.Context, meta ClientMeta, parent *Resource) error {
expectedResource = parent
return fmt.Errorf("random failure")
}
_, err := execDefault.ResolveTable(context.Background(), mockedClient, nil)
assert.Nil(t, err)
assert.Equal(t, expectedResource.data["name"], "defaultValue")
assert.Len(t, execDefault.PartialFetchFailureResult, 1)
assert.Equal(t, "post resource resolver failed: random failure", execDefault.PartialFetchFailureResult[0].Error)
})
}

// ClientMeta is an autogenerated mock type for the ClientMeta type
Expand Down

0 comments on commit 2831d0c

Please sign in to comment.