class TestConfigurationReaders(unittest.TestCase): configuration = config.Configuration() def setUp(self): os.environ['SEARCHCVE_MONGODB'] = 'testdb' os.environ['SEARCHCVE_MONGOHOST'] = 'localhost' os.environ['SEARCHCVE_MONGOPORT'] = '27017' os.environ['SEARCHCVE_CVESTARTYEAR'] = '2022' def test_MongoDB(self): dbname = 'testdb' self.assertEqual(self.configuration.getMongoDB(), dbname) def test_MongoHost(self): self.assertEqual(self.configuration.getMongoHost(), 'localhost') def test_MongoPort(self): self.assertEqual(self.configuration.getMongoPort(), '27017') def test_CVEStartYear(self): startyear = '2002' self.assertEqual(str((self.configuration.getCVEStartYear())), startyear)
loader=FileSystemLoader(template_directory), autoescape=select_autoescape(['html', 'xml'])) self.template = template_environment.get_template(template_file) def render_template(self, *args, **kwargs): """ Render the loaded template and write it to <output_file>. """ data_dict = kwargs.get('data_dict', {}) rendered_template = self.template.render(**kwargs) return rendered_template confs = conf.Configuration() sender = confs.getMailSender() host = confs.getMailHost() port = confs.getMailPort() username = confs.getMailUsername() password = confs.getMailPassword() tls = confs.getMailTls() mailer = Mailer( dict(transport=dict(use='smtp', host=host, port=port, tls=tls, username=username, password=password)))