Revision bb8fb9af snf-cyclades-app/synnefo/db/tests.py
b/snf-cyclades-app/synnefo/db/tests.py | ||
---|---|---|
38 | 38 |
|
39 | 39 |
from synnefo.db.models import * |
40 | 40 |
from synnefo.db import models_factory as mfact |
41 |
from synnefo.db.pools import IPPool |
|
41 |
from synnefo.db.pools import IPPool, EmptyPool
|
|
42 | 42 |
|
43 | 43 |
from django.db import IntegrityError |
44 | 44 |
from django.core.exceptions import MultipleObjectsReturned |
... | ... | |
202 | 202 |
|
203 | 203 |
class BridgePoolTest(TestCase): |
204 | 204 |
def test_no_pool(self): |
205 |
self.assertRaises(BridgePoolTable.DoesNotExist,
|
|
205 |
self.assertRaises(EmptyPool,
|
|
206 | 206 |
BridgePoolTable.get_pool) |
207 | 207 |
|
208 | 208 |
def test_two_pools(self): |
Also available in: Unified diff