From ff7696a92e6ac3fcac8cf7a3699c0bfd8a111dc6 Mon Sep 17 00:00:00 2001
From: Christian Kellner <christian@kellner.me>
Date: Fri, 15 Jan 2021 18:29:37 +0100
Subject: [PATCH] pipeline: return objects from `add` methods

Return the Assembler and Stage that got added from their respective
methods.
---
 osbuild/pipeline.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/osbuild/pipeline.py b/osbuild/pipeline.py
index b7136e34..45c61786 100644
--- a/osbuild/pipeline.py
+++ b/osbuild/pipeline.py
@@ -185,9 +185,11 @@ class Pipeline:
         self.stages.append(stage)
         if self.assembler:
             self.assembler.base = stage.id
+        return stage
 
     def set_assembler(self, name, options=None):
         self.assembler = Assembler(name, self.build, self.tree_id, options or {})
+        return self.assembler
 
     def build_stages(self, object_store, monitor, libdir):
         results = {"success": True}