def test_correct_number_of_fields(self): b = Builder() b.set_template(TMPL_TWO_FIELDS) b.prepare() self.assertEqual(2, b.count())
def test_get_raw(self): b = Builder() b.set_template("foo[1] = 1, foe[2] = 2") b.prepare() self.assertTrue(b.is_correct()) target = "\x01\x00\x02" self.assertEqual(target, b.get_raw())
def test_correct_sizes_of_fields(self): b = Builder() b.set_template(TMPL_TWO_FIELDS) b.prepare() fields = b.get_fields() self.assertEqual(1, fields["foo"].size) self.assertEqual(2, fields["foe"].size)
def test_correct_names_of_fields(self): b = Builder() b.set_template(TMPL_TWO_FIELDS) b.prepare() fields = b.get_fields() self.assertEqual(2, len(fields)) self.assertTrue("foo" in fields) self.assertTrue("foe" in fields)
def test_is_correct(self): b = Builder() b.set_template(TMPL_TWO_FIELDS) b.prepare() self.assertFalse(b.is_correct()) b = Builder() b.set_template("foo[1] = 128") b.prepare() self.assertTrue(b.is_correct()) b = Builder() b.set_template("foo[1] = 0, foe[2], bar[1] = 2") b.prepare() self.assertFalse(b.is_correct()) b.fields["foe"].value = 3 self.assertTrue(b.is_correct())
def test_set_template(self): b = Builder() b.set_template(TMPL_TWO_FIELDS) self.assertEqual(TMPL_TWO_FIELDS, b.template)