test/cli: check a normal invocation

Check a normal, i.e. successful, invocation and ensure relevant
functions are called with the right parameters.

test/unit/test_cli.py

@@ -6,6 +6,7 @@
 import contextlib
 import io
 import koji
+import koji_cli.lib as kl
 from flexmock import flexmock
 
 from plugintest import PluginTest
@@ -14,6 +15,53 @@ from plugintest import PluginTest
 @PluginTest.load_plugin("cli")
 class TestCliPlugin(PluginTest):
 
+    @staticmethod
+    def mock_koji_lib(*, bg=False, task_result=0):
+        kojilib = flexmock(OptionParser=kl.OptionParser)
+
+        kojilib.should_receive("get_usage_str").and_return("usage")
+        kojilib.should_receive("activate_session").once()
+        kojilib.should_receive("_running_in_bg").and_return(bg)
+        kojilib.should_receive("watch_tasks").and_return(task_result)
+
+        return kojilib
+
+    @staticmethod
+    def mock_options(*, quiet=False):
+        options = flexmock(
+            quiet=quiet,
+            weburl="http://osbuild.org/"
+        )
+        return options
+
+
+    @staticmethod
+    def mock_session_add_valid_tag(session):
+        build_target = {
+            "build_tag": 23,
+            "build_tag_name": "target-build",
+            "dest_tag": 42,
+            "dest_tag_name": "target-dest"
+        }
+
+        tag_info = {
+            "id": build_target["dest_tag"],
+            "name": build_target["dest_tag_name"],
+            "locked": False
+        }
+
+        session.should_receive("getBuildTarget") \
+               .with_args("target") \
+               .and_return(build_target) \
+               .once()
+
+        session.should_receive("getTag") \
+               .with_args(build_target["dest_tag"]) \
+               .and_return(tag_info) \
+               .once()
+
+        return session
+
     def test_basic_invocation(self):
         # check we get the right amount of arguments
         # i.e. we are missing the architecture here
@@ -27,6 +75,35 @@ class TestCliPlugin(PluginTest):
             self.assertIn("osbuild-image", f.getvalue())
         f.close()
 
+        # check one successful invocation
+        argv = ["name", "version", "distro", "target", "arch1"]
+        task_result = {"compose_id": "42", "build_id": 23}
+        task_id = 1
+        koji_lib = self.mock_koji_lib()
+
+        options = self.mock_options()
+        session = flexmock()
+
+        self.mock_session_add_valid_tag(session)
+
+        session.should_receive("osbuildImage") \
+               .with_args(str, str, str, list, str, list, opts=dict) \
+               .and_return(task_id) \
+               .once()
+
+        session.should_receive("logout") \
+               .with_args() \
+               .once()
+
+        session.should_receive("getTaskResult") \
+               .with_args(task_id) \
+               .and_return(task_result) \
+               .once()
+
+        setattr(self.plugin, "kl", koji_lib)
+        r = self.plugin.handle_osbuild_image(options, session, argv)
+        self.assertEqual(r, 0)
+
     def test_target_check(self):
         # unknown build target
         session = flexmock()