def test_inject_file_cmd_exception(self): test = file_inject.FileInject() error, message = test.injectfile_cmd('Unencoded data') self.assertEqual(error, "500", "Did not get expected error code") self.assertEqual( message, "Error doing base64 decoding of data", "Did not get expected message on error" )
def test_inject_file_cmd(self): test = file_inject.FileInject() encode_details = base64.b64encode( b'/tmp/tests/test_file_write,Testing the inject' ) error, message = test.injectfile_cmd(encode_details) self.assertEqual(error, "0", "Did not get expected error code") self.assertEqual( message, "", "Did not get expected message on success" ) files = glob.glob('/tmp/tests/test_file*') self.assertEqual( len(files), 1, 'Did not find any files' ) found_file = False temp_contents = None for f in files: if '/tmp/tests/test_file_write' == f: with open(f) as temp_file: temp_contents = temp_file.read() found_file = True assert found_file, 'Did not find written file in expected path' self.assertEqual( temp_contents, 'Testing the inject', 'Written data in file is not what was expected' ) permissions = os.stat('/tmp/tests/test_file_write') try: readable_perms = stat.filemode(permissions.st_mode) except Exception: readable_perms = self.permissions_to_unix_name(permissions) self.assertEqual( readable_perms, '-rw-------', 'Permissions are not 600 as expected' )
def test_inject_file_cmd(self): test = file_inject.FileInject() encode_details = base64.b64encode( b'/tmp/tests/test_file_write,Testing the inject') error, message = test.injectfile_cmd(encode_details) self.assertEqual(error, "0", "Did not get expected error code") self.assertEqual(message, "", "Did not get expected message on success") files = glob.glob('/tmp/tests/test_file*') self.assertEqual(len(files), 1, 'Did not find any files') found_file = False temp_contents = None for f in files: if '/tmp/tests/test_file_write' == f: with open(f) as temp_file: temp_contents = temp_file.read() found_file = True assert found_file, 'Did not find written file in expected path' self.assertEqual(temp_contents, 'Testing the inject', 'Written data in file is not what was expected')
def test_instantiate_file_inject(self): file_inject.FileInject() assert True, 'Class did not generate exception'