def test_consistency_levels(self):
read_cl = ConsistencyLevel.ONE
write_cl = ConsistencyLevel.LOCAL_QUORUM
# set directly
opts = GraphOptions(graph_read_consistency_level=read_cl,
graph_write_consistency_level=write_cl)
self.assertEqual(opts.graph_read_consistency_level, read_cl)
self.assertEqual(opts.graph_write_consistency_level, write_cl)
# mapping from base
opt_map = opts.get_options_map()
self.assertEqual(opt_map['graph-read-consistency'],
six.b(ConsistencyLevel.value_to_name[read_cl]))
self.assertEqual(opt_map['graph-write-consistency'],
six.b(ConsistencyLevel.value_to_name[write_cl]))
# empty by default
new_opts = GraphOptions()
opt_map = new_opts.get_options_map()
self.assertNotIn('graph-read-consistency', opt_map)
self.assertNotIn('graph-write-consistency', opt_map)
# set from other
opt_map = new_opts.get_options_map(opts)
self.assertEqual(opt_map['graph-read-consistency'],
six.b(ConsistencyLevel.value_to_name[read_cl]))
self.assertEqual(opt_map['graph-write-consistency'],
six.b(ConsistencyLevel.value_to_name[write_cl]))
def test_consistency_levels(self):
read_cl = ConsistencyLevel.ONE
write_cl = ConsistencyLevel.LOCAL_QUORUM
# set directly
opts = GraphOptions(graph_read_consistency_level=read_cl, graph_write_consistency_level=write_cl)
self.assertEqual(opts.graph_read_consistency_level, read_cl)
self.assertEqual(opts.graph_write_consistency_level, write_cl)
# mapping from base
opt_map = opts.get_options_map()
self.assertEqual(opt_map['graph-read-consistency'], six.b(ConsistencyLevel.value_to_name[read_cl]))
self.assertEqual(opt_map['graph-write-consistency'], six.b(ConsistencyLevel.value_to_name[write_cl]))
# empty by default
new_opts = GraphOptions()
opt_map = new_opts.get_options_map()
self.assertNotIn('graph-read-consistency', opt_map)
self.assertNotIn('graph-write-consistency', opt_map)
# set from other
opt_map = new_opts.get_options_map(opts)
self.assertEqual(opt_map['graph-read-consistency'], six.b(ConsistencyLevel.value_to_name[read_cl]))
self.assertEqual(opt_map['graph-write-consistency'], six.b(ConsistencyLevel.value_to_name[write_cl]))
def test_get_options(self):
# nothing set --> base map
base = GraphOptions(**self.api_params)
self.assertEqual(GraphOptions().get_options_map(base), base._graph_options)
# something set overrides
kwargs = self.api_params.copy() # this test concept got strange after we added default values for a couple GraphOption attrs
kwargs['graph_name'] = 'unit_test'
other = GraphOptions(**kwargs)
options = base.get_options_map(other)
updated = self.opt_mapping['graph_name']
self.assertEqual(options[updated], six.b('unit_test'))
for name in (n for n in self.opt_mapping.values() if n != updated):
self.assertEqual(options[name], base._graph_options[name])
# base unchanged
self._verify_api_params(base, self.api_params)
def test_get_options(self):
# nothing set --> base map
base = GraphOptions(**self.api_params)
self.assertEqual(GraphOptions().get_options_map(base), base._graph_options)
# something set overrides
kwargs = self.api_params.copy() # this test concept got strange after we added default values for a couple GraphOption attrs
kwargs['graph_name'] = 'unit_test'
other = GraphOptions(**kwargs)
options = base.get_options_map(other)
updated = self.opt_mapping['graph_name']
self.assertEqual(options[updated], six.b('unit_test'))
for name in (n for n in self.opt_mapping.values() if n != updated):
self.assertEqual(options[name], base._graph_options[name])
# base unchanged
self._verify_api_params(base, self.api_params)
def test_set_attr(self):
expected = 'test@@@@'
opts = GraphOptions(graph_name=expected)
self.assertEqual(opts.graph_name, six.b(expected))
expected = 'somethingelse####'
opts.graph_name = expected
self.assertEqual(opts.graph_name, six.b(expected))
# will update options with set value
another = GraphOptions()
self.assertIsNone(another.graph_name)
another.update(opts)
self.assertEqual(another.graph_name, six.b(expected))
opts.graph_name = None
self.assertIsNone(opts.graph_name)
# will not update another with its set-->unset value
another.update(opts)
self.assertEqual(another.graph_name, six.b(expected)) # remains unset
opt_map = another.get_options_map(opts)
self.assertEqual(opt_map, another._graph_options)
def test_set_attr(self):
expected = 'test@@@@'
opts = GraphOptions(graph_name=expected)
self.assertEqual(opts.graph_name, six.b(expected))
expected = 'somethingelse####'
opts.graph_name = expected
self.assertEqual(opts.graph_name, six.b(expected))
# will update options with set value
another = GraphOptions()
self.assertIsNone(another.graph_name)
another.update(opts)
self.assertEqual(another.graph_name, six.b(expected))
opts.graph_name = None
self.assertIsNone(opts.graph_name)
# will not update another with its set-->unset value
another.update(opts)
self.assertEqual(another.graph_name, six.b(expected)) # remains unset
opt_map = another.get_options_map(opts)
self.assertEqual(opt_map, another._graph_options)
