aboutsummaryrefslogtreecommitdiffstats
path: root/servo/tests
diff options
context:
space:
mode:
Diffstat (limited to 'servo/tests')
-rw-r--r--servo/tests/test_models.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/servo/tests/test_models.py b/servo/tests/test_models.py
index b0ef6e7..ba3b714 100644
--- a/servo/tests/test_models.py
+++ b/servo/tests/test_models.py
@@ -7,6 +7,19 @@ from servo.models.common import Configuration
from servo.models.order import Order
from servo.models.account import User
from servo.models.queue import Queue
+from servo.models.device import Device
+
+
+class DeviceTests(TestCase):
+ def test_purchase_country_code(self):
+ device = Device()
+ device.purchase_country = 'SE'
+ self.assertEquals(device.get_purchase_country(), 'Sweden')
+
+ def test_purchase_country(self):
+ device = Device()
+ device.purchase_country = 'Sweden'
+ self.assertEquals(device.get_purchase_country(), 'Sweden')
class ConfigurationTests(TestCase):