fix: add tests to ensure that Docker mount points are being rewritten properly (#173)

agneum · agneum · commit db86b2afe9f0 · 2020-10-16T12:58:46.000+07:00
diff --git a/pkg/retrieval/engine/postgres/tools/tools_test.go b/pkg/retrieval/engine/postgres/tools/tools_test.go
@@ -9,6 +9,8 @@ import (
 	"os"
 	"testing"
 
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/mount"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -33,3 +35,48 @@ func TestIfDirectoryEmpty(t *testing.T) {
 	require.NoError(t, err)
 	assert.False(t, isEmpty)
 }
+
+func TestGetMountsFromMountPoints(t *testing.T) {
+	testCases := []struct {
+		dataDir        string
+		mountPoints    []types.MountPoint
+		expectedPoints []mount.Mount
+	}{
+		{
+			dataDir: "/var/lib/dblab/clones/dblab_clone_6000/data",
+			mountPoints: []types.MountPoint{{
+				Source:      "/var/lib/pgsql/data",
+				Destination: "/var/lib/postgresql/data",
+			}},
+			expectedPoints: []mount.Mount{{
+				Source:   "/var/lib/pgsql/data",
+				Target:   "/var/lib/postgresql/data",
+				ReadOnly: true,
+				BindOptions: &mount.BindOptions{
+					Propagation: "",
+				},
+			}},
+		},
+
+		{
+			dataDir: "/var/lib/dblab/clones/dblab_clone_6000/data",
+			mountPoints: []types.MountPoint{{
+				Source:      "/var/lib/postgresql",
+				Destination: "/var/lib/dblab",
+			}},
+			expectedPoints: []mount.Mount{{
+				Source:   "/var/lib/postgresql/clones/dblab_clone_6000/data",
+				Target:   "/var/lib/dblab/clones/dblab_clone_6000/data",
+				ReadOnly: true,
+				BindOptions: &mount.BindOptions{
+					Propagation: "",
+				},
+			}},
+		},
+	}
+
+	for _, tc := range testCases {
+		mounts := GetMountsFromMountPoints(tc.dataDir, tc.mountPoints)
+		assert.Equal(t, tc.expectedPoints, mounts)
+	}
+}
