79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
# standard imports
|
|
import unittest
|
|
|
|
# local imports
|
|
from shep import State
|
|
from shep.error import (
|
|
StateExists,
|
|
StateItemExists,
|
|
StateInvalid,
|
|
StateItemNotFound,
|
|
)
|
|
|
|
|
|
class TestStateItems(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.states = State(4)
|
|
self.states.add('foo')
|
|
self.states.add('bar')
|
|
self.states.add('baz')
|
|
self.states.alias('xyzzy', self.states.BAZ | self.states.BAR)
|
|
self.states.alias('plugh', self.states.FOO | self.states.BAR)
|
|
|
|
|
|
def test_put(self):
|
|
item = b'foo'
|
|
|
|
# put in initial (no) state
|
|
self.states.put(item)
|
|
|
|
with self.assertRaises(StateItemExists):
|
|
self.states.put(item)
|
|
|
|
with self.assertRaises(StateItemExists):
|
|
self.states.put(item, self.states.BAZ)
|
|
|
|
|
|
def test_item_state(self):
|
|
item = b'foo'
|
|
self.states.put(item, self.states.XYZZY)
|
|
self.assertEqual(self.states.state(item), self.states.XYZZY)
|
|
|
|
|
|
def test_item_move(self):
|
|
item = b'foo'
|
|
self.states.put(item, self.states.FOO)
|
|
self.states.move(item, self.states.BAR)
|
|
self.assertEqual(self.states.state(item), self.states.BAR)
|
|
|
|
|
|
def test_item_move_from_alias(self):
|
|
item = b'foo'
|
|
self.states.put(item, self.states.FOO)
|
|
self.states.move(item, self.states.XYZZY)
|
|
self.assertEqual(self.states.state(item), self.states.XYZZY)
|
|
self.states.move(item, self.states.BAR)
|
|
self.assertEqual(self.states.state(item), self.states.BAR)
|
|
|
|
|
|
def test_item_move_from_new(self):
|
|
item = b'foo'
|
|
self.states.put(item)
|
|
self.assertEqual(self.states.state(item), self.states.NEW)
|
|
self.states.move(item, self.states.XYZZY)
|
|
self.assertEqual(self.states.state(item), self.states.XYZZY)
|
|
|
|
|
|
def test_item_purge(self):
|
|
item = b'foo'
|
|
self.states.put(item, self.states.BAZ)
|
|
self.assertEqual(self.states.state(item), self.states.BAZ)
|
|
self.states.purge(item)
|
|
with self.assertRaises(StateItemNotFound):
|
|
self.states.state(item)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|