tests: Pretend to run a whole build

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/tests/build.py b/tests/build.py
index daac2c61d97d97fedc26389974384abeee015898..de159b241967a288a632f2ba814a2700d9bf1500 100755 (executable)
--- a/tests/build.py
+++ b/tests/build.py
@@ -94,6 +94,35 @@ class BuildTestCase(test.TestCase):
                # The score should now be 1
                self.assertEqual(build.score, 1)
 
+       async def test_run(self):
+               """
+                       This test creates a build and tries to emulate a job being completed.
+               """
+               path = self.source_path("tests/data/beep-1.3-2.ip3.src.pfm")
+
+               # Create the build
+               with self.db.transaction():
+                       build = await self._create_build(path)
+
+               # Assign all jobs to the default builder
+               for job in build.jobs:
+                       job.assign(self.builder)
+
+                       # This job should now be running
+                       self.assertTrue(job.is_running())
+
+                       # The builder should match
+                       self.assertEqual(job.builder, self.builder)
+
+               path = self.source_path("tests/data/beep-1.3-2.ip3.x86_64.pfm")
+
+               # Upload a package
+               upload = await self._create_upload(path)
+
+               # Pretend that all jobs finished
+               for job in build.jobs:
+                       await job.finished(success=True, packages=[upload])
+
 
 if __name__ == "__main__":
        unittest.main()

diff --git a/tests/test.py b/tests/test.py
index cd832777037cf4e21e435fb271c1f021664209ea..75b2989f8f8343cb00ff3851abb4e22b81e34bba 100644 (file)
--- a/tests/test.py
+++ b/tests/test.py
@@ -3,6 +3,7 @@
 import configparser
 import functools
 import os
+import shutil
 import socket
 import tempfile
 import unittest
@@ -14,6 +15,7 @@ from buildservice import builds
 from buildservice import database
 from buildservice import misc
 from buildservice import packages
+from buildservice import uploads
 
 class TestCase(unittest.IsolatedAsyncioTestCase):
        """
@@ -165,3 +167,32 @@ class TestCase(unittest.IsolatedAsyncioTestCase):
                self.assertIsInstance(build, builds.Build)
 
                return build
+
+       async def _create_upload(self, path, filename=None, user=None, **kwargs):
+               """
+                       Helper function to create an upload from path
+               """
+               if filename is None:
+                       filename = os.path.basename(path)
+
+               if user is None:
+                       user = self.user
+
+               # Allocate a new destination file
+               dst = self.backend.uploads.allocate_file()
+
+               # Open the source file
+               with open(path, "rb") as src:
+                       # Copy the entire content
+                       shutil.copyfileobj(src, dst)
+
+               # Close the destination file
+               dst.close()
+
+               # Create the upload object
+               upload = self.backend.uploads.create(filename, dst.name, user=user, **kwargs)
+
+               # Check if received the correct type
+               self.assertIsInstance(upload, uploads.Upload)
+
+               return upload