def test_dagda_check_full_happy_path(self): sys.argv = ['dagda.py', 'check', '-i', 'jboss/wildfly'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'check') self.assertEqual(parsed_args.get_extra_args().get_docker_image_name(), 'jboss/wildfly') self.assertEqual(parsed_args.get_extra_args().get_container_id(), None)
def test_dagda_monitor_full_happy_path(self): sys.argv = ['dagda.py', 'monitor', '69dbf26ab368', '--start'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'monitor') self.assertEqual(parsed_args.get_extra_args().get_container_id(), '69dbf26ab368') self.assertTrue(parsed_args.get_extra_args().is_start()) self.assertFalse(parsed_args.get_extra_args().is_stop())
def test_dagda_agent_full_happy_path(self): sys.argv = ['dagda.py', 'agent', 'localhost:5000', '-i', 'alpine'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'agent') self.assertEqual(parsed_args.get_extra_args().get_dagda_server(), 'localhost:5000') self.assertEqual(parsed_args.get_extra_args().get_docker_image_name(), 'alpine') self.assertEqual(parsed_args.get_extra_args().get_container_id(), None)
def test_dagda_history_full_happy_path(self): sys.argv = ['dagda.py', 'history', 'jboss/wildfly'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'history') self.assertEqual(parsed_args.get_extra_args().get_docker_image_name(), 'jboss/wildfly') self.assertEqual(parsed_args.get_extra_args().get_report_id(), None)
def test_dagda_start_full_happy_path(self): sys.argv = ['dagda.py', 'start', '--server_host', 'localhost', '--server_port', '5555'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'start') self.assertEqual(parsed_args.get_extra_args().get_server_host(), 'localhost') self.assertEqual(parsed_args.get_extra_args().get_server_port(), 5555) self.assertEqual(parsed_args.get_extra_args().get_mongodb_host(), None) self.assertEqual(parsed_args.get_extra_args().get_mongodb_port(), None)
def test_dagda_vuln_full_happy_path(self): sys.argv = ['dagda.py', 'vuln', '--product', 'openldap', '--product_version', '2.2.20'] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'vuln') self.assertEqual(parsed_args.get_extra_args().get_product(), 'openldap') self.assertEqual(parsed_args.get_extra_args().get_product_version(), '2.2.20') self.assertFalse(parsed_args.get_extra_args().is_initialization_required()) self.assertFalse(parsed_args.get_extra_args().is_init_status_requested())
def test_dagda_docker_happy_path(self): sys.argv = [ 'dagda.py', 'docker', 'images', ] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'docker') self.assertEqual(parsed_args.get_extra_args().get_command(), 'images')
def test_dagda_history_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'history'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/history')
def test_dagda_monitor_stop_full_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'monitor', '69dbf26ab368', '--stop'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/monitor/containers/69dbf26ab368/stop')
def test_dagda_check_container_full_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'check', '-c', '697f6f235558'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/check/containers/697f6f235558')
def test_dagda_vuln_product_version_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'vuln', '--product', 'openldap', '--product_version', '2.2.20'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/vuln/products/openldap/2.2.20')
def test_dagda_vuln_exploit_info_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'vuln', '--exploit_db_info', '2002'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/vuln/exploit/2002/details')
def test_dagda_history_is_fp_with_version_full_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = ['dagda.py', 'history', 'jboss/wildfly', '--is_fp', 'openldap:2.2.20'] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual(err, '/v1/history/jboss/wildfly/fp/openldap/2.2.20')
def test_dagda_docker_events_happy_path(self): os.environ['DAGDA_HOST'] = str('localhost') os.environ['DAGDA_PORT'] = str('50000') sys.argv = [ 'dagda.py', 'docker', 'events', '--event_from', 'FROM', '--event_type', 'TYPE', '--event_action', 'ACTION' ] parsed_args = DagdaCLIParser() try: execute_dagda_cmd(parsed_args.get_command(), parsed_args.get_extra_args()) self.fail() except requests.exceptions.ConnectionError as err: err = DagdaCLITestSuite._get_path(err) self.assertEqual( err, '/v1/docker/events?event_action=ACTION&event_from=FROM&event_type=TYPE' )
def test_dagda_docker_happy_path(self): sys.argv = ['dagda.py', 'docker', 'images', ] parsed_args = DagdaCLIParser() self.assertEqual(parsed_args.get_command(), 'docker') self.assertEqual(parsed_args.get_extra_args().get_command(), 'images')