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):