138 lines
4.2 KiB
Python
138 lines
4.2 KiB
Python
"""Tests for general API tests for the plugin app."""
|
|
|
|
from django.urls import reverse
|
|
|
|
from InvenTree.api_tester import InvenTreeAPITestCase, PluginMixin
|
|
|
|
|
|
class PluginDetailAPITest(PluginMixin, InvenTreeAPITestCase):
|
|
"""Tests the plugin API endpoints."""
|
|
|
|
roles = [
|
|
'admin.add',
|
|
'admin.view',
|
|
'admin.change',
|
|
'admin.delete',
|
|
]
|
|
|
|
def setUp(self):
|
|
"""Setup for all tests."""
|
|
self.MSG_NO_PKG = 'Either packagename of URL must be provided'
|
|
|
|
self.PKG_NAME = 'minimal'
|
|
self.PKG_URL = 'git+https://github.com/geoffrey-a-reed/minimal'
|
|
super().setUp()
|
|
|
|
def test_plugin_install(self):
|
|
"""Test the plugin install command."""
|
|
url = reverse('api-plugin-install')
|
|
|
|
# valid - Pypi
|
|
data = self.post(
|
|
url,
|
|
{
|
|
'confirm': True,
|
|
'packagename': self.PKG_NAME
|
|
},
|
|
expected_code=201,
|
|
).data
|
|
|
|
self.assertEqual(data['success'], True)
|
|
|
|
# valid - github url
|
|
data = self.post(
|
|
url,
|
|
{
|
|
'confirm': True,
|
|
'url': self.PKG_URL
|
|
},
|
|
expected_code=201,
|
|
).data
|
|
self.assertEqual(data['success'], True)
|
|
|
|
# valid - github url and packagename
|
|
data = self.post(
|
|
url,
|
|
{
|
|
'confirm': True,
|
|
'url': self.PKG_URL,
|
|
'packagename': 'minimal',
|
|
},
|
|
expected_code=201,
|
|
).data
|
|
self.assertEqual(data['success'], True)
|
|
|
|
# invalid tries
|
|
# no input
|
|
self.post(url, {}, expected_code=400)
|
|
|
|
# no package info
|
|
data = self.post(url, {
|
|
'confirm': True,
|
|
}, expected_code=400).data
|
|
|
|
self.assertEqual(data['url'][0].title().upper(), self.MSG_NO_PKG.upper())
|
|
self.assertEqual(data['packagename'][0].title().upper(), self.MSG_NO_PKG.upper())
|
|
|
|
# not confirmed
|
|
self.post(url, {
|
|
'packagename': self.PKG_NAME
|
|
}, expected_code=400)
|
|
|
|
data = self.post(url, {
|
|
'packagename': self.PKG_NAME,
|
|
'confirm': False,
|
|
}, expected_code=400).data
|
|
|
|
self.assertEqual(data['confirm'][0].title().upper(), 'Installation not confirmed'.upper())
|
|
|
|
def test_admin_action(self):
|
|
"""Test the PluginConfig action commands."""
|
|
url = reverse('admin:plugin_pluginconfig_changelist')
|
|
|
|
test_plg = self.plugin_confs.first()
|
|
# deactivate plugin
|
|
response = self.client.post(url, {
|
|
'action': 'plugin_deactivate',
|
|
'index': 0,
|
|
'_selected_action': [test_plg.pk],
|
|
}, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
# deactivate plugin - deactivate again -> nothing will hapen but the nothing 'changed' function is triggered
|
|
response = self.client.post(url, {
|
|
'action': 'plugin_deactivate',
|
|
'index': 0,
|
|
'_selected_action': [test_plg.pk],
|
|
}, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
# activate plugin
|
|
response = self.client.post(url, {
|
|
'action': 'plugin_activate',
|
|
'index': 0,
|
|
'_selected_action': [test_plg.pk],
|
|
}, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
# save to deactivate a plugin
|
|
response = self.client.post(reverse('admin:plugin_pluginconfig_change', args=(test_plg.pk, )), {
|
|
'_save': 'Save',
|
|
}, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_model(self):
|
|
"""Test the PluginConfig model."""
|
|
# check mixin registry
|
|
plg = self.plugin_confs.first()
|
|
mixin_dict = plg.mixins()
|
|
self.assertIn('base', mixin_dict)
|
|
self.assertDictContainsSubset({'base': {'key': 'base', 'human_name': 'base'}}, mixin_dict)
|
|
|
|
# check reload on save
|
|
with self.assertWarns(Warning) as cm:
|
|
plg_inactive = self.plugin_confs.filter(active=False).first()
|
|
plg_inactive.active = True
|
|
plg_inactive.save()
|
|
self.assertEqual(cm.warning.args[0], 'A reload was triggered')
|