def test_add_model_to_repo(self): import tempfile fn = self.repo.get_models()['variables'] dm = DispersalModel(fn) temp_fn = tempfile.mktemp(suffix='.xml') dm.set_name('variable_test') dm.set_location('grass_location') dm.save_model(temp_fn) f = open(temp_fn,'r') data = f.read() f.close() webui.add_model_to_repo(data) os.remove(temp_fn) self.assertTrue('variable_test' in self.repo.get_models()) self.repo.remove_model('variable_test',force=True)
def test_add_and_delete_model(self): import tempfile fn = self.repo.get_models()['variables'] dm = DispersalModel(fn) temp_fn = tempfile.mktemp(suffix='.xml') dm.set_name('variable_test') dm.set_location('grass_location') dm.save_model(temp_fn) f = open(temp_fn,'r') data = f.read() f.close() webui.add_model_to_repo(data) os.remove(temp_fn) self.assertTrue('variable_test' in self.repo.get_models()) r = self.urlopen('/models/variable_test/del',method='POST') self.assertEqual(r['code'],303) self.assertTrue('variable_test' not in self.repo.get_models())