testBCDataStream.py

#
# Unit tests for BCDataStream class
#

import unittest

import BCDataStream

class Tests(unittest.TestCase):
  def setUp(self):
    self.ds = BCDataStream.BCDataStream()

  def testString(self):
    t = {
      "\x07setting" : "setting",
      "\xfd\x00\x07setting" : "setting",
      "\xfe\x00\x00\x00\x07setting" : "setting",
      }
    for (input, output) in t.iteritems():
      self.ds.clear()
      self.ds.write(input)
      got = self.ds.read_string()
      self.assertEqual(output, got)

if __name__ == "__main__":
  unittest.main()