add basic cli test to cover headers

jasonrbriggs · Jan 25, 2016 · eab730d · eab730d
1 parent b206a88
commit eab730d
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/stomp/test/cli_test.py b/stomp/test/cli_test.py
@@ -86,6 +86,24 @@ def testsendfile(self):
         teststdout.expect('Shutting down, please wait')
         cli.onecmd('quit')
 
+    def testsendfileheaders(self):
+        f = create_test_file()
+
+        teststdin = TestStdin()
+        teststdout = TestStdout(self)
+        teststdout.expect('CONNECTED')
+
+        cli = StompCLI(host, port, username, password, '1.0', stdin=teststdin, stdout=teststdout)
+
+        time.sleep(3)
+
+        cli.onecmd('sendfile /queue/testsendfile %s { "custom" : "header" }' % f.name)
+
+        time.sleep(3)
+
+        teststdout.expect('Shutting down, please wait')
+        cli.onecmd('quit')
+
     def testabort(self):
         teststdin = TestStdin()
         teststdout = TestStdout(self)