aljaz/cowrie
1
0
Fork 0
mirror of https://github.com/aljazceru/cowrie.git synced 2026-02-21 06:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

convert testing framework to python3

Browse Source
This commit is contained in:
Michel Oosterhof
2016-08-16 12:15:44 +00:00
parent 66384911c7
commit 4a307e0c38
3 changed files with 56 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -26,7 +26,7 @@ Additional functionality over standard kippo:
Software required:
* Python 2.7+
* Python 2.7+, 3.3+
* Zope Interface 3.6.0+
* Twisted 8.0+
* python-crypto

8
cowrie/test/fake_transport.py
View File

@@ -40,13 +40,13 @@ class FakeTransport(proto_helpers.StringTransport):
width = 80
height = 24
void = object()
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, N_COLORS = range(9)
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, N_COLORS = list(range(9))
for keyID in ('UP_ARROW', 'DOWN_ARROW', 'RIGHT_ARROW', 'LEFT_ARROW',
'HOME', 'INSERT', 'DELETE', 'END', 'PGUP', 'PGDN',
'F1', 'F2', 'F3', 'F4', 'F5', 'F6', 'F7', 'F8', 'F9',
'F10', 'F11', 'F12'):
exec '%s = object()' % (keyID,)
exec('%s = object()' % (keyID,))
TAB = '\x09'
BACKSPACE = '\x08'
@@ -136,7 +136,7 @@ class FakeTransport(proto_helpers.StringTransport):
def eraseDisplay(self):
self.lines = [self._emptyLine(self.width) for i in xrange(self.height)]
self.lines = [self._emptyLine(self.width) for i in range(self.height)]
def _currentFormattingState(self):
@@ -146,5 +146,5 @@ class FakeTransport(proto_helpers.StringTransport):
def _emptyLine(self, width):
return [(self.void, self._currentFormattingState())
for i in xrange(width)]
for i in range(width)]

102
cowrie/test/test_base_commands.py
View File

