Function expectBuild

coderd/wsbuilder/wsbuilder_test.go:1560–1573

expectBuild captures a call to InsertWorkspaceBuild and runs the provided assertions against it.

(
	assertions func(job database.InsertWorkspaceBuildParams),
)

// expectBuild captures a call to InsertWorkspaceBuild and runs the provided assertions
// against it.
func expectBuild(
	assertions func(job database.InsertWorkspaceBuildParams),
) func(mTx *dbmock.MockStore) {
	return func(mTx *dbmock.MockStore) {
		mTx.EXPECT().InsertWorkspaceBuild(gomock.Any(), gomock.Any()).
			Times(1).
			DoAndReturn(
				func(ctx context.Context, params database.InsertWorkspaceBuildParams) error {
					assertions(params)
					return nil
				},
			)
	}
}

// expectBuildParameters captures a call to InsertWorkspaceBuildParameters and runs the provided assertions
// against it.