Rework state testing to allow State structs to be empty
Pod and container State structs are now allowed to be empty on first being retrieved from the database. Rework pod and container equality functions used in testing to account for this change. Signed-off-by: Matthew Heon <matthew.heon@gmail.com> Closes: #1186 Approved by: rhatdan
This commit is contained in:
Matthew Heon
Atomic Bot
parent
f4120f9662
commit
c7c56d800c
|
|
@ -4,6 +4,7 @@ import (
|
|||
"encoding/json"
|
||||
"net"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
|
@ -145,23 +146,27 @@ func getTestPod2(lockPath string) (*Pod, error) {
|
|||
|
||||
// This horrible hack tests if containers are equal in a way that should handle
|
||||
// empty arrays being dropped to nil pointers in the spec JSON
|
||||
func testContainersEqual(t *testing.T, a, b *Container) {
|
||||
// For some operations (container retrieval from the database) state is allowed
|
||||
// to be empty. This is controlled by the allowedEmpty bool.
|
||||
func testContainersEqual(t *testing.T, a, b *Container, allowedEmpty bool) {
|
||||
if a == nil && b == nil {
|
||||
return
|
||||
}
|
||||
require.NotNil(t, a)
|
||||
require.NotNil(t, b)
|
||||
|
||||
assert.NotNil(t, a.config)
|
||||
assert.NotNil(t, b.config)
|
||||
assert.NotNil(t, a.state)
|
||||
assert.NotNil(t, b.state)
|
||||
require.NotNil(t, a.config)
|
||||
require.NotNil(t, b.config)
|
||||
require.NotNil(t, a.state)
|
||||
require.NotNil(t, b.state)
|
||||
|
||||
aConfig := new(ContainerConfig)
|
||||
bConfig := new(ContainerConfig)
|
||||
aState := new(containerState)
|
||||
bState := new(containerState)
|
||||
|
||||
blankState := new(containerState)
|
||||
|
||||
assert.Equal(t, a.valid, b.valid)
|
||||
|
||||
aConfigJSON, err := json.Marshal(a.config)
|
||||
|
|
@ -186,25 +191,38 @@ func testContainersEqual(t *testing.T, a, b *Container) {
|
|||
err = json.Unmarshal(bStateJSON, bState)
|
||||
assert.NoError(t, err)
|
||||
|
||||
assert.EqualValues(t, aState, bState)
|
||||
if allowedEmpty {
|
||||
assert.True(t, reflect.DeepEqual(aState, bState) || reflect.DeepEqual(aState, blankState))
|
||||
} else {
|
||||
assert.EqualValues(t, aState, bState)
|
||||
}
|
||||
}
|
||||
|
||||
// Test if pods are equal
|
||||
func testPodsEqual(t *testing.T, a, b *Pod) {
|
||||
// Test if pods are equal.
|
||||
// For some operations (pod retrieval from the database) state is allowed to be
|
||||
// empty. This is controlled by the allowedEmpty bool.
|
||||
func testPodsEqual(t *testing.T, a, b *Pod, allowedEmpty bool) {
|
||||
if a == nil && b == nil {
|
||||
return
|
||||
}
|
||||
|
||||
assert.NotNil(t, a)
|
||||
assert.NotNil(t, b)
|
||||
blankState := new(podState)
|
||||
|
||||
assert.NotNil(t, a.config)
|
||||
assert.NotNil(t, b.config)
|
||||
assert.NotNil(t, a.state)
|
||||
assert.NotNil(t, b.state)
|
||||
require.NotNil(t, a)
|
||||
require.NotNil(t, b)
|
||||
|
||||
require.NotNil(t, a.config)
|
||||
require.NotNil(t, b.config)
|
||||
require.NotNil(t, a.state)
|
||||
require.NotNil(t, b.state)
|
||||
|
||||
assert.Equal(t, a.valid, b.valid)
|
||||
|
||||
assert.EqualValues(t, a.config, b.config)
|
||||
assert.EqualValues(t, a.state, b.state)
|
||||
|
||||
if allowedEmpty {
|
||||
assert.True(t, reflect.DeepEqual(a.state, b.state) || reflect.DeepEqual(a.state, blankState))
|
||||
} else {
|
||||
assert.EqualValues(t, a.state, b.state)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -108,7 +108,7 @@ func TestAddAndGetContainer(t *testing.T) {
|
|||
retrievedCtr, err := state.Container(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -128,7 +128,7 @@ func TestAddAndGetContainerFromMultiple(t *testing.T) {
|
|||
retrievedCtr, err := state.Container(testCtr1.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr1, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -278,7 +278,7 @@ func TestAddCtrDepInPodFails(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
require.Len(t, ctrs, 1)
|
||||
|
||||
testContainersEqual(t, testCtr1, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -328,7 +328,7 @@ func TestAddCtrDepInDifferentNamespaceFails(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(ctrs))
|
||||
|
||||
testContainersEqual(t, testCtr1, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -347,7 +347,7 @@ func TestAddCtrSameNamespaceSucceeds(t *testing.T) {
|
|||
retrievedCtr, err := state.Container(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -417,7 +417,7 @@ func TestGetContainerInSameNamespaceSucceeds(t *testing.T) {
|
|||
ctr, err := state.Container(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, ctr)
|
||||
testContainersEqual(t, ctr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -434,7 +434,7 @@ func TestGetContainerInNamespaceWhileNotInNamespaceSucceeds(t *testing.T) {
|
|||
ctr, err := state.Container(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, ctr)
|
||||
testContainersEqual(t, ctr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -463,7 +463,7 @@ func TestLookupContainerByFullID(t *testing.T) {
|
|||
retrievedCtr, err := state.LookupContainer(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -478,7 +478,7 @@ func TestLookupContainerByUniquePartialID(t *testing.T) {
|
|||
retrievedCtr, err := state.LookupContainer(testCtr.ID()[0:8])
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -511,7 +511,7 @@ func TestLookupContainerByName(t *testing.T) {
|
|||
retrievedCtr, err := state.LookupContainer(testCtr.Name())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -556,7 +556,7 @@ func TestLookupCtrInSameNamespaceSucceeds(t *testing.T) {
|
|||
ctr, err := state.LookupContainer(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, ctr)
|
||||
testContainersEqual(t, ctr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -597,7 +597,7 @@ func TestLookupContainerMatchInDifferentNamespaceSucceeds(t *testing.T) {
|
|||
ctr, err := state.LookupContainer("000")
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr2, ctr)
|
||||
testContainersEqual(t, ctr, testCtr2, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -701,7 +701,7 @@ func TestSaveAndUpdateContainer(t *testing.T) {
|
|||
err = state.UpdateContainer(testCtr)
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, testCtr, retrievedCtr, false)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -730,7 +730,7 @@ func TestSaveAndUpdateContainerSameNamespaceSucceeds(t *testing.T) {
|
|||
err = state.UpdateContainer(testCtr)
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, testCtr, retrievedCtr, false)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -883,7 +883,7 @@ func TestGetAllContainersWithOneContainer(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
require.Len(t, ctrs, 1)
|
||||
|
||||
testContainersEqual(t, testCtr, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -946,7 +946,7 @@ func TestGetContainerOneContainerInNamespace(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(ctrs))
|
||||
|
||||
testContainersEqual(t, testCtr1, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1314,7 +1314,7 @@ func TestGetPodOnePod(t *testing.T) {
|
|||
statePod, err := state.Pod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1335,7 +1335,7 @@ func TestGetOnePodFromTwo(t *testing.T) {
|
|||
statePod, err := state.Pod(testPod1.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod1, statePod)
|
||||
testPodsEqual(t, statePod, testPod1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1386,7 +1386,7 @@ func TestGetPodInNamespaceSucceeds(t *testing.T) {
|
|||
statePod, err := state.Pod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1432,7 +1432,7 @@ func TestLookupPodFullID(t *testing.T) {
|
|||
statePod, err := state.LookupPod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1447,7 +1447,7 @@ func TestLookupPodUniquePartialID(t *testing.T) {
|
|||
statePod, err := state.LookupPod(testPod.ID()[0:8])
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1481,7 +1481,7 @@ func TestLookupPodByName(t *testing.T) {
|
|||
statePod, err := state.LookupPod(testPod.Name())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, statePod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1526,7 +1526,7 @@ func TestLookupPodInSameNamespaceSucceeds(t *testing.T) {
|
|||
statePod, err := state.LookupPod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1570,7 +1570,7 @@ func TestLookupPodOneInDifferentNamespaceFindsRightPod(t *testing.T) {
|
|||
pod, err := state.LookupPod(strings.Repeat("1", 5))
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod1, pod)
|
||||
testPodsEqual(t, pod, testPod1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1686,8 +1686,7 @@ func TestAddPodValidPodSucceeds(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allPods))
|
||||
|
||||
testPodsEqual(t, testPod, allPods[0])
|
||||
assert.Equal(t, testPod.valid, allPods[0].valid)
|
||||
testPodsEqual(t, allPods[0], testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1807,8 +1806,7 @@ func TestAddPodSameNamespaceSucceeds(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allPods))
|
||||
|
||||
testPodsEqual(t, testPod, allPods[0])
|
||||
assert.Equal(t, testPod.valid, allPods[0].valid)
|
||||
testPodsEqual(t, allPods[0], testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1888,7 +1886,7 @@ func TestRemovePodFromPods(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allPods))
|
||||
|
||||
testPodsEqual(t, testPod2, allPods[0])
|
||||
testPodsEqual(t, allPods[0], testPod2, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -1986,7 +1984,7 @@ func TestAllPodsFindsPod(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allPods))
|
||||
|
||||
testPodsEqual(t, testPod, allPods[0])
|
||||
testPodsEqual(t, allPods[0], testPod, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -2070,7 +2068,7 @@ func TestAllPodsOnePodInDifferentNamespace(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allPods))
|
||||
|
||||
testPodsEqual(t, testPod1, allPods[0])
|
||||
testPodsEqual(t, allPods[0], testPod1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -2338,7 +2336,7 @@ func TestPodContainersOneContainer(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
require.Len(t, ctrs, 1)
|
||||
|
||||
testContainersEqual(t, testCtr, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -2644,8 +2642,8 @@ func TestAddContainerToPodSucceeds(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
require.Len(t, allCtrs, 1)
|
||||
|
||||
testContainersEqual(t, testCtr, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], allCtrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr, true)
|
||||
testContainersEqual(t, ctrs[0], allCtrs[0], false)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -2710,7 +2708,7 @@ func TestAddContainerToPodWithAddContainer(t *testing.T) {
|
|||
assert.NoError(t, err)
|
||||
assert.Equal(t, 2, len(allCtrs))
|
||||
|
||||
testContainersEqual(t, testCtr1, ctrs[0])
|
||||
testContainersEqual(t, ctrs[0], testCtr1, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -3029,7 +3027,7 @@ func TestAddContainerToPodSameNamespaceSucceeds(t *testing.T) {
|
|||
allCtrs, err := state.AllContainers()
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 1, len(allCtrs))
|
||||
testContainersEqual(t, testCtr, allCtrs[0])
|
||||
testContainersEqual(t, allCtrs[0], testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -3123,7 +3121,7 @@ func TestAddCtrToPodSameNamespaceSucceeds(t *testing.T) {
|
|||
retrievedCtr, err := state.Container(testCtr.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testContainersEqual(t, testCtr, retrievedCtr)
|
||||
testContainersEqual(t, retrievedCtr, testCtr, true)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -3477,7 +3475,7 @@ func TestSaveAndUpdatePod(t *testing.T) {
|
|||
statePod, err := state.Pod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
|
||||
testPod.state.CgroupPath = "/new/path/for/test"
|
||||
|
||||
|
|
@ -3487,7 +3485,7 @@ func TestSaveAndUpdatePod(t *testing.T) {
|
|||
err = state.UpdatePod(statePod)
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, false)
|
||||
})
|
||||
}
|
||||
|
||||
|
|
@ -3506,7 +3504,7 @@ func TestSaveAndUpdatePodSameNamespace(t *testing.T) {
|
|||
statePod, err := state.Pod(testPod.ID())
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, statePod, testPod, true)
|
||||
|
||||
testPod.state.CgroupPath = "/new/path/for/test"
|
||||
|
||||
|
|
@ -3516,6 +3514,6 @@ func TestSaveAndUpdatePodSameNamespace(t *testing.T) {
|
|||
err = state.UpdatePod(statePod)
|
||||
assert.NoError(t, err)
|
||||
|
||||
testPodsEqual(t, testPod, statePod)
|
||||
testPodsEqual(t, testPod, statePod, false)
|
||||
})
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue