File: network_store_test.go

package info (click to toggle)
docker.io 28.5.2%2Bdfsg1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 69,048 kB
  • sloc: sh: 5,867; makefile: 863; ansic: 184; python: 162; asm: 159
file content (70 lines) | stat: -rw-r--r-- 1,977 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package libnetwork

import (
	"context"
	"slices"
	"strings"
	"testing"

	"github.com/docker/docker/libnetwork/config"
	"gotest.tools/v3/assert"
	is "gotest.tools/v3/assert/cmp"
)

func TestNetworkStore(t *testing.T) {
	configOption := config.OptionDataDir(t.TempDir())
	c, err := New(context.Background(), configOption)
	assert.NilError(t, err)
	defer c.Stop()

	// Insert a first network
	nw1 := &Network{id: "testNetwork1", configFrom: "config-network"}
	err = c.storeNetwork(context.Background(), nw1)
	assert.NilError(t, err)

	// Then a second network
	nw2 := &Network{id: "testNetwork2"}
	err = c.storeNetwork(context.Background(), nw2)
	assert.NilError(t, err)

	// Check that we can find both networks, and that the returned values are
	// not copies of the original ones.
	for _, tc := range []struct {
		name        string
		filter      func(nw *Network) bool
		expNetworks []*Network
	}{
		{
			name:        "no filter",
			filter:      func(nw *Network) bool { return true },
			expNetworks: []*Network{nw1, nw2},
		},
		{
			name:        "filter by configFrom",
			filter:      filterNetworkByConfigFrom("config-network"),
			expNetworks: []*Network{nw1},
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			found := c.findNetworks(tc.filter)
			assert.Equal(t, len(found), len(tc.expNetworks))
			slices.SortFunc(found, func(a, b *Network) int { return strings.Compare(a.id, b.id) })
			for i, nw := range tc.expNetworks {
				assert.Check(t, is.Equal(found[i], nw), "got: %s; expected: %s", found[i].id, nw.id)
			}
		})
	}

	// Delete the first network
	err = c.deleteStoredNetwork(nw1)
	assert.NilError(t, err)

	// Check that we can only find the second network
	found := c.findNetworks(func(nw *Network) bool { return true })
	assert.Equal(t, len(found), 1)
	assert.Check(t, is.Equal(found[0], nw2), "got: %s; expected: %s", found[0].id, nw2.id)

	// Store the second network again
	err = c.storeNetwork(context.Background(), nw2)
	assert.NilError(t, err)
}