Release database connections after updating build statuses.

This commit is contained in:
Jake Moshenko 2017-07-26 12:04:04 -04:00
parent 543bf21e6b
commit 0011ddda16
3 changed files with 43 additions and 46 deletions

View file

@ -2,7 +2,7 @@ import unittest
from unittest import TestCase
from app import app
from data.model.build import get_repository_build, update_phase, create_repository_build
from data.model.build import get_repository_build, update_phase_then_close, create_repository_build
from initdb import setup_database_for_testing, finished_database_for_testing
from data import model
from data.database import RepositoryBuild, Image, ImageStorage, BUILD_PHASE
@ -83,7 +83,7 @@ class TestUpdatePhase(unittest.TestCase):
repo_build = get_repository_build(build.uuid)
self.assertEqual(repo_build.phase, BUILD_PHASE.WAITING)
self.assertTrue(update_phase(build.uuid, BUILD_PHASE.COMPLETE))
self.assertTrue(update_phase_then_close(build.uuid, BUILD_PHASE.COMPLETE))
repo_build = get_repository_build(build.uuid)
@ -91,7 +91,7 @@ class TestUpdatePhase(unittest.TestCase):
repo_build.delete_instance()
self.assertFalse(update_phase(repo_build.uuid, BUILD_PHASE.PULLING))
self.assertFalse(update_phase_then_close(repo_build.uuid, BUILD_PHASE.PULLING))
@staticmethod
def create_build(repository):