embee/pyaccuwage
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

hopefully fix STaticField

Browse source
This commit is contained in:
Mark Riedesel 2021-09-02 17:40:35 -05:00
parent 7867a52a0c
commit 558e3fd232
2 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pyaccuwage/fields.py
View file

@ -198,12 +198,14 @@ class StaticField(TextField):
def __init__(self, name=None, required=True, value=None): def __init__(self, name=None, required=True, value=None):
super(StaticField, self).__init__(name=name, required=required, super(StaticField, self).__init__(name=name, required=required,
max_length=len(value)) max_length=len(value))
self._value = value self._static_value = value
def get_data(self):
return str(self._static_value).encode('ascii')[:self.max_length]
def parse(self, s): def parse(self, s):
pass pass
class BlankField(TextField): class BlankField(TextField):
is_read_only = True is_read_only = True

6
tests/test_fields.py
View file

@ -1,5 +1,6 @@
import unittest import unittest
from pyaccuwage.fields import TextField from pyaccuwage.fields import TextField
from pyaccuwage.fields import StaticField
# from pyaccuwage.fields import IntegerField # from pyaccuwage.fields import IntegerField
# from pyaccuwage.fields import StateField # from pyaccuwage.fields import StateField
# from pyaccuwage.fields import BlankField # from pyaccuwage.fields import BlankField
@ -31,3 +32,8 @@ class TestTextField(unittest.TestCase):
data = field.get_data() data = field.get_data()
self.assertEqual(len(data), field.max_length) self.assertEqual(len(data), field.max_length)
self.assertEqual(data, b'HELLO,') self.assertEqual(data, b'HELLO,')
class TestStaticField(unittest.TestCase):
def test_static_field(self):
field = StaticField(value='TEST')
self.assertEqual(field.get_data(), b'TEST')