@@ -7,8 +7,8 @@ from twisted.trial import unittest
from cowrie.core import protocol
from cowrie.core import config
import fake_server
import fake_transport
from . import fake_server
from . import fake_transport
class ShellBaseCommandsTests(unittest.TestCase):
@@ -22,179 +22,179 @@ class ShellBaseCommandsTests(unittest.TestCase):
def test_whoami_command(self):
self.proto.lineReceived('whoami \n')
print self.tr.value()
print(self.tr.value())
def test_users_command(self):
self.proto.lineReceived('users \n')
print self.tr.value()
print(self.tr.value())
def test_help_command(self):
self.proto.lineReceived('help \n')
print self.tr.value()
print(self.tr.value())
def test_w_command(self):
self.proto.lineReceived('w \n')
print self.tr.value()
print(self.tr.value())
def test_who_command(self):
self.proto.lineReceived('who \n')
print self.tr.value()
print(self.tr.value())
def test_echo_command(self):
self.proto.lineReceived('echo "test worked correctly" \n')
print self.tr.value()
print(self.tr.value())
def test_exit_command(self):
self.proto.lineReceived('exit \n')
print self.tr.value()
print(self.tr.value())
def test_logout_command(self):
self.proto.lineReceived('logout \n')
print self.tr.value()
print(self.tr.value())
def test_clear_command(self):
self.proto.lineReceived('clear \n')
print self.tr.value()
print(self.tr.value())
def test_reset_command(self):
self.proto.lineReceived('hostname unitChanged\n')
print self.tr.value()
print(self.tr.value())
def test_hostname_command(self):
self.proto.lineReceived('hostname unitChanged\n')
print self.tr.value()
print(self.tr.value())
def test_ps_command(self):
self.proto.lineReceived('ps\n')
print self.tr.value()
print(self.tr.value())
def test_id_command(self):
self.proto.lineReceived('id\n')
print self.tr.value()
print(self.tr.value())
def test_passwd_command(self):
self.proto.lineReceived('passwd\n')
self.proto.lineReceived('changeme\n')
self.proto.lineReceived('changeme\n')
print self.tr.value()
print(self.tr.value())
def test_shutdown_command(self):
self.proto.lineReceived('shutdown\n')
print self.tr.value()
print(self.tr.value())
def test_poweroff_command(self):
self.proto.lineReceived('poweroff\n')
print self.tr.value()
print(self.tr.value())
def test_history_command(self):
self.proto.lineReceived('history\n')
print self.tr.value()
print(self.tr.value())
def test_date_command(self):
self.proto.lineReceived('date\n')
print self.tr.value()
print(self.tr.value())
def test_bash_command(self):
self.proto.lineReceived('bash\n')
print self.tr.value()
print(self.tr.value())
def test_sh_command(self):
self.proto.lineReceived('sh\n')
print self.tr.value()
print(self.tr.value())
def test_php_command(self):
self.proto.lineReceived('php\n')
print self.tr.value()
print(self.tr.value())
def test_chattr_command(self):
self.proto.lineReceived('chattr\n')
print self.tr.value()
print(self.tr.value())
def test_umask_command(self):
self.proto.lineReceived('umask\n')
print self.tr.value()
print(self.tr.value())
def test_set_command(self):
self.proto.lineReceived('set\n')
print self.tr.value()
print(self.tr.value())
def test_unset_command(self):
self.proto.lineReceived('unset\n')
print self.tr.value()
print(self.tr.value())
def test_export_command(self):
self.proto.lineReceived('export\n')
print self.tr.value()
print(self.tr.value())
def test_alias_command(self):
self.proto.lineReceived('alias\n')
print self.tr.value()
print(self.tr.value())
def test_jobs_command(self):
self.proto.lineReceived('jobs\n')
print self.tr.value()
print(self.tr.value())
def test_kill_command(self):
self.proto.lineReceived('/bin/kill\n')
print self.tr.value()
print(self.tr.value())
def test_pkill_command(self):
self.proto.lineReceived('/bin/pkill\n')
print self.tr.value()
print(self.tr.value())
def test_killall_command(self):
self.proto.lineReceived('/bin/killall\n')
print self.tr.value()
print(self.tr.value())
def test_killall5_command(self):
self.proto.lineReceived('/bin/killall5\n')
print self.tr.value()
print(self.tr.value())
def test_su_command(self):
self.proto.lineReceived('su\n')
print self.tr.value()
print(self.tr.value())
def test_chown_command(self):
self.proto.lineReceived('chown\n')
print self.tr.value()
print(self.tr.value())
def test_chgrp_command(self):
self.proto.lineReceived('chgrp\n')
print self.tr.value()
print(self.tr.value())
def tearDown(self):
@@ -212,63 +212,63 @@ class ShellFileCommandsTests(unittest.TestCase):
def test_cat_output(self):
self.proto.lineReceived('cat /proc/cpuinfo')
print self.tr.value()
print(self.tr.value())
def test_grep_output(self):
self.proto.lineReceived('grep cpu /proc/cpuinfo')
print self.tr.value()
print(self.tr.value())
def test_tail_output(self):
self.proto.lineReceived('tail -n 10 /proc/cpuinfo')
print self.tr.value()
print(self.tr.value())
def test_head_output(self):
self.proto.lineReceived('head -n 10 /proc/cpuinfo')
print self.tr.value()
print(self.tr.value())
def test_cd_output(self):
self.proto.lineReceived('cd /usr/bin')
print self.tr.value()
print(self.tr.value())
def test_rm_output(self):
self.proto.lineReceived('rm /usr/bin/gcc')
print self.tr.value()
print(self.tr.value())
def test_cp_output(self):
self.proto.lineReceived('cp /usr/bin/gcc /tmp')
print self.tr.value()
print(self.tr.value())
def test_mv_output(self):
self.proto.lineReceived('mv /usr/bin/gcc /tmp')
print self.tr.value()
print(self.tr.value())
def test_mkdir_output(self):
self.proto.lineReceived('mkdir /tmp/hello')
print self.tr.value()
print(self.tr.value())
def test_rmdir_output(self):
self.proto.lineReceived('mkdir /tmp/blah')
self.proto.lineReceived('rmdir /tmp/blah')
print self.tr.value()
print(self.tr.value())
def test_pwd_output(self):
self.proto.lineReceived('pwd')
print self.tr.value()
print(self.tr.value())
def test_touch_output(self):
self.proto.lineReceived('touch unittests.txt')
print self.tr.value()
print(self.tr.value())
def tearDown(self):
@@ -286,12 +286,12 @@ class ShellPipeCommandsTests(unittest.TestCase):
def test_shell_pipe_with_ls_tail(self):
self.proto.lineReceived('ls -la | tail -n 2\n')
print self.tr.value()
print(self.tr.value())
def test_shell_pipe_with_cat_head(self):
self.proto.lineReceived('cat /proc/cpuinfo | head -n 5 \n')
print self.tr.value()
print(self.tr.value())
def test_shell_busybox_with_cat_and_sudo_grep(self):