← Back to branch summary

~raxnetworking/nova/melange

« back to all changes in this revision

Viewing changes to melange/tests/unit/test_sqlalchemy_api.py

  • Committer: Rajaram Mallya
  • Date: 2011-09-05 13:19:18 UTC
  • Revision ID: rajarammallya@gmail.com-20110905131918-impvi6rmge75qk1x
Rajaram/Vinkesh|fixed more imports to import  only modules and misc style improvements

Show diffs side-by-side

added added

removed removed

Lines of Context:
melange/tests/unit/test_sqlalchemy_api.py
14
14
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15
15
#    License for the specific language governing permissions and limitations
16
16
#    under the License.
 
17
 
17
18
from melange.db.sqlalchemy import api as db_api
18
19
from melange.db.sqlalchemy import session
19
 
from melange.ipam.models import IpBlock
20
 
from melange.tests import BaseTest
21
 
from melange.tests import unit
22
 
from melange.tests.factories.models import IpBlockFactory
23
 
 
24
 
 
25
 
class TestSqlalchemyApi(BaseTest):
 
20
from melange.ipam import models
 
21
from melange import tests
 
22
from melange.tests import factories
 
23
 
 
24
 
 
25
class TestSqlalchemyApi(tests.BaseTest):
26
26
 
27
27
    def test_delete_does_soft_deletion(self):
28
 
        model = IpBlockFactory()
 
28
        model = factories.models.IpBlockFactory()
29
29
 
30
30
        db_api.delete(model)
31
31
 
32
 
        deleted_model = session.raw_query(IpBlock).get(model.id)
 
32
        deleted_model = session.raw_query(models.IpBlock).get(model.id)
33
33
        self.assertIsNotNone(deleted_model)
34
34
        self.assertTrue(deleted_model.deleted)
35
35
 
36
36
    def test_delete_all_does_soft_deletion(self):
37
 
        model1 = IpBlockFactory()
38
 
        model2 = IpBlockFactory()
 
37
        model1 = factories.models.IpBlockFactory()
 
38
        model2 = factories.models.IpBlockFactory()
39
39
 
40
 
        db_api.delete_all(IpBlock)
 
40
        db_api.delete_all(models.IpBlock)
41
41
 
42
42
        for model in [model1, model2]:
43
 
            deleted_model = session.raw_query(IpBlock).get(model.id)
 
43
            deleted_model = session.raw_query(models.IpBlock).get(model.id)
44
44
            self.assertIsNotNone(deleted_model)
45
45
            self.assertTrue(deleted_model.deleted)