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