class GmpCloneScannerTestCase(unittest.TestCase): def setUp(self): self.connection = MockConnection() self.gmp = Gmp(self.connection) def test_clone(self): self.gmp.clone_scanner('a1') self.connection.send.has_been_called_with('<create_scanner>' '<copy>a1</copy>' '</create_scanner>') def test_missing_id(self): with self.assertRaises(RequiredArgument): self.gmp.clone_scanner('') with self.assertRaises(RequiredArgument): self.gmp.clone_scanner(None)