vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Making some refactors to make it easier to cancel the build at any time.

Browse source
This commit is contained in:
Charlton Austin 2016-10-21 10:05:17 -04:00
parent 42ed8522fd
commit 1cde22e76c
3 changed files with 86 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

54
test/test_queries.py
View file

@ -1,13 +1,16 @@
import unittest
from unittest import TestCase
from app import app
from data.model.build import get_repository_build, update_phase, create_repository_build
from initdb import setup_database_for_testing, finished_database_for_testing
from data import model
from data.database import RepositoryBuild, Repository, Image, ImageStorage
from data.database import RepositoryBuild, Image, ImageStorage, BUILD_PHASE
ADMIN_ACCESS_USER = 'devtable'
SIMPLE_REPO = 'simple'
class TestSpecificQueries(unittest.TestCase):
def setUp(self):
setup_database_for_testing(self)
@ -63,6 +66,53 @@ class TestSpecificQueries(unittest.TestCase):
self.fail('Expected BlobDoesNotExist exception')
class TestUpdatePhase(unittest.TestCase):
def setUp(self):
setup_database_for_testing(self)
self.app = app.test_client()
self.ctx = app.test_request_context()
self.ctx.__enter__()
def tearDown(self):
finished_database_for_testing(self)
self.ctx.__exit__(True, None, None)
def testUpdatePhase(self):
build = self.create_build(model.repository.get_repository("devtable", "building"))
repo_build = get_repository_build(build.uuid)
self.assertEqual(repo_build.phase, BUILD_PHASE.WAITING)
self.assertTrue(update_phase(build.uuid, BUILD_PHASE.COMPLETE))
repo_build = get_repository_build(build.uuid)
self.assertEqual(repo_build.phase, BUILD_PHASE.COMPLETE)
repo_build.delete_instance()
self.assertFalse(update_phase(repo_build.uuid, BUILD_PHASE.PULLING))
@staticmethod
def create_build(repository):
new_token = model.token.create_access_token(repository, 'write', 'build-worker')
repo = 'ci.devtable.com:5000/%s/%s' % (repository.namespace_user.username, repository.name)
job_config = {
'repository': repo,
'docker_tags': ['latest'],
'build_subdir': '',
'trigger_metadata': {
'commit': '3482adc5822c498e8f7db2e361e8d57b3d77ddd9',
'ref': 'refs/heads/master',
'default_branch': 'master'
}
}
build = create_repository_build(repository, new_token, job_config,
'68daeebd-a5b9-457f-80a0-4363b882f8ea',
"build_name")
build.save()
return build
if __name__ == '__main__':
unittest.main()
unittest.main()