Building out models for more complete testing

Author: JackAtOmenApps

tests/migrations/0002_auto_20201101_1652.py

@@ -0,0 +1,38 @@
+# Generated by Django 3.0.8 on 2020-11-01 16:52
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('tests', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='EdgeSet',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=100)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='NodeSet',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=100)),
+            ],
+        ),
+        migrations.AddField(
+            model_name='networkedge',
+            name='edge_set',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='tests.EdgeSet'),
+        ),
+        migrations.AddField(
+            model_name='networknode',
+            name='edge_set',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='tests.EdgeSet'),
+        ),
+    ]

tests/models.py

@@ -1,9 +1,22 @@
 from django.db import models
 from django_postgresql_dag.models import node_factory, edge_factory
 
+
+class EdgeSet(models.Model):
+    """A model designed as a container for a set of edges"""
+    name = models.CharField(max_length=100)
+
+
+class NodeSet(models.Model):
+    """A model designed as a container for a set of nodes"""
+    name = models.CharField(max_length=100)
+
+
 class NetworkEdge(edge_factory("NetworkNode", concrete=False)):
     name = models.CharField(max_length=100)
 
+    edge_set = models.ForeignKey(EdgeSet, null=True, blank=True, on_delete=models.CASCADE)
+
     def __str__(self):
         return self.name
 
@@ -18,8 +31,10 @@ class Meta:
 class NetworkNode(node_factory(NetworkEdge)):
     name = models.CharField(max_length=100)
 
+    edge_set = models.ForeignKey(EdgeSet, null=True, blank=True, on_delete=models.CASCADE)
+
     def __str__(self):
         return self.name
 
     class Meta:
-        app_label = 'tests'
+        app_label = 'tests'