jebus/enviPy-bayer
1
0
Fork 0
forked from enviPath/enviPy
Code Issues Pull Requests Actions Releases Activity

[Chore] Linted Files (#150)

Browse Source 
Co-authored-by: Tim Lorsbach <tim@lorsba.ch>
Reviewed-on: enviPath/enviPy#150
This commit is contained in:
jebus
2025-10-09 07:25:13 +13:00
parent 22f0bbe10b
commit afeb56622c
50 changed files with 5616 additions and 4408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
tests/views/test_rule_views.py
View File

@ -12,22 +12,29 @@ class RuleViewTest(TestCase):
@classmethod
def setUpClass(cls):
super(RuleViewTest, cls).setUpClass()
cls.user1 = UserManager.create_user("user1", "user1@envipath.com", "SuperSafe",
set_setting=False, add_to_group=True, is_active=True)
cls.user1 = UserManager.create_user(
"user1",
"user1@envipath.com",
"SuperSafe",
set_setting=False,
add_to_group=True,
is_active=True,
)
cls.user1_default_package = cls.user1.default_package
cls.package = PackageManager.create_package(cls.user1, 'Test', 'Test Pack')
cls.package = PackageManager.create_package(cls.user1, "Test", "Test Pack")
def setUp(self):
self.client.force_login(self.user1)
def test_create_rule(self):
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -38,18 +45,21 @@ class RuleViewTest(TestCase):
self.assertEqual(r.package, self.user1_default_package)
self.assertEqual(r.name, "Test Rule")
self.assertEqual(r.description, "Just a Description")
self.assertEqual(r.smirks,
"[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]")
self.assertEqual(
r.smirks,
"[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
)
self.assertEqual(self.user1_default_package.rules.count(), 1)
# Adding the same rule again should return the existing one, hence not increasing the number of rules
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.url, rule_url)
@ -58,12 +68,13 @@ class RuleViewTest(TestCase):
# Adding the same rule in a different package should create a new rule
response = self.client.post(
reverse("package rule list", kwargs={'package_uuid': self.package.uuid}), {
reverse("package rule list", kwargs={"package_uuid": self.package.uuid}),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -72,12 +83,13 @@ class RuleViewTest(TestCase):
# Edit
def test_edit_rule(self):
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -86,13 +98,17 @@ class RuleViewTest(TestCase):
r = Rule.objects.get(url=rule_url)
response = self.client.post(
reverse("package rule detail", kwargs={
'package_uuid': str(self.user1_default_package.uuid),
'rule_uuid': str(r.uuid)
}), {
reverse(
"package rule detail",
kwargs={
"package_uuid": str(self.user1_default_package.uuid),
"rule_uuid": str(r.uuid),
},
),
{
"rule-name": "Test Rule Adjusted",
"rule-description": "New Description",
}
},
)
self.assertEqual(response.status_code, 302)
@ -108,7 +124,7 @@ class RuleViewTest(TestCase):
"Test Desc",
"2025-10",
"soil",
[Temperature(interval=Interval(start=20, end=30))]
[Temperature(interval=Interval(start=20, end=30))],
)
s2 = Scenario.create(
@ -117,16 +133,17 @@ class RuleViewTest(TestCase):
"Test Desc2",
"2025-10",
"soil",
[Temperature(interval=Interval(start=10, end=20))]
[Temperature(interval=Interval(start=10, end=20))],
)
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -134,48 +151,48 @@ class RuleViewTest(TestCase):
r = Rule.objects.get(url=rule_url)
response = self.client.post(
reverse("package rule detail", kwargs={
'package_uuid': str(r.package.uuid),
'rule_uuid': str(r.uuid)
}), {
"selected-scenarios": [s1.url, s2.url]
}
reverse(
"package rule detail",
kwargs={"package_uuid": str(r.package.uuid), "rule_uuid": str(r.uuid)},
),
{"selected-scenarios": [s1.url, s2.url]},
)
self.assertEqual(len(r.scenarios.all()), 2)
response = self.client.post(
reverse("package rule detail", kwargs={
'package_uuid': str(r.package.uuid),
'rule_uuid': str(r.uuid)
}), {
"selected-scenarios": [s1.url]
}
reverse(
"package rule detail",
kwargs={"package_uuid": str(r.package.uuid), "rule_uuid": str(r.uuid)},
),
{"selected-scenarios": [s1.url]},
)
self.assertEqual(len(r.scenarios.all()), 1)
self.assertEqual(r.scenarios.first().url, s1.url)
response = self.client.post(
reverse("package rule detail", kwargs={
'package_uuid': str(r.package.uuid),
'rule_uuid': str(r.uuid)
}), {
reverse(
"package rule detail",
kwargs={"package_uuid": str(r.package.uuid), "rule_uuid": str(r.uuid)},
),
{
# We have to set an empty string to avoid that the parameter is removed
"selected-scenarios": ""
}
},
)
self.assertEqual(len(r.scenarios.all()), 0)
def test_copy(self):
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -183,12 +200,13 @@ class RuleViewTest(TestCase):
r = Rule.objects.get(url=rule_url)
response = self.client.post(
reverse("package detail", kwargs={
'package_uuid': str(self.package.uuid),
}), {
"hidden": "copy",
"object_to_copy": r.url
}
reverse(
"package detail",
kwargs={
"package_uuid": str(self.package.uuid),
},
),
{"hidden": "copy", "object_to_copy": r.url},
)
self.assertEqual(response.status_code, 200)
@ -202,26 +220,29 @@ class RuleViewTest(TestCase):
# Copy to the same package should fail
response = self.client.post(
reverse("package detail", kwargs={
'package_uuid': str(r.package.uuid),
}), {
"hidden": "copy",
"object_to_copy": r.url
}
reverse(
"package detail",
kwargs={
"package_uuid": str(r.package.uuid),
},
),
{"hidden": "copy", "object_to_copy": r.url},
)
self.assertEqual(response.status_code, 400)
self.assertEqual(response.json()['error'], f"Can't copy object {rule_url} to the same package!")
self.assertEqual(
response.json()["error"], f"Can't copy object {rule_url} to the same package!"
)
def test_delete(self):
response = self.client.post(
reverse("rules"), {
reverse("rules"),
{
"rule-name": "Test Rule",
"rule-description": "Just a Description",
"rule-smirks": "[H:5][C:1]([#6:6])([#1,#9,#17,#35,#53:4])[#9,#17,#35,#53]>>[H:5][C:1]([#6:6])([#8])[#1,#9,#17,#35,#53:4]",
"rule-type": "SimpleAmbitRule",
}
},
)
self.assertEqual(response.status_code, 302)
@ -229,12 +250,11 @@ class RuleViewTest(TestCase):
r = Rule.objects.get(url=rule_url)
response = self.client.post(
reverse("package rule detail", kwargs={
'package_uuid': str(r.package.uuid),
'rule_uuid': str(r.uuid)
}), {
"hidden": "delete"
}
reverse(
"package rule detail",
kwargs={"package_uuid": str(r.package.uuid), "rule_uuid": str(r.uuid)},
),
{"hidden": "delete"},
)
self.assertEqual(self.user1_default_package.rules.count(), 0)