export_test.go RawHistoryFolder

1// SPDX-License-Identifier: AGPL-3.0-or-later
2
3package kvstore
4
5import "context"
6
7// SetAfterSettingsSnapshot installs the hook that runs between the snapshot of
8// DeleteEnvironment and its batch, and returns the function that removes it.
9func SetAfterSettingsSnapshot(f func()) (restore func()) {
10	afterSettingsSnapshot = f
11	return func() { afterSettingsSnapshot = nil }
12}
13
14// DecodeConcurrencyRecord, EncodeConcurrencyRecord and EncodeConcurrencyMark
15// expose the stored layout of a concurrency group to the tests.
16var (
17	DecodeConcurrencyRecord = decodeConcurrencyRecord
18	EncodeConcurrencyRecord = encodeConcurrencyRecord
19	EncodeConcurrencyMark   = encodeConcurrencyMark
20)
21
22// RawVersionForTest is the kv version of a runner record: it moves on every
23// write.
24func (c *CI) RawVersionForTest(ctx context.Context, id string) (uint64, bool) {
25	item, ok := c.db.Get(ctx, runnerKey(id))
26	return item.Version, ok
27}