From 86bd2bc1a6bbfe669af6ebc2724b051ea03b6ea2 Mon Sep 17 00:00:00 2001 From: Federico Capoano Date: Fri, 13 Jan 2017 16:39:04 +0100 Subject: [PATCH] [test] Reuse django_netjsonconfig.tests.CreateConfigMixin --- owm_legacy/tests.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/owm_legacy/tests.py b/owm_legacy/tests.py index a3e944e..4ff9371 100644 --- a/owm_legacy/tests.py +++ b/owm_legacy/tests.py @@ -4,26 +4,20 @@ from django.test import TestCase from django_netjsonconfig.models import Config +from django_netjsonconfig.tests import CreateConfigMixin from owm_legacy.settings import ALLOWED_SUBNETS -class TestOwmLegacy(TestCase): +class TestOwmLegacy(CreateConfigMixin, TestCase): """ tests for owm_legacy """ - def _create_config(self): - d = Config(name='test', - backend='netjsonconfig.OpenWrt', - config={'general':{'hostname':'test'}}, - key='00:11:22:33:44:55') - d.full_clean() - d.save() - return d + config_model = Config def test_get_config_md5(self): d = self._create_config() response = self.client.get(reverse('owm:get_config_md5', args=[d.key])) - self.assertEqual(response['Content-Disposition'], 'attachment; filename=00:11:22:33:44:55') + self.assertEqual(response['Content-Disposition'], 'attachment; filename={0}'.format(self.TEST_KEY)) self.assertEqual(len(response.content), 32) checksum1 = response.content sleep(1)