From 09ef2e20882af372ccdc71923c9f15a8f16f8443 Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <filipp@mac.com>
Date: Sat, 16 Aug 2014 14:20:54 +0300
Subject: Added ShipTo argument to warranty check

---
 gsxws/core.py       | 8 ++++----
 gsxws/products.py   | 5 ++++-
 tests/test_gsxws.py | 9 ++++++++-
 3 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/gsxws/core.py b/gsxws/core.py
index 413e3fe..79b5687 100644
--- a/gsxws/core.py
+++ b/gsxws/core.py
@@ -478,10 +478,10 @@ def connect(user_id, password, sold_to,
     global GSX_LOCALE
     global GSX_REGION
 
-    GSX_LANG = language
-    GSX_REGION = region
-    GSX_LOCALE = locale
-    GSX_ENV = environment
+    GSX_LANG    = language
+    GSX_REGION  = region
+    GSX_LOCALE  = locale
+    GSX_ENV     = environment
 
     act = GsxSession(user_id, password, sold_to, language, timezone)
     return act.login()
diff --git a/gsxws/products.py b/gsxws/products.py
index b1bd5a7..f0bab00 100644
--- a/gsxws/products.py
+++ b/gsxws/products.py
@@ -49,7 +49,7 @@ class Product(object):
         self.configCode = result.configCode
         return result
 
-    def warranty(self, parts=[], date_received=None):
+    def warranty(self, parts=[], date_received=None, ship_to=None):
         """
         The Warranty Status API retrieves the same warranty details
         displayed on the GSX Coverage screen.
@@ -71,6 +71,9 @@ class Product(object):
         if self.should_check_activation:
             self.activation()
 
+        if ship_to:
+            self._gsx.shipTo = ship_to
+            
         try:
             self._gsx.partNumbers = []
             for k, v in parts:
diff --git a/tests/test_gsxws.py b/tests/test_gsxws.py
index 8f11633..1de1906 100644
--- a/tests/test_gsxws.py
+++ b/tests/test_gsxws.py
@@ -181,7 +181,14 @@ class TestPartFunction(RemoteTestCase):
         self.assertIsInstance(parts[0].partNumber, basestring)
 
 
-class TestWarrantyFunctions(TestCase):
+class TestRemoteWarrantyFunctions(RemoteTestCase):
+    def test_warranty_lookup(self):
+        product = Product('DGKFL06JDHJP')
+        wty = product.warranty(ship_to=env['GSX_SHIPTO'])
+        self.assertEqual(wty.warrantyStatus, 'Out Of Warranty (No Coverage)')
+
+
+class TestLocalWarrantyFunctions(TestCase):
     def setUp(self):
         self.data = parse('tests/fixtures/warranty_status.xml',
                           'warrantyDetailInfo')
-- 
cgit v1.2.3