Revision 3e7c63f8 snf-cyclades-app/synnefo/db/pools/tests.py
| b/snf-cyclades-app/synnefo/db/pools/tests.py | ||
|---|---|---|
| 1 | 1 |
import sys |
| 2 | 2 |
from synnefo.db.pools import (PoolManager, EmptyPool, BridgePool, |
| 3 | 3 |
MacPrefixPool, IPPool) |
| 4 |
from bitarray import bitarray |
|
| 4 | 5 |
|
| 5 | 6 |
# Use backported unittest functionality if Python < 2.7 |
| 6 | 7 |
try: |
| ... | ... | |
| 34 | 35 |
def test_created_pool(self): |
| 35 | 36 |
obj = DummyObject(42) |
| 36 | 37 |
pool = DummyPool(obj) |
| 37 |
self.assertEqual(pool.to_01(), '1' * 42 + '0' * 6) |
|
| 38 |
self.assertEqual(pool.to_map(), '.' * 42 + 'X' * 6) |
|
| 38 |
self.assertEqual(pool.to_01(), '1' * 42) |
|
| 39 |
self.assertEqual(pool.to_map(), '.' * 42) |
|
| 40 |
self.assertEqual(pool.available, bitarray('1' * 42 + '0' * 6))
|
|
| 41 |
self.assertEqual(pool.reserved, bitarray('1' * 42 + '0' * 6))
|
|
| 39 | 42 |
|
| 40 | 43 |
def test_save_pool(self): |
| 41 | 44 |
obj = DummyObject(42) |
| ... | ... | |
| 89 | 92 |
pool.put(0) |
| 90 | 93 |
self.assertEqual(pool.get(), 1) |
| 91 | 94 |
|
| 95 |
def test_extend_pool(self): |
|
| 96 |
obj = DummyObject(42) |
|
| 97 |
pool = DummyPool(obj) |
|
| 98 |
pool.extend(7) |
|
| 99 |
self.assertEqual(pool.to_01(), '1' * 49) |
|
| 100 |
self.assertEqual(pool.to_map(), '.' * 49) |
|
| 101 |
self.assertEqual(pool.available, bitarray('1' * 49 + '0' * 7))
|
|
| 102 |
self.assertEqual(pool.reserved, bitarray('1' * 49 + '0' * 7))
|
|
| 103 |
|
|
| 104 |
def test_shrink_pool(self): |
|
| 105 |
obj = DummyObject(42) |
|
| 106 |
pool = DummyPool(obj) |
|
| 107 |
pool.shrink(3) |
|
| 108 |
self.assertEqual(pool.to_01(), '1' * 39) |
|
| 109 |
self.assertEqual(pool.to_map(), '.' * 39) |
|
| 110 |
self.assertEqual(pool.available, bitarray('1' * 39 + '0' * 1))
|
|
| 111 |
self.assertEqual(pool.reserved, bitarray('1' * 39 + '0' * 1))
|
|
| 112 |
|
|
| 92 | 113 |
|
| 93 | 114 |
class BridgePoolTestCase(unittest.TestCase): |
| 94 | 115 |
def test_bridge_conversion(self): |
| ... | ... | |
| 96 | 117 |
obj.base = "bridge" |
| 97 | 118 |
pool = BridgePool(obj) |
| 98 | 119 |
for i in range(0, 13): |
| 99 |
self.assertEqual("bridge" + str(i), pool.get())
|
|
| 120 |
self.assertEqual("bridge" + str(i + 1), pool.get())
|
|
| 100 | 121 |
pool.put("bridge2")
|
| 101 | 122 |
pool.put("bridge6")
|
| 102 | 123 |
self.assertEqual("bridge2", pool.get())
|
Also available in: Unified diff