Function TestRestoreHandler

pkg/controller/controller_test.go:122–171

(t *testing.T)

}

func TestRestoreHandler(t *testing.T) {
	ctrl, client := newTestController()
	sgName := "foo"
	sgNamespace := "default"
	sg := newSnapshotGroup(sgName, sgNamespace)
	snaps, err := snapshots.ListSnapshots(sg)
	assert.NoError(t, err)
	assert.Equal(t, 0, len(snaps))

	_, err = client.SnapshotGroupClient.SnapshotGroups(sgNamespace).Create(context.Background(), sg, metav1.CreateOptions{})
	assert.NoError(t, err)

	event := workItem{
		name:          sgName,
		namespace:     sgNamespace,
		snapshotGroup: sg,
		task:          backupTask,
	}
	err = ctrl.syncHandler(event)
	assert.NoError(t, err)

	snaps, err = snapshots.ListSnapshots(sg)
	assert.NoError(t, err)
	assert.Equal(t, 1, len(snaps))
	assert.Equal(t, []string{"1 second"}, snaps[0].Intervals)

	pvcClient := client.K8s.CoreV1().PersistentVolumeClaims(sg.ObjectMeta.Namespace)
	pvc, err := pvcClient.Get(context.TODO(), sg.ObjectMeta.Name, metav1.GetOptions{})
	assert.NoError(t, err)
	assert.Equal(t, "gemini", pvc.ObjectMeta.Annotations["app.kubernetes.io/managed-by"])
	assert.Equal(t, "", pvc.ObjectMeta.Annotations["gemini.fairwinds.com/restore"])

	time.Sleep(time.Second)
	timestamp := strconv.Itoa(int(snaps[0].Timestamp.Unix()))
	sg.ObjectMeta.Annotations["gemini.fairwinds.com/restore"] = timestamp
	event.task = restoreTask
	err = ctrl.syncHandler(event)
	assert.NoError(t, err)

	pvc, err = pvcClient.Get(context.TODO(), sg.ObjectMeta.Name, metav1.GetOptions{})
	assert.NoError(t, err)
	assert.Equal(t, "gemini", pvc.ObjectMeta.Annotations["app.kubernetes.io/managed-by"])
	assert.Equal(t, timestamp, pvc.ObjectMeta.Annotations["gemini.fairwinds.com/restore"])

	snaps, err = snapshots.ListSnapshots(sg)
	assert.NoError(t, err)
	assert.Equal(t, 2, len(snaps))
	assert.Equal(t, []string{"1 second"}, snaps[1].Intervals)
	assert.Equal(t, []string{}, snaps[0].Intervals)
}

func TestDeleteHandler(t *testing.T) {
	ctrl, _ := newTestController()