| @@ 24-175 (lines=152) @@ | ||
| 21 | from gvm.errors import RequiredArgument, InvalidArgumentType |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyConfigSetFamilySelectionTestCase: |
|
| 25 | def test_modify_config_set_family_selection(self): |
|
| 26 | self.gmp.modify_config_set_family_selection( |
|
| 27 | config_id='c1', families=['foo'] |
|
| 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_config_set_family_selection( |
|
| 44 | config_id='c1', families=['foo', 'bar'] |
|
| 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_config_set_family_selection( |
|
| 66 | config_id='c1', families=('foo', 'bar') |
|
| 67 | ) |
|
| 68 | ||
| 69 | self.connection.send.has_been_called_with( |
|
| 70 | '<modify_config config_id="c1">' |
|
| 71 | '<family_selection>' |
|
| 72 | '<growing>1</growing>' |
|
| 73 | '<family>' |
|
| 74 | '<name>foo</name>' |
|
| 75 | '<all>1</all>' |
|
| 76 | '<growing>1</growing>' |
|
| 77 | '</family>' |
|
| 78 | '<family>' |
|
| 79 | '<name>bar</name>' |
|
| 80 | '<all>1</all>' |
|
| 81 | '<growing>1</growing>' |
|
| 82 | '</family>' |
|
| 83 | '</family_selection>' |
|
| 84 | '</modify_config>' |
|
| 85 | ) |
|
| 86 | ||
| 87 | def test_modify_config_set_family_selection_missing_config_id(self): |
|
| 88 | with self.assertRaises(RequiredArgument): |
|
| 89 | self.gmp.modify_config_set_family_selection( |
|
| 90 | config_id=None, families=['foo'] |
|
| 91 | ) |
|
| 92 | ||
| 93 | with self.assertRaises(RequiredArgument): |
|
| 94 | self.gmp.modify_config_set_family_selection( |
|
| 95 | config_id='', families=['foo'] |
|
| 96 | ) |
|
| 97 | ||
| 98 | with self.assertRaises(RequiredArgument): |
|
| 99 | self.gmp.modify_config_set_family_selection('', ['foo']) |
|
| 100 | ||
| 101 | def test_modify_config_set_family_selection_invalid_families(self): |
|
| 102 | with self.assertRaises(InvalidArgumentType): |
|
| 103 | self.gmp.modify_config_set_family_selection( |
|
| 104 | config_id='c1', families=None |
|
| 105 | ) |
|
| 106 | ||
| 107 | with self.assertRaises(InvalidArgumentType): |
|
| 108 | self.gmp.modify_config_set_family_selection( |
|
| 109 | config_id='c1', families='' |
|
| 110 | ) |
|
| 111 | ||
| 112 | with self.assertRaises(InvalidArgumentType): |
|
| 113 | self.gmp.modify_config_set_family_selection('c1', '') |
|
| 114 | ||
| 115 | def test_modify_config_set_family_selection_with_auto_add_new_families( |
|
| 116 | self, |
|
| 117 | ): |
|
| 118 | self.gmp.modify_config_set_family_selection( |
|
| 119 | config_id='c1', families=['foo'], auto_add_new_families=True |
|
| 120 | ) |
|
| 121 | ||
| 122 | self.connection.send.has_been_called_with( |
|
| 123 | '<modify_config config_id="c1">' |
|
| 124 | '<family_selection>' |
|
| 125 | '<growing>1</growing>' |
|
| 126 | '<family>' |
|
| 127 | '<name>foo</name>' |
|
| 128 | '<all>1</all>' |
|
| 129 | '<growing>1</growing>' |
|
| 130 | '</family>' |
|
| 131 | '</family_selection>' |
|
| 132 | '</modify_config>' |
|
| 133 | ) |
|
| 134 | ||
| 135 | self.gmp.modify_config_set_family_selection( |
|
| 136 | config_id='c1', families=['foo'], auto_add_new_families=False |
|
| 137 | ) |
|
| 138 | ||
| 139 | self.connection.send.has_been_called_with( |
|
| 140 | '<modify_config config_id="c1">' |
|
| 141 | '<family_selection>' |
|
| 142 | '<growing>0</growing>' |
|
| 143 | '<family>' |
|
| 144 | '<name>foo</name>' |
|
| 145 | '<all>1</all>' |
|
| 146 | '<growing>1</growing>' |
|
| 147 | '</family>' |
|
| 148 | '</family_selection>' |
|
| 149 | '</modify_config>' |
|
| 150 | ) |
|
| 151 | ||
| 152 | def test_modify_config_set_family_selection_with_auto_add_new_nvts(self): |
|
| 153 | self.gmp.modify_config_set_family_selection( |
|
| 154 | config_id='c1', families=['foo'], auto_add_new_nvts=True |
|
| 155 | ) |
|
| 156 | ||
| 157 | self.connection.send.has_been_called_with( |
|
| 158 | '<modify_config config_id="c1">' |
|
| 159 | '<family_selection>' |
|
| 160 | '<growing>1</growing>' |
|
| 161 | '<family>' |
|
| 162 | '<name>foo</name>' |
|
| 163 | '<all>1</all>' |
|
| 164 | '<growing>1</growing>' |
|
| 165 | '</family>' |
|
| 166 | '</family_selection>' |
|
| 167 | '</modify_config>' |
|
| 168 | ) |
|
| 169 | ||
| 170 | self.gmp.modify_config_set_family_selection( |
|
| 171 | config_id='c1', families=['foo'], auto_add_new_nvts=False |
|
| 172 | ) |
|
| 173 | ||
| 174 | self.connection.send.has_been_called_with( |
|
| 175 | '<modify_config config_id="c1">' |
|
| 176 | '<family_selection>' |
|
| 177 | '<growing>1</growing>' |
|
| 178 | '<family>' |
|
| @@ 24-175 (lines=152) @@ | ||
| 21 | from gvm.errors import RequiredArgument, InvalidArgumentType |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyPolicySetFamilySelectionTestCase: |
|
| 25 | def test_modify_config_set_family_selection(self): |
|
| 26 | self.gmp.modify_policy_set_family_selection( |
|
| 27 | policy_id='c1', families=['foo'] |
|
| 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', 'bar'] |
|
| 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', families=('foo', 'bar') |
|
| 67 | ) |
|
| 68 | ||
| 69 | self.connection.send.has_been_called_with( |
|
| 70 | '<modify_config config_id="c1">' |
|
| 71 | '<family_selection>' |
|
| 72 | '<growing>1</growing>' |
|
| 73 | '<family>' |
|
| 74 | '<name>foo</name>' |
|
| 75 | '<all>1</all>' |
|
| 76 | '<growing>1</growing>' |
|
| 77 | '</family>' |
|
| 78 | '<family>' |
|
| 79 | '<name>bar</name>' |
|
| 80 | '<all>1</all>' |
|
| 81 | '<growing>1</growing>' |
|
| 82 | '</family>' |
|
| 83 | '</family_selection>' |
|
| 84 | '</modify_config>' |
|
| 85 | ) |
|
| 86 | ||
| 87 | def test_modify_config_set_family_selection_missing_config_id(self): |
|
| 88 | with self.assertRaises(RequiredArgument): |
|
| 89 | self.gmp.modify_policy_set_family_selection( |
|
| 90 | policy_id=None, families=['foo'] |
|
| 91 | ) |
|
| 92 | ||
| 93 | with self.assertRaises(RequiredArgument): |
|
| 94 | self.gmp.modify_policy_set_family_selection( |
|
| 95 | policy_id='', families=['foo'] |
|
| 96 | ) |
|
| 97 | ||
| 98 | with self.assertRaises(RequiredArgument): |
|
| 99 | self.gmp.modify_policy_set_family_selection('', ['foo']) |
|
| 100 | ||
| 101 | def test_modify_config_set_family_selection_invalid_families(self): |
|
| 102 | with self.assertRaises(InvalidArgumentType): |
|
| 103 | self.gmp.modify_policy_set_family_selection( |
|
| 104 | policy_id='c1', families=None |
|
| 105 | ) |
|
| 106 | ||
| 107 | with self.assertRaises(InvalidArgumentType): |
|
| 108 | self.gmp.modify_policy_set_family_selection( |
|
| 109 | policy_id='c1', families='' |
|
| 110 | ) |
|
| 111 | ||
| 112 | with self.assertRaises(InvalidArgumentType): |
|
| 113 | self.gmp.modify_policy_set_family_selection('c1', '') |
|
| 114 | ||
| 115 | def test_modify_config_set_family_selection_with_auto_add_new_families( |
|
| 116 | self, |
|
| 117 | ): |
|
| 118 | self.gmp.modify_policy_set_family_selection( |
|
| 119 | policy_id='c1', families=['foo'], auto_add_new_families=True |
|
| 120 | ) |
|
| 121 | ||
| 122 | self.connection.send.has_been_called_with( |
|
| 123 | '<modify_config config_id="c1">' |
|
| 124 | '<family_selection>' |
|
| 125 | '<growing>1</growing>' |
|
| 126 | '<family>' |
|
| 127 | '<name>foo</name>' |
|
| 128 | '<all>1</all>' |
|
| 129 | '<growing>1</growing>' |
|
| 130 | '</family>' |
|
| 131 | '</family_selection>' |
|
| 132 | '</modify_config>' |
|
| 133 | ) |
|
| 134 | ||
| 135 | self.gmp.modify_policy_set_family_selection( |
|
| 136 | policy_id='c1', families=['foo'], auto_add_new_families=False |
|
| 137 | ) |
|
| 138 | ||
| 139 | self.connection.send.has_been_called_with( |
|
| 140 | '<modify_config config_id="c1">' |
|
| 141 | '<family_selection>' |
|
| 142 | '<growing>0</growing>' |
|
| 143 | '<family>' |
|
| 144 | '<name>foo</name>' |
|
| 145 | '<all>1</all>' |
|
| 146 | '<growing>1</growing>' |
|
| 147 | '</family>' |
|
| 148 | '</family_selection>' |
|
| 149 | '</modify_config>' |
|
| 150 | ) |
|
| 151 | ||
| 152 | def test_modify_config_set_family_selection_with_auto_add_new_nvts(self): |
|
| 153 | self.gmp.modify_policy_set_family_selection( |
|
| 154 | policy_id='c1', families=['foo'], auto_add_new_nvts=True |
|
| 155 | ) |
|
| 156 | ||
| 157 | self.connection.send.has_been_called_with( |
|
| 158 | '<modify_config config_id="c1">' |
|
| 159 | '<family_selection>' |
|
| 160 | '<growing>1</growing>' |
|
| 161 | '<family>' |
|
| 162 | '<name>foo</name>' |
|
| 163 | '<all>1</all>' |
|
| 164 | '<growing>1</growing>' |
|
| 165 | '</family>' |
|
| 166 | '</family_selection>' |
|
| 167 | '</modify_config>' |
|
| 168 | ) |
|
| 169 | ||
| 170 | self.gmp.modify_policy_set_family_selection( |
|
| 171 | policy_id='c1', families=['foo'], auto_add_new_nvts=False |
|
| 172 | ) |
|
| 173 | ||
| 174 | self.connection.send.has_been_called_with( |
|
| 175 | '<modify_config config_id="c1">' |
|
| 176 | '<family_selection>' |
|
| 177 | '<growing>1</growing>' |
|
| 178 | '<family>' |
|