test_import_pocsuite_execute.py

import unittest
from pocsuite3.api import init_pocsuite
from pocsuite3.api import start_pocsuite
from pocsuite3.api import get_results


class TestCase(unittest.TestCase):
    def setUp(self):
        self.config = {
            'url': 'http://127.0.0.1:8080',
            'poc': 'ecshop_rce',
        }
        init_pocsuite(self.config)

    def tearDown(self):
        pass

    def verify_result(self):
        result = get_results().pop()
        self.assertTrue(result[5] == 'success')

    @unittest.skip(reason='test')
    def test_import_run(self):
        start_pocsuite()
        self.verify_result()