1package files
2
3import (
4	"github.com/aptly-dev/aptly/aptly"
5	"github.com/aptly-dev/aptly/utils"
6)
7
8type mockChecksumStorage struct {
9	store map[string]utils.ChecksumInfo
10}
11
12// NewMockChecksumStorage creates aptly.ChecksumStorage for tests
13func NewMockChecksumStorage() aptly.ChecksumStorage {
14	return &mockChecksumStorage{
15		store: make(map[string]utils.ChecksumInfo),
16	}
17}
18
19func (st *mockChecksumStorage) Get(path string) (*utils.ChecksumInfo, error) {
20	c, ok := st.store[path]
21	if !ok {
22		return nil, nil
23	}
24
25	return &c, nil
26}
27
28func (st *mockChecksumStorage) Update(path string, c *utils.ChecksumInfo) error {
29	st.store[path] = *c
30	return nil
31}
32
33// Check interface
34var (
35	_ aptly.ChecksumStorage = &mockChecksumStorage{}
36)
37