From bebb4d465b790bed5424f46f22844e063d2f2029 Mon Sep 17 00:00:00 2001 From: Mouse Reeve Date: Tue, 12 May 2020 14:45:30 -0700 Subject: [PATCH] Tests base model absolute id generator --- fedireads/models/base_model.py | 2 +- fedireads/tests/test_base_model.py | 13 +++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) create mode 100644 fedireads/tests/test_base_model.py diff --git a/fedireads/models/base_model.py b/fedireads/models/base_model.py index cd1cae07..506e063f 100644 --- a/fedireads/models/base_model.py +++ b/fedireads/models/base_model.py @@ -11,7 +11,7 @@ class FedireadsModel(models.Model): @property def absolute_id(self): ''' constructs the absolute reference to any db object ''' - if self.remote_id: + if hasattr(self, 'remote_id') and self.remote_id: return self.remote_id base_path = 'https://%s' % DOMAIN diff --git a/fedireads/tests/test_base_model.py b/fedireads/tests/test_base_model.py new file mode 100644 index 00000000..79f34a20 --- /dev/null +++ b/fedireads/tests/test_base_model.py @@ -0,0 +1,13 @@ +''' testing models ''' +from django.test import TestCase + +from fedireads.models.base_model import FedireadsModel +from fedireads.settings import DOMAIN + + +class BaseModel(TestCase): + def test_absolute_id(self): + instance = FedireadsModel() + instance.id = 1 + expected = instance.absolute_id + self.assertEqual(expected, 'https://%s/fedireadsmodel/1' % DOMAIN)