Ejemplo n.º 1
0
func TestInputRCAndPod(t *testing.T) {
	input := testTempFileInput(t)
	defer os.RemoveAll(input.Path())

	// setup pod
	kubePod := testKubePod()

	podFile := fmt.Sprintf("test.%s.%s", PodExtension, testRandomExtension())
	podPath := path.Join(input.Path(), podFile)
	testWriteYAMLToFile(t, podPath, kubePod)
	testClearTypeInfo(kubePod)
	pod, err := entity.NewPod(kubePod, kube.ObjectMeta{}, podPath)
	assert.NoError(t, err)

	// setup rc
	objects := testWriteRandomObjects(t, input.Path(), 5)
	kubeRC := &kube.ReplicationController{
		ObjectMeta: kube.ObjectMeta{
			Name:      "spread",
			Namespace: kube.NamespaceDefault,
		},
		TypeMeta: unversioned.TypeMeta{
			Kind:       "ReplicationController",
			APIVersion: "v1",
		},
		Spec: kube.ReplicationControllerSpec{
			Selector: map[string]string{"valid": "selector"},
			Template: &kube.PodTemplateSpec{
				Spec: kubePod.Spec,
			},
		},
	}

	rcFile := fmt.Sprintf("test.%s.%s", RCExtension, testRandomExtension())
	rcPath := path.Join(input.Path(), rcFile)
	testWriteYAMLToFile(t, rcPath, kubeRC)
	testClearTypeInfo(kubeRC)
	rc, err := entity.NewReplicationController(kubeRC, kube.ObjectMeta{}, rcPath)
	assert.NoError(t, err)

	expected, err := entity.NewApp(nil, kube.ObjectMeta{}, input.Path(), objects...)
	assert.NoError(t, err)

	// attach rc and pod
	err = expected.Attach(rc)
	assert.NoError(t, err)
	err = expected.Attach(pod)
	assert.NoError(t, err)

	// generate entity
	actual, err := input.Build()
	assert.NoError(t, err, "should have built entity successfully")

	testCompareEntity(t, expected, actual)
}
Ejemplo n.º 2
0
func TestSourcePods(t *testing.T) {
	fs := testTempFileSource(t)
	defer os.RemoveAll(string(fs))

	podFile := path.Join(string(fs), "test.pod.yaml")

	terminationPeriod := int64(30)
	kubePod := &kube.Pod{
		ObjectMeta: kube.ObjectMeta{
			Name:      "example-pod",
			Namespace: kube.NamespaceDefault,
		},
		TypeMeta: unversioned.TypeMeta{
			Kind:       "Pod",
			APIVersion: "v1",
		},
		Spec: kube.PodSpec{
			Containers: []kube.Container{
				{
					Name:                   "wiki",
					Image:                  "mediawiki",
					ImagePullPolicy:        kube.PullAlways,
					TerminationMessagePath: kube.TerminationMessagePathDefault,
				},
				{
					Name:                   "db",
					Image:                  "postgres",
					ImagePullPolicy:        kube.PullAlways,
					TerminationMessagePath: kube.TerminationMessagePathDefault,
				},
			},
			SecurityContext:               &kube.PodSecurityContext{},
			RestartPolicy:                 kube.RestartPolicyAlways,
			DNSPolicy:                     kube.DNSDefault,
			TerminationGracePeriodSeconds: &terminationPeriod,
		},
	}

	testWriteYAMLToFile(t, podFile, kubePod)

	pods, err := fs.Entities(entity.EntityPod)
	assert.NoError(t, err, "should be okay")
	assert.Len(t, pods, 1, "should have single pod")

	expectedKubePod := kubePod
	testClearTypeInfo(expectedKubePod)

	expected, err := entity.NewPod(expectedKubePod, kube.ObjectMeta{}, podFile)

	actual := pods[0]

	testCompareEntity(t, expected, actual)
}
Ejemplo n.º 3
0
func TestInputPod(t *testing.T) {
	input := testTempFileInput(t)
	defer os.RemoveAll(input.Path())

	kubePod := testKubePod()

	podFile := fmt.Sprintf("test.%s.%s", PodExtension, testRandomExtension())
	podPath := path.Join(input.Path(), podFile)
	testWriteYAMLToFile(t, podPath, kubePod)

	testClearTypeInfo(kubePod)

	objects := testWriteRandomObjects(t, input.Path(), 5)

	expectedPod, err := entity.NewPod(kubePod, kube.ObjectMeta{}, podPath)
	assert.NoError(t, err)
	expected, err := entity.NewApp([]entity.Entity{expectedPod}, kube.ObjectMeta{}, input.Path(), objects...)
	assert.NoError(t, err)

	actual, err := input.Build()
	assert.NoError(t, err, "should have built entity successfully")

	testCompareEntity(t, expected, actual)
}
Ejemplo n.º 4
0
// pods returns Pods for the rcs in the PodFile
func (fs FileSource) pods(objects []deploy.KubeObject) (pods []entity.Entity, err error) {
	pattern := fmt.Sprintf("%s/*%s", fs, PodExtension)

	patternExts := withExtensions(pattern)
	paths, err := multiGlob(patternExts...)
	if err != nil {
		return pods, err
	}

	for _, filePath := range paths {
		err = walkPathForObjects(filePath, func(info *resource.Info, err error) error {
			kubePod, ok := info.Object.(*kube.Pod)
			if !ok {
				return fmt.Errorf("expected type `Pod` but found `%s`", info.Object.GetObjectKind().GroupVersionKind().Kind)
			}

			pod, err := entity.NewPod(kubePod, kube.ObjectMeta{}, filePath, objects...)
			if err != nil {
				return err
			}

			pods = append(pods, pod)
			return nil
		})

		if checkErrPathDoesNotExist(err) {
			// it's okay if directory doesn't exist
			err = nil
		}

		if err != nil {
			return
		}
	}
	return
}