diff options
author | Patrick Williams <patrick@stwcx.xyz> | 2015-09-15 22:41:29 +0300 |
---|---|---|
committer | Patrick Williams <patrick@stwcx.xyz> | 2015-09-15 22:41:29 +0300 |
commit | 21f9b84b4b729fbd7acbd465e7a3f726e4d20f91 (patch) | |
tree | eb2d091d427ca0813b445509d59cc8e27e8ad25f /yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py | |
parent | 101cef31e2bf54c678501155cd2106251acbd076 (diff) | |
parent | c124f4f2e04dca16a428a76c89677328bc7bf908 (diff) | |
download | openbmc-21f9b84b4b729fbd7acbd465e7a3f726e4d20f91.tar.xz |
Merge commit 'c124f4f2e04dca16a428a76c89677328bc7bf908' as 'yocto-poky'
Diffstat (limited to 'yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py')
-rw-r--r-- | yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py | 190 |
1 files changed, 190 insertions, 0 deletions
diff --git a/yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py b/yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py new file mode 100644 index 000000000..5dbc77fda --- /dev/null +++ b/yocto-poky/bitbake/lib/toaster/bldcontrol/tests.py @@ -0,0 +1,190 @@ +""" +This file demonstrates writing tests using the unittest module. These will pass +when you run "manage.py test". + +Replace this with more appropriate tests for your application. +""" + +from django.test import TestCase + +from bldcontrol.bbcontroller import BitbakeController, BuildSetupException +from bldcontrol.localhostbecontroller import LocalhostBEController +from bldcontrol.sshbecontroller import SSHBEController +from bldcontrol.models import BuildEnvironment, BuildRequest +from bldcontrol.management.commands.runbuilds import Command + +import socket +import subprocess +import os + +# standard poky data hardcoded for testing +BITBAKE_LAYERS = [type('bitbake_info', (object,), { "giturl": "git://git.yoctoproject.org/poky.git", "dirpath": "", "commit": "HEAD"})] +POKY_LAYERS = [ + type('poky_info', (object,), { "name": "meta", "giturl": "git://git.yoctoproject.org/poky.git", "dirpath": "meta", "commit": "HEAD"}), + type('poky_info', (object,), { "name": "meta-yocto", "giturl": "git://git.yoctoproject.org/poky.git", "dirpath": "meta-yocto", "commit": "HEAD"}), + type('poky_info', (object,), { "name": "meta-yocto-bsp", "giturl": "git://git.yoctoproject.org/poky.git", "dirpath": "meta-yocto-bsp", "commit": "HEAD"}), + ] + + + +# we have an abstract test class designed to ensure that the controllers use a single interface +# specific controller tests only need to override the _getBuildEnvironment() method + +test_sourcedir = os.getenv("TTS_SOURCE_DIR") +test_builddir = os.getenv("TTS_BUILD_DIR") +test_address = os.getenv("TTS_TEST_ADDRESS", "localhost") + +if test_sourcedir == None or test_builddir == None or test_address == None: + raise Exception("Please set TTTS_SOURCE_DIR, TTS_BUILD_DIR and TTS_TEST_ADDRESS") + +# The bb server will expect a toaster-pre.conf file to exist. If it doesn't exit then we make +# an empty one here. +open(test_builddir + 'conf/toaster-pre.conf', 'a').close() + +class BEControllerTests(object): + + def _serverForceStop(self, bc): + err = bc._shellcmd("netstat -tapn 2>/dev/null | grep 8200 | awk '{print $7}' | sort -fu | cut -d \"/\" -f 1 | grep -v -- - | tee /dev/fd/2 | xargs -r kill") + self.assertTrue(err == '', "bitbake server pid %s not stopped" % err) + + def test_serverStartAndStop(self): + from bldcontrol.sshbecontroller import NotImplementedException + obe = self._getBuildEnvironment() + bc = self._getBEController(obe) + try: + # setting layers, skip any layer info + bc.setLayers(BITBAKE_LAYERS, POKY_LAYERS) + except NotImplementedException, e: + print "Test skipped due to command not implemented yet" + return True + # We are ok with the exception as we're handling the git already exists + except BuildSetupException: + pass + + bc.pokydirname = test_sourcedir + bc.islayerset = True + + hostname = test_address.split("@")[-1] + + # test start server and stop + bc.startBBServer() + + self.assertFalse(socket.socket(socket.AF_INET, socket.SOCK_STREAM).connect_ex((hostname, int(bc.be.bbport))), "Server not answering") + + bc.stopBBServer() + self.assertTrue(socket.socket(socket.AF_INET, socket.SOCK_STREAM).connect_ex((hostname, int(bc.be.bbport))), "Server not stopped") + + self._serverForceStop(bc) + + def test_getBBController(self): + from bldcontrol.sshbecontroller import NotImplementedException + obe = self._getBuildEnvironment() + bc = self._getBEController(obe) + layerSet = False + try: + # setting layers, skip any layer info + layerSet = bc.setLayers(BITBAKE_LAYERS, POKY_LAYERS) + except NotImplementedException: + print "Test skipped due to command not implemented yet" + return True + # We are ok with the exception as we're handling the git already exists + except BuildSetupException: + pass + + bc.pokydirname = test_sourcedir + bc.islayerset = True + + bbc = bc.getBBController() + self.assertTrue(isinstance(bbc, BitbakeController)) + bc.stopBBServer() + + self._serverForceStop(bc) + +class LocalhostBEControllerTests(TestCase, BEControllerTests): + def __init__(self, *args): + super(LocalhostBEControllerTests, self).__init__(*args) + + + def _getBuildEnvironment(self): + return BuildEnvironment.objects.create( + lock = BuildEnvironment.LOCK_FREE, + betype = BuildEnvironment.TYPE_LOCAL, + address = test_address, + sourcedir = test_sourcedir, + builddir = test_builddir ) + + def _getBEController(self, obe): + return LocalhostBEController(obe) + +class SSHBEControllerTests(TestCase, BEControllerTests): + def __init__(self, *args): + super(SSHBEControllerTests, self).__init__(*args) + + def _getBuildEnvironment(self): + return BuildEnvironment.objects.create( + lock = BuildEnvironment.LOCK_FREE, + betype = BuildEnvironment.TYPE_SSH, + address = test_address, + sourcedir = test_sourcedir, + builddir = test_builddir ) + + def _getBEController(self, obe): + return SSHBEController(obe) + + def test_pathExists(self): + obe = BuildEnvironment.objects.create(betype = BuildEnvironment.TYPE_SSH, address= test_address) + sbc = SSHBEController(obe) + self.assertTrue(sbc._pathexists("/")) + self.assertFalse(sbc._pathexists("/.deadbeef")) + self.assertTrue(sbc._pathexists(sbc._shellcmd("pwd"))) + + +class RunBuildsCommandTests(TestCase): + def test_bec_select(self): + """ + Tests that we can find and lock a build environment + """ + + obe = BuildEnvironment.objects.create(lock = BuildEnvironment.LOCK_FREE, betype = BuildEnvironment.TYPE_LOCAL) + command = Command() + bec = command._selectBuildEnvironment() + + # make sure we select the object we've just built + self.assertTrue(bec.be.id == obe.id, "Environment is not properly selected") + # we have a locked environment + self.assertTrue(bec.be.lock == BuildEnvironment.LOCK_LOCK, "Environment is not locked") + # no more selections possible here + self.assertRaises(IndexError, command._selectBuildEnvironment) + + def test_br_select(self): + from orm.models import Project, Release, BitbakeVersion, Branch + p = Project.objects.create_project("test", Release.objects.get_or_create(name = "HEAD", bitbake_version = BitbakeVersion.objects.get_or_create(name="HEAD", branch=Branch.objects.get_or_create(name="HEAD"))[0])[0]) + obr = BuildRequest.objects.create(state = BuildRequest.REQ_QUEUED, project = p) + command = Command() + br = command._selectBuildRequest() + + # make sure we select the object we've just built + self.assertTrue(obr.id == br.id, "Request is not properly selected") + # we have a locked environment + self.assertTrue(br.state == BuildRequest.REQ_INPROGRESS, "Request is not updated") + # no more selections possible here + self.assertRaises(IndexError, command._selectBuildRequest) + + +class UtilityTests(TestCase): + def test_reduce_path(self): + from bldcontrol.management.commands.loadconf import _reduce_canon_path, _get_id_for_sourcetype + + self.assertTrue( _reduce_canon_path("/") == "/") + self.assertTrue( _reduce_canon_path("/home/..") == "/") + self.assertTrue( _reduce_canon_path("/home/../ana") == "/ana") + self.assertTrue( _reduce_canon_path("/home/../ana/..") == "/") + self.assertTrue( _reduce_canon_path("/home/ana/mihai/../maria") == "/home/ana/maria") + + def test_get_id_for_sorucetype(self): + from bldcontrol.management.commands.loadconf import _reduce_canon_path, _get_id_for_sourcetype + self.assertTrue( _get_id_for_sourcetype("layerindex") == 1) + self.assertTrue( _get_id_for_sourcetype("local") == 0) + self.assertTrue( _get_id_for_sourcetype("imported") == 2) + with self.assertRaises(Exception): + _get_id_for_sourcetype("unknown") |