greenbone /
python-gvm
| @@ 24-182 (lines=159) @@ | ||
| 21 | from gvm.errors import RequiredArgument, InvalidArgumentType |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyPolicySetFamilySelectionTestCase: |
|
| 25 | def test_modify_policy_set_family_selection(self): |
|
| 26 | self.gmp.modify_policy_set_family_selection( |
|
| 27 | policy_id='c1', families=[('foo', True, True)] |
|
| 28 | ) |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<modify_config config_id="c1">' |
|
| 32 | '<family_selection>' |
|
| 33 | '<growing>1</growing>' |
|
| 34 | '<family>' |
|
| 35 | '<name>foo</name>' |
|
| 36 | '<all>1</all>' |
|
| 37 | '<growing>1</growing>' |
|
| 38 | '</family>' |
|
| 39 | '</family_selection>' |
|
| 40 | '</modify_config>' |
|
| 41 | ) |
|
| 42 | ||
| 43 | self.gmp.modify_policy_set_family_selection( |
|
| 44 | policy_id='c1', families=(('foo', True, True), ('bar', True, True)) |
|
| 45 | ) |
|
| 46 | ||
| 47 | self.connection.send.has_been_called_with( |
|
| 48 | '<modify_config config_id="c1">' |
|
| 49 | '<family_selection>' |
|
| 50 | '<growing>1</growing>' |
|
| 51 | '<family>' |
|
| 52 | '<name>foo</name>' |
|
| 53 | '<all>1</all>' |
|
| 54 | '<growing>1</growing>' |
|
| 55 | '</family>' |
|
| 56 | '<family>' |
|
| 57 | '<name>bar</name>' |
|
| 58 | '<all>1</all>' |
|
| 59 | '<growing>1</growing>' |
|
| 60 | '</family>' |
|
| 61 | '</family_selection>' |
|
| 62 | '</modify_config>' |
|
| 63 | ) |
|
| 64 | ||
| 65 | self.gmp.modify_policy_set_family_selection( |
|
| 66 | policy_id='c1', |
|
| 67 | families=[('foo', True, False), ('bar', False, True)], |
|
| 68 | ) |
|
| 69 | ||
| 70 | self.connection.send.has_been_called_with( |
|
| 71 | '<modify_config config_id="c1">' |
|
| 72 | '<family_selection>' |
|
| 73 | '<growing>1</growing>' |
|
| 74 | '<family>' |
|
| 75 | '<name>foo</name>' |
|
| 76 | '<all>0</all>' |
|
| 77 | '<growing>1</growing>' |
|
| 78 | '</family>' |
|
| 79 | '<family>' |
|
| 80 | '<name>bar</name>' |
|
| 81 | '<all>1</all>' |
|
| 82 | '<growing>0</growing>' |
|
| 83 | '</family>' |
|
| 84 | '</family_selection>' |
|
| 85 | '</modify_config>' |
|
| 86 | ) |
|
| 87 | ||
| 88 | def test_modify_policy_set_family_selection_missing_config_id(self): |
|
| 89 | with self.assertRaises(RequiredArgument): |
|
| 90 | self.gmp.modify_policy_set_family_selection( |
|
| 91 | policy_id=None, families=[('foo', True, True)] |
|
| 92 | ) |
|
| 93 | ||
| 94 | with self.assertRaises(RequiredArgument): |
|
| 95 | self.gmp.modify_policy_set_family_selection( |
|
| 96 | policy_id='', families=[('foo', True, True)] |
|
| 97 | ) |
|
| 98 | ||
| 99 | with self.assertRaises(RequiredArgument): |
|
| 100 | self.gmp.modify_policy_set_family_selection( |
|
| 101 | '', [('foo', True, True)] |
|
| 102 | ) |
|
| 103 | ||
| 104 | def test_modify_policy_set_family_selection_invalid_families(self): |
|
| 105 | with self.assertRaises(InvalidArgumentType): |
|
| 106 | self.gmp.modify_policy_set_family_selection( |
|
| 107 | policy_id='c1', families=None |
|
| 108 | ) |
|
| 109 | ||
| 110 | with self.assertRaises(InvalidArgumentType): |
|
| 111 | self.gmp.modify_policy_set_family_selection( |
|
| 112 | policy_id='c1', families='' |
|
| 113 | ) |
|
| 114 | ||
| 115 | with self.assertRaises(InvalidArgumentType): |
|
| 116 | self.gmp.modify_policy_set_family_selection('c1', '') |
|
| 117 | ||
| 118 | def test_modify_policy_set_family_selection_with_auto_add_new_families( |
|
| 119 | self, |
|
| 120 | ): |
|
| 121 | self.gmp.modify_policy_set_family_selection( |
|
| 122 | policy_id='c1', |
|
| 123 | families=[('foo', True, True)], |
|
| 124 | auto_add_new_families=True, |
|
| 125 | ) |
|
| 126 | ||
| 127 | self.connection.send.has_been_called_with( |
|
| 128 | '<modify_config config_id="c1">' |
|
| 129 | '<family_selection>' |
|
| 130 | '<growing>1</growing>' |
|
| 131 | '<family>' |
|
| 132 | '<name>foo</name>' |
|
| 133 | '<all>1</all>' |
|
| 134 | '<growing>1</growing>' |
|
| 135 | '</family>' |
|
| 136 | '</family_selection>' |
|
| 137 | '</modify_config>' |
|
| 138 | ) |
|
| 139 | ||
| 140 | self.gmp.modify_policy_set_family_selection( |
|
| 141 | policy_id='c1', |
|
| 142 | families=[('foo', True, True)], |
|
| 143 | auto_add_new_families=False, |
|
| 144 | ) |
|
| 145 | ||
| 146 | self.connection.send.has_been_called_with( |
|
| 147 | '<modify_config config_id="c1">' |
|
| 148 | '<family_selection>' |
|
| 149 | '<growing>0</growing>' |
|
| 150 | '<family>' |
|
| 151 | '<name>foo</name>' |
|
| 152 | '<all>1</all>' |
|
| 153 | '<growing>1</growing>' |
|
| 154 | '</family>' |
|
| 155 | '</family_selection>' |
|
| 156 | '</modify_config>' |
|
| 157 | ) |
|
| 158 | ||
| 159 | def test_modify_policy_set_family_selection_with_auto_add_new_nvts(self): |
|
| 160 | self.gmp.modify_policy_set_family_selection( |
|
| 161 | policy_id='c1', families=[('foo', True, True)] |
|
| 162 | ) |
|
| 163 | ||
| 164 | self.connection.send.has_been_called_with( |
|
| 165 | '<modify_config config_id="c1">' |
|
| 166 | '<family_selection>' |
|
| 167 | '<growing>1</growing>' |
|
| 168 | '<family>' |
|
| 169 | '<name>foo</name>' |
|
| 170 | '<all>1</all>' |
|
| 171 | '<growing>1</growing>' |
|
| 172 | '</family>' |
|
| 173 | '</family_selection>' |
|
| 174 | '</modify_config>' |
|
| 175 | ) |
|
| 176 | ||
| 177 | self.gmp.modify_policy_set_family_selection( |
|
| 178 | policy_id='c1', families=[('foo', False, True)] |
|
| 179 | ) |
|
| 180 | ||
| 181 | self.connection.send.has_been_called_with( |
|
| 182 | '<modify_config config_id="c1">' |
|
| 183 | '<family_selection>' |
|
| 184 | '<growing>1</growing>' |
|
| 185 | '<family>' |
|
| @@ 24-182 (lines=159) @@ | ||
| 21 | from gvm.errors import RequiredArgument, InvalidArgumentType |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyPolicySetFamilySelectionTestCase: |
|
| 25 | def test_modify_policy_set_family_selection(self): |
|
| 26 | self.gmp.modify_policy_set_family_selection( |
|
| 27 | policy_id='c1', families=[('foo', True, True)] |
|
| 28 | ) |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<modify_config config_id="c1">' |
|
| 32 | '<family_selection>' |
|
| 33 | '<growing>1</growing>' |
|
| 34 | '<family>' |
|
| 35 | '<name>foo</name>' |
|
| 36 | '<all>1</all>' |
|
| 37 | '<growing>1</growing>' |
|
| 38 | '</family>' |
|
| 39 | '</family_selection>' |
|
| 40 | '</modify_config>' |
|
| 41 | ) |
|
| 42 | ||
| 43 | self.gmp.modify_policy_set_family_selection( |
|
| 44 | policy_id='c1', families=(('foo', True, True), ('bar', True, True)) |
|
| 45 | ) |
|
| 46 | ||
| 47 | self.connection.send.has_been_called_with( |
|
| 48 | '<modify_config config_id="c1">' |
|
| 49 | '<family_selection>' |
|
| 50 | '<growing>1</growing>' |
|
| 51 | '<family>' |
|
| 52 | '<name>foo</name>' |
|
| 53 | '<all>1</all>' |
|
| 54 | '<growing>1</growing>' |
|
| 55 | '</family>' |
|
| 56 | '<family>' |
|
| 57 | '<name>bar</name>' |
|
| 58 | '<all>1</all>' |
|
| 59 | '<growing>1</growing>' |
|
| 60 | '</family>' |
|
| 61 | '</family_selection>' |
|
| 62 | '</modify_config>' |
|
| 63 | ) |
|
| 64 | ||
| 65 | self.gmp.modify_policy_set_family_selection( |
|
| 66 | policy_id='c1', |
|
| 67 | families=[('foo', True, False), ('bar', False, True)], |
|
| 68 | ) |
|
| 69 | ||
| 70 | self.connection.send.has_been_called_with( |
|
| 71 | '<modify_config config_id="c1">' |
|
| 72 | '<family_selection>' |
|
| 73 | '<growing>1</growing>' |
|
| 74 | '<family>' |
|
| 75 | '<name>foo</name>' |
|
| 76 | '<all>0</all>' |
|
| 77 | '<growing>1</growing>' |
|
| 78 | '</family>' |
|
| 79 | '<family>' |
|
| 80 | '<name>bar</name>' |
|
| 81 | '<all>1</all>' |
|
| 82 | '<growing>0</growing>' |
|
| 83 | '</family>' |
|
| 84 | '</family_selection>' |
|
| 85 | '</modify_config>' |
|
| 86 | ) |
|
| 87 | ||
| 88 | def test_modify_policy_set_family_selection_missing_config_id(self): |
|
| 89 | with self.assertRaises(RequiredArgument): |
|
| 90 | self.gmp.modify_policy_set_family_selection( |
|
| 91 | policy_id=None, families=[('foo', True, True)] |
|
| 92 | ) |
|
| 93 | ||
| 94 | with self.assertRaises(RequiredArgument): |
|
| 95 | self.gmp.modify_policy_set_family_selection( |
|
| 96 | policy_id='', families=[('foo', True, True)] |
|
| 97 | ) |
|
| 98 | ||
| 99 | with self.assertRaises(RequiredArgument): |
|
| 100 | self.gmp.modify_policy_set_family_selection( |
|
| 101 | '', [('foo', True, True)] |
|
| 102 | ) |
|
| 103 | ||
| 104 | def test_modify_policy_set_family_selection_invalid_families(self): |
|
| 105 | with self.assertRaises(InvalidArgumentType): |
|
| 106 | self.gmp.modify_policy_set_family_selection( |
|
| 107 | policy_id='c1', families=None |
|
| 108 | ) |
|
| 109 | ||
| 110 | with self.assertRaises(InvalidArgumentType): |
|
| 111 | self.gmp.modify_policy_set_family_selection( |
|
| 112 | policy_id='c1', families='' |
|
| 113 | ) |
|
| 114 | ||
| 115 | with self.assertRaises(InvalidArgumentType): |
|
| 116 | self.gmp.modify_policy_set_family_selection('c1', '') |
|
| 117 | ||
| 118 | def test_modify_policy_set_family_selection_with_auto_add_new_families( |
|
| 119 | self, |
|
| 120 | ): |
|
| 121 | self.gmp.modify_policy_set_family_selection( |
|
| 122 | policy_id='c1', |
|
| 123 | families=[('foo', True, True)], |
|
| 124 | auto_add_new_families=True, |
|
| 125 | ) |
|
| 126 | ||
| 127 | self.connection.send.has_been_called_with( |
|
| 128 | '<modify_config config_id="c1">' |
|
| 129 | '<family_selection>' |
|
| 130 | '<growing>1</growing>' |
|
| 131 | '<family>' |
|
| 132 | '<name>foo</name>' |
|
| 133 | '<all>1</all>' |
|
| 134 | '<growing>1</growing>' |
|
| 135 | '</family>' |
|
| 136 | '</family_selection>' |
|
| 137 | '</modify_config>' |
|
| 138 | ) |
|
| 139 | ||
| 140 | self.gmp.modify_policy_set_family_selection( |
|
| 141 | policy_id='c1', |
|
| 142 | families=[('foo', True, True)], |
|
| 143 | auto_add_new_families=False, |
|
| 144 | ) |
|
| 145 | ||
| 146 | self.connection.send.has_been_called_with( |
|
| 147 | '<modify_config config_id="c1">' |
|
| 148 | '<family_selection>' |
|
| 149 | '<growing>0</growing>' |
|
| 150 | '<family>' |
|
| 151 | '<name>foo</name>' |
|
| 152 | '<all>1</all>' |
|
| 153 | '<growing>1</growing>' |
|
| 154 | '</family>' |
|
| 155 | '</family_selection>' |
|
| 156 | '</modify_config>' |
|
| 157 | ) |
|
| 158 | ||
| 159 | def test_modify_policy_set_family_selection_with_auto_add_new_nvts(self): |
|
| 160 | self.gmp.modify_policy_set_family_selection( |
|
| 161 | policy_id='c1', families=[('foo', True, True)] |
|
| 162 | ) |
|
| 163 | ||
| 164 | self.connection.send.has_been_called_with( |
|
| 165 | '<modify_config config_id="c1">' |
|
| 166 | '<family_selection>' |
|
| 167 | '<growing>1</growing>' |
|
| 168 | '<family>' |
|
| 169 | '<name>foo</name>' |
|
| 170 | '<all>1</all>' |
|
| 171 | '<growing>1</growing>' |
|
| 172 | '</family>' |
|
| 173 | '</family_selection>' |
|
| 174 | '</modify_config>' |
|
| 175 | ) |
|
| 176 | ||
| 177 | self.gmp.modify_policy_set_family_selection( |
|
| 178 | policy_id='c1', families=[('foo', False, True)] |
|
| 179 | ) |
|
| 180 | ||
| 181 | self.connection.send.has_been_called_with( |
|
| 182 | '<modify_config config_id="c1">' |
|
| 183 | '<family_selection>' |
|
| 184 | '<growing>1</growing>' |
|
| 185 | '<family>' |
|
