1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
|
from sprox.fillerbase import FillerBase, TableFiller, EditFormFiller, AddFormFiller, FormFiller, ConfigBaseError
from sprox.test.base import setup_database, sorted_user_columns, SproxTest, User, Example
from nose.tools import raises, eq_
session = None
engine = None
connection = None
trans = None
def setup():
global session, engine, metadata, trans
session, engine, metadata = setup_database()
class UserFiller(TableFiller):
__entity__ = User
class TestFillerBase(SproxTest):
def setup(self):
super(TestFillerBase, self).setup()
class UserFiller(FillerBase):
__entity__ = User
self.filler = UserFiller(session)
def test_get_value(self):
value = self.filler.get_value()
assert value =={}, value
class TestTableFiller(SproxTest):
def setup(self):
super(TestTableFiller, self).setup()
self.filler = UserFiller(session)
def test_create(self):
pass
def test_get_value(self):
value = self.filler.get_value()
eq_(len(value), 1)
value = value[0]
eq_(value['groups'], u'4')
eq_(value['town'], 'Arvada')
def test_get_value_with_binary_field(self):
class ExampleFiller(TableFiller):
__entity__ = Example
example = Example(binary='datadatadata')
session.add(example)
filler = ExampleFiller(session)
value = filler.get_value()
eq_(value[0]['binary'], '<file>')
def test_get_list_data_value_array_values(self):
r = self.filler._get_list_data_value(User, ['something', 'something else'])
assert r == ['something', 'something else'], r
@raises(ConfigBaseError)
def test_count_without_get(self):
self.filler.get_count()
def test_count(self):
self.filler.get_value()
c = self.filler.get_count()
assert c == 1, c
class TestEditFormFiller(SproxTest):
def setup(self):
super(TestEditFormFiller, self).setup()
self.filler = EditFormFiller(session)
self.filler.__entity__ = User
def test_create(self):
pass
def test_get_value(self):
value = self.filler.get_value(values={'user_id':1})
eq_(value['groups'], [5])
eq_(value['town'], 1)
class TestAddFormFiller(SproxTest):
def setup(self):
super(TestAddFormFiller, self).setup()
self.filler = AddFormFiller(session)
self.filler.__entity__ = User
def test_create(self):
pass
def test_get_value(self):
value = self.filler.get_value(values={'user_id':1})
eq_(value['user_id'], 1)
|