test coverage: generators and websockets to 100%

1 files changed, 11 insertions, 0 deletions

diff --git a/test/test_language_generators.py b/test/test_language_generators.py
index 3db1ac91..854b8485 100644
--- a/test/test_language_generators.py
+++ b/test/test_language_generators.py
@@ -36,3 +36,14 @@ def test_filegenerator():
         # remove all references to FileGenerator instance to close the file
         # handle.
         del g
+
+
+def test_transform_generator():
+    def trans(offset, data):
+        return "a" * len(data)
+    g = generators.LiteralGenerator("one")
+    t = generators.TransformGenerator(g, trans)
+    assert len(t) == len(g)
+    assert t[0] == "a"
+    assert t[:] == "a" * len(g)
+    assert repr(t)