From c4d5edaf0d3d3638a25bd5f989a0d97d7f351740 Mon Sep 17 00:00:00 2001
From: Jiri Kortus <jikortus@redhat.com>
Date: Mon, 30 Mar 2020 19:48:40 +0200
Subject: [PATCH] Add unit tests for Pipeline struct

Refs: #406
---
 internal/osbuild/osbuild_test.go | 59 ++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 internal/osbuild/osbuild_test.go

diff --git a/internal/osbuild/osbuild_test.go b/internal/osbuild/osbuild_test.go
new file mode 100644
index 000000000..718315c1a
--- /dev/null
+++ b/internal/osbuild/osbuild_test.go
@@ -0,0 +1,59 @@
+// Package osbuild provides primitives for representing and (un)marshalling
+// OSBuild types.
+package osbuild
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPipeline_SetBuild(t *testing.T) {
+	expectedPipeline := &Pipeline{
+		Build: &Build{
+			Pipeline: &Pipeline{},
+			Runner:   "org.osbuild.fedora31",
+		},
+	}
+	actualPipeline := &Pipeline{}
+	actualPipeline.SetBuild(&Pipeline{}, "org.osbuild.fedora31")
+	assert.Equal(t, expectedPipeline, actualPipeline)
+}
+
+func TestPipeline_AddStage(t *testing.T) {
+	expectedPipeline := &Pipeline{
+		Build: &Build{
+			Pipeline: &Pipeline{},
+			Runner:   "org.osbuild.fedora31",
+		},
+		Stages: []*Stage{
+			{
+				Name: "org.osbuild.rpm",
+			},
+		},
+	}
+	actualPipeline := &Pipeline{
+		Build: &Build{
+			Pipeline: &Pipeline{},
+			Runner:   "org.osbuild.fedora31",
+		},
+	}
+	actualPipeline.AddStage(&Stage{
+		Name: "org.osbuild.rpm",
+	})
+	assert.Equal(t, expectedPipeline, actualPipeline)
+	assert.Equal(t, 1, len(actualPipeline.Stages))
+}
+
+func TestPipeline_SetAssembler(t *testing.T) {
+	expectedPipeline := &Pipeline{
+		Assembler: &Assembler{
+			Name: "org.osbuild.testassembler",
+		},
+	}
+	actualPipeline := &Pipeline{}
+	actualPipeline.SetAssembler(&Assembler{
+		Name: "org.osbuild.testassembler",
+	})
+	assert.Equal(t, expectedPipeline, actualPipeline)
+}