def test_query(self):
manager = get_manager(self.plugin_info['1']['plugin'], db.session, user=self.plugin_info['1']['user'])
obj = TestModel()
manager.add(obj)
# We have created the object, so querying for it should give us the first instance.
data = manager.query_first(obj.plugin_proxy, obj.metric_proxy)
self.assertTrue(isinstance(data, TestModel))
def test_wrong_permissions(self):
manager = get_manager(self.plugin_info['1']['plugin'], db.session, user=self.plugin_info['1']['user'])
bad_plugin = PluginFactory()
obj = TestLoosePermissions()
manager.add(obj)
# This is 1 because of the object we made, and another because the setUp function makes an instance, which is 2.
self.run_query_assert(obj, bad_plugin, 2)
def test_query(self):
manager = get_manager(self.plugin_info['1']['plugin'],
db.session,
user=self.plugin_info['1']['user'])
obj = TestModel()
manager.add(obj)
# We have created the object, so querying for it should give us the first instance.
data = manager.query_first(obj.plugin_proxy, obj.metric_proxy)
self.assertTrue(isinstance(data, TestModel))
def test_wrong_permissions(self):
manager = get_manager(self.plugin_info['1']['plugin'],
db.session,
user=self.plugin_info['1']['user'])
bad_plugin = PluginFactory()
obj = TestLoosePermissions()
manager.add(obj)
# This is 1 because of the object we made, and another because the setUp function makes an instance, which is 2.
self.run_query_assert(obj, bad_plugin, 2)
def test_manager(self):
manager = get_manager(self.plugin_info['1']['plugin'], db.session, user=self.plugin_info['1']['user'])
bad_user = UserFactory()
bad_plugin = PluginFactory()
t_obj = TestModel()
manager.add(t_obj)
# We are the wrong user and plugin, so we don't have permissions for this yet.
self.run_query_assert(t_obj, bad_plugin, 0, bad_user)
# Right user, wrong plugin. Should give us nothing.
self.run_query_assert(t_obj, self.plugin_info['1']['plugin'], 0, bad_user)
# Wrong user, right plugin. Should give us nothing.
self.run_query_assert(t_obj, bad_plugin, 0, self.plugin_info['1']['user'])
# We are now the correct user and plugin, so we should have permission.
self.run_query_assert(t_obj, self.plugin_info['1']['plugin'], 1, self.plugin_info['1']['user'])
def test_manager(self):
manager = get_manager(self.plugin_info['1']['plugin'],
db.session,
user=self.plugin_info['1']['user'])
bad_user = UserFactory()
bad_plugin = PluginFactory()
t_obj = TestModel()
manager.add(t_obj)
# We are the wrong user and plugin, so we don't have permissions for this yet.
self.run_query_assert(t_obj, bad_plugin, 0, bad_user)
# Right user, wrong plugin. Should give us nothing.
self.run_query_assert(t_obj, self.plugin_info['1']['plugin'], 0,
bad_user)
# Wrong user, right plugin. Should give us nothing.
self.run_query_assert(t_obj, bad_plugin, 0,
self.plugin_info['1']['user'])
# We are now the correct user and plugin, so we should have permission.
self.run_query_assert(t_obj, self.plugin_info['1']['plugin'], 1,
self.plugin_info['1']['user'])
def run_query_assert(self, obj, plugin, times_len, user=None):
manager = get_manager(plugin, db.session, user=user)
times = manager.query_filter(obj.plugin_proxy, obj.metric_proxy)
self.assertEqual(len(times), times_len)
def run_query_assert(self, obj, plugin, times_len, user=None):
manager = get_manager(plugin, db.session, user=user)
times = manager.query_filter(obj.plugin_proxy, obj.metric_proxy)
self.assertEqual(len(times), times_len)
