def create_options(self, **kwargs):
"""
Create a mock options object. Send in a dictionary with the option destination and values
and they will be set. Note that you must use the destination and not just the option name.
For example, if the option is --my-option, the destination (unless customized) will be
my_option.
"""
p = OptionParser()
migrate.add_parser_options(p)
# Set the list of acceptable attributes for this Mock.
valid_options = filter(lambda x: x.dest is not None, p.option_list)
mock_opts = Mock(spec=[o.dest for o in valid_options])
# Set everything to the default
def set_default(opt):
# Optparse uses a tuple to indicate if no default has been set.
if opt.default != ("NO", "DEFAULT"):
val = opt.default
else:
val = None
setattr(mock_opts, opt.dest, val)
map(lambda x: set_default(x), valid_options)
if not kwargs:
kwargs = {}
map(lambda (k, v): setattr(mock_opts, k, v), kwargs.items())
# The five_to_six option is set after argument parsing in the module so we set it
# for convenience.
if 'five_to_six' not in kwargs:
mock_opts.five_to_six = False
return mock_opts
def test_activation_key_forbids_destination_credentials(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--activation-key", "bar", "--destination-user", "x"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
(options, args) = parser.parse_args(["--activation-key", "bar", "--destination-password", "y"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_activation_key_forbids_destination_credentials(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--activation-key", "bar", "--destination-user", "x"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
(options, args) = parser.parse_args(["--activation-key", "bar", "--destination-password", "y"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_choices_for_registration_state(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
valid = ["keep", "unentitle", "purge"]
for opt in valid:
(options, args) = parser.parse_args(["--registration-state", opt])
self.assertRaises(SystemExit, parser.parse_args, ["--registration-state", "blah"])
def create_options(self, **kwargs):
"""
Create a mock options object. Send in a dictionary with the option destination and values
and they will be set. Note that you must use the destination and not just the option name.
For example, if the option is --my-option, the destination (unless customized) will be
my_option.
"""
p = OptionParser()
migrate.add_parser_options(p)
# Set the list of acceptable attributes for this Mock.
valid_options = filter(lambda x: x.dest is not None, p.option_list)
mock_opts = Mock(spec=[o.dest for o in valid_options])
# Set everything to the default
def set_default(opt):
# Optparse uses a tuple to indicate if no default has been set.
if opt.default != ("NO", "DEFAULT"):
val = opt.default
else:
val = None
setattr(mock_opts, opt.dest, val)
map(lambda x: set_default(x), valid_options)
if not kwargs:
kwargs = {}
map(lambda (k, v): setattr(mock_opts, k, v), kwargs.items())
# The five_to_six option is set after argument parsing in the module so we set it
# for convenience.
if 'five_to_six' not in kwargs:
mock_opts.five_to_six = False
return mock_opts
def test_choices_for_registration_state(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
valid = ["keep", "unentitle", "purge"]
for opt in valid:
(options, args) = parser.parse_args(["--registration-state", opt])
self.assertRaises(SystemExit, parser.parse_args,
["--registration-state", "blah"])
def test_classic_migration_options(self):
parser = OptionParser()
migrate.add_parser_options(parser)
self.assertFalse(parser.has_option("--registration-state"))
self.assertTrue(parser.has_option("--org"))
self.assertTrue(parser.has_option("--environment"))
self.assertTrue(parser.has_option("--force"))
(opts, args) = parser.parse_args([])
migrate.set_defaults(opts, five_to_six_script=False)
self.assertFalse(opts.five_to_six)
self.assertEquals("purge", opts.registration_state)
def test_classic_migration_options(self):
parser = OptionParser()
migrate.add_parser_options(parser)
self.assertFalse(parser.has_option("--registration-state"))
self.assertTrue(parser.has_option("--org"))
self.assertTrue(parser.has_option("--environment"))
self.assertTrue(parser.has_option("--force"))
(opts, args) = parser.parse_args([])
migrate.set_defaults(opts, five_to_six_script=False)
self.assertFalse(opts.five_to_six)
self.assertEquals("purge", opts.registration_state)
def test_5to6_options(self):
five_to_six = True
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six)
self.assertTrue(parser.has_option("--registration-state"))
self.assertFalse(parser.has_option("--org"))
self.assertFalse(parser.has_option("--environment"))
self.assertFalse(parser.has_option("--force"))
(opts, args) = parser.parse_args([])
migrate.set_defaults(opts, five_to_six)
self.assertTrue(opts.five_to_six)
self.assertEquals(None, opts.org)
self.assertEquals(None, opts.environment)
self.assertTrue(opts.force)
def test_choices_for_registration_state(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
valid = ["keep", "unentitle", "purge"]
for opt in valid:
(options, args) = parser.parse_args(["--registration-state", opt])
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=False)
(options, args) = parser.parse_args(["--keep"])
self.assertEquals("keep", options.registration_state)
(options, args) = parser.parse_args([""])
self.assertEquals("purge", options.registration_state)
def test_5to6_options(self):
five_to_six = True
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six)
self.assertTrue(parser.has_option("--registration-state"))
self.assertFalse(parser.has_option("--org"))
self.assertFalse(parser.has_option("--environment"))
self.assertFalse(parser.has_option("--force"))
(opts, args) = parser.parse_args([])
migrate.set_defaults(opts, five_to_six)
self.assertTrue(opts.five_to_six)
self.assertEquals(None, opts.org)
self.assertEquals(None, opts.environment)
self.assertTrue(opts.force)
def test_choices_for_registration_state(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
valid = ["keep", "unentitle", "purge"]
for opt in valid:
(options, args) = parser.parse_args(["--registration-state", opt])
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=False)
(options, args) = parser.parse_args(["--keep"])
self.assertEquals("keep", options.registration_state)
(options, args) = parser.parse_args([""])
self.assertEquals("purge", options.registration_state)
def test_registration_state_default(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
(options, args) = parser.parse_args([])
self.assertEquals("unentitle", options.registration_state)
def test_mutually_exclusive_activation_keys_and_environment(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--environment", "foo", "--activation-key", "bar"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_activation_keys_require_org(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--activation-key", "bar"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_mutually_exclusive_activation_keys_and_environment(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--environment", "foo", "--activation-key", "bar"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_mutually_exclusive_auto_service_level_options(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--no-auto", "--servicelevel", "foo"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_registration_state_default(self):
parser = OptionParser()
migrate.add_parser_options(parser, five_to_six_script=True)
(options, args) = parser.parse_args([])
self.assertEquals("unentitle", options.registration_state)
def test_activation_keys_require_org(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options, args) = parser.parse_args(["--activation-key", "bar"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
def test_mutually_exclusive_auto_service_level_options(self):
parser = OptionParser()
migrate.add_parser_options(parser)
(options,
args) = parser.parse_args(["--no-auto", "--service-level", "foo"])
self.assertRaises(SystemExit, migrate.validate_options, (options))
