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()