From 558e3fd23286059d07eb3b03d43cc3fe42ca55b9 Mon Sep 17 00:00:00 2001 From: Mark Riedesel Date: Thu, 2 Sep 2021 17:40:35 -0500 Subject: [PATCH] hopefully fix STaticField --- pyaccuwage/fields.py | 6 ++++-- tests/test_fields.py | 6 ++++++ 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/pyaccuwage/fields.py b/pyaccuwage/fields.py index 0d6fcd3..27c10dc 100644 --- a/pyaccuwage/fields.py +++ b/pyaccuwage/fields.py @@ -198,12 +198,14 @@ class StaticField(TextField): def __init__(self, name=None, required=True, value=None): super(StaticField, self).__init__(name=name, required=required, 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): pass - class BlankField(TextField): is_read_only = True diff --git a/tests/test_fields.py b/tests/test_fields.py index 3d8fd3e..2d32755 100644 --- a/tests/test_fields.py +++ b/tests/test_fields.py @@ -1,5 +1,6 @@ import unittest from pyaccuwage.fields import TextField +from pyaccuwage.fields import StaticField # from pyaccuwage.fields import IntegerField # from pyaccuwage.fields import StateField # from pyaccuwage.fields import BlankField @@ -31,3 +32,8 @@ class TestTextField(unittest.TestCase): data = field.get_data() self.assertEqual(len(data), field.max_length) 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')