greenbone / ospd

        params = response.findall('vts/vt/vt_params/vt_param')

        self.assertEqual(2, len(params))

    def testGetVTs_VTs_with_refs(self):

        daemon = DummyWrapper([])

        daemon.add_vt('1.2.3.4',

                      'A vulnerability test',

                      vt_params="a",

                      custom="b",

                      vt_refs="c")

        response = secET.fromstring(

            daemon.handle_command('<get_vts vt_id="1.2.3.4"></get_vts>'))

        # The status of the response must be success (i.e. 200)

        self.assertEqual(response.get('status'), '200')

        # The response root element must have the correct name

        self.assertEqual(response.tag, 'get_vts_response')

        # The response must contain a 'vts' element

        self.assertIsNotNone(response.find('vts'))

        vt_params = response[0][0].findall('vt_params')

        self.assertEqual(1, len(vt_params))

        custom = response[0][0].findall('custom')

        self.assertEqual(1, len(custom))

        refs = response.findall('vts/vt/vt_refs/ref')

        self.assertEqual(2, len(refs))

    def testGetVTs_VTs_with_dependencies(self):

        daemon = DummyWrapper([])

        daemon.add_vt('123456789', 'Yet another vulnerability test')

        response = secET.fromstring(daemon.handle_command('<get_vts />'))

    def testGetVTs_VTs_with_params(self):

        daemon = DummyWrapper([])

        daemon.add_vt('1.2.3.4', 'A vulnerability test', vt_params="a", custom="b")

        response = secET.fromstring(daemon.handle_command('<get_vts vt_id="1.2.3.4"></get_vts>'))

        # The status of the response must be success (i.e. 200)

        self.assertEqual(response.get('status'), '200')

        # The response root element must have the correct name

        self.assertEqual(response.tag, 'get_vts_response')

        # The response must contain a 'scanner_params' element

        self.assertIsNotNone(response.find('vts'))

        vt_params = response[0][0].findall('vt_params')

        self.assertEqual(1, len(vt_params))

        custom = response[0][0].findall('custom')

        self.assertEqual(1, len(custom))

        params = response.findall('vts/vt/vt_params/vt_param')

        self.assertEqual(2, len(params))

    def testGetVTs_VTs_with_refs(self):

        daemon = DummyWrapper([])