def test_set_version(self, pkg_mock): test_object = ReflexInitializer(os.getcwd(), False) version_mock = MagicMock() version_mock.version = "4.3.21" pkg_mock.require.return_value = [version_mock] test_object.set_version() self.assertTrue(test_object.configs["version"] == "4.3.21")
def test_set_global_values(self, input_mock): input_mock.return_value = "*****@*****.**" empty_initializer = ReflexInitializer(False, os.getcwd()) self.assertTrue("globals" not in empty_initializer.configs.keys()) self.assertEqual( "*****@*****.**", empty_initializer.set_global_values()["default_email"], )
def test_query_possible_measures(self, input_mock, discoverer_mock): """Test our logic for measures is correct""" input_mock.return_value = "y" measure_mock = MagicMock() measure_mock.name = "test" measure_mock.version = "test" discoverer_mock = measure_mock discoverer_mock.discovered_measures = [measure_mock] test_object = ReflexInitializer(os.getcwd(), False) single_valid_template = test_object.query_possible_measures() self.assertEqual(single_valid_template, []) discoverer_mock.return_value = ["test"] no_valid_template = test_object.query_possible_measures() self.assertTrue(no_valid_template == [])
def overwrite_reflex_config(self): """If any upgrades possible, overwrite current reflex config.""" initializer = ReflexInitializer(False, self.config_file, self.auto_overwrite) initializer.config_file = self.config_file initializer.configs = self.current_config.raw_configuration initializer.write_config_file()
def cli(context, select_all): """Creates a new reflex ready directory structure.""" LOGGER.debug("Initializing reflex directory in: %s", context.home) LOGGER.info("Generating reflex.yaml config file in: %s", context.home) initializer = ReflexInitializer(context.home, select_all) initializer.determine_config_values() initializer.write_config_file()
def cli(interactive, config): """ Generates a reflex configuration yaml file to be later used to build reflex infrastructure. Default configuration file name is reflex.yaml """ LOGGER.debug("Initializing reflex directory in: %s", config) LOGGER.info("%s📝 Generating reflex.yaml config file in: %s%s", BOLD, config, ENDC) initializer = ReflexInitializer(interactive, config) initializer.determine_config_values() initializer.write_config_file()
def setUp(self): self.initializer = ReflexInitializer(os.getcwd(), False)
class ReflexInitializerTestCase(unittest.TestCase): """Test class for the environment context of our CLI tool.""" def setUp(self): self.initializer = ReflexInitializer(False, os.getcwd()) @patch("reflex_cli.reflex_initializer.RuleDiscoverer") @patch("reflex_cli.reflex_initializer.UserInput.get_input") def test_query_possible_rules(self, input_mock, discoverer_mock): """Test our logic for rules is correct""" input_mock.return_value = "y" rule_mock = MagicMock() rule_mock.name = "test" rule_mock.version = "test" discoverer_mock = rule_mock discoverer_mock.discovered_rules = [rule_mock] test_object = ReflexInitializer(os.getcwd(), False) single_valid_template = test_object.query_possible_rules() self.assertEqual(single_valid_template, []) discoverer_mock.return_value = ["test"] no_valid_template = test_object.query_possible_rules() self.assertTrue(no_valid_template == []) @patch("reflex_cli.reflex_initializer.pkg_resources") def test_get_reflex_version(self, pkg_mock): test_object = ReflexInitializer(False, os.getcwd()) version_mock = MagicMock() version_mock.version = "4.3.21" pkg_mock.require.return_value = [version_mock] self.assertTrue(test_object.get_reflex_version() == "4.3.21") def test_strip_rule_common_names(self): example_rule_array = [ { "reflex-aws-test": { "example": "example1" } }, { "reflex-aws-example": "value" }, { "non-reflex": "value" }, ] desired_rule_array = [ { "test": { "example": "example1" } }, { "example": "value" }, { "non-reflex": "value" }, ] processed_array = self.initializer.strip_rule_common_names( example_rule_array) self.assertNotEqual(example_rule_array, processed_array) self.assertEqual(processed_array, desired_rule_array) @patch("reflex_cli.reflex_initializer.UserInput.collect_default_email") def test_set_global_values(self, input_mock): input_mock.return_value = "*****@*****.**" empty_initializer = ReflexInitializer(False, os.getcwd()) self.assertTrue("globals" not in empty_initializer.configs.keys()) self.assertEqual( "*****@*****.**", empty_initializer.set_global_values()["default_email"], )
def test_get_reflex_version(self, pkg_mock): test_object = ReflexInitializer(False, os.getcwd()) version_mock = MagicMock() version_mock.version = "4.3.21" pkg_mock.require.return_value = [version_mock] self.assertTrue(test_object.get_reflex_version() == "4.3.21")
def overwrite_reflex_config(self): """If any upgrades possible, overwrite current reflex config.""" initializer = ReflexInitializer(self.config_file, False) initializer.config_file = self.config_file initializer.configs = self.current_config initializer.write_config_file()