Adding unit tests

zenoss · Mar 15, 2017 · 444ee78 · 444ee78
1 parent fefeffb
commit 444ee78
Showing 1 changed file with 67 additions and 0 deletions.
diff --git a/Products/Zuul/tests/test_devicefacade.py b/Products/Zuul/tests/test_devicefacade.py
@@ -176,6 +176,73 @@ def test_deviceSortByProdState(self):
         self.assertEquals(resultIter.next().getProductionState(), 1000)
         self.assertEquals(resultIter.next().getProductionState(), 400)
 
+    def test_deviceSearchByProdStateAndLocationReturnsCorrectDevices(self):
+        manage_addLocation(self.dmd.Locations, "test1")
+
+        devMaintenance = self.dmd.Devices.createInstance('devMaintenance')
+        devMaintenance.setLocation("/test1")
+        devMaintenance.setProdState(400)
+
+        devProduction = self.dmd.Devices.createInstance('devProduction')
+        devProduction.setLocation("/test1")
+        devProduction.setProdState(1000)
+
+        # this device should never be returned
+        devOther = self.dmd.Devices.createInstance('devOther')
+        devOther.setLocation("/test1")
+        devOther.setProdState(1)
+
+        results = self.facade.getDeviceBrains(uid="/zport/dmd/Devices", params=dict(productionState=[400], location="test1"))
+        self.assertEquals(1, results.total)
+        device = iter(results).next()
+        self.assertEquals(device.getProductionState(), 400)
+        self.assertEquals(device.location.getRelatedId(), "test1")
+
+        results = self.facade.getDeviceBrains(uid="/zport/dmd/Devices", params=dict(productionState=[1000], location="test1"))
+        self.assertEquals(1, results.total)
+        device = iter(results).next()
+        self.assertEquals(device.getProductionState(), 1000)
+        self.assertEquals(device.location.getRelatedId(), "test1")
+
+    def test_deviceSortByProdStateWithLocationFilterReturnsCorrectDevicesAndSortsCorrectly(self):
+        manage_addLocation(self.dmd.Locations, "test1")
+        manage_addLocation(self.dmd.Locations, "test2")
+
+        devMaintenance = self.dmd.Devices.createInstance('devMaintenance')
+        devMaintenance.setLocation("/test1")
+        devMaintenance.setProdState(400)
+
+        devProduction = self.dmd.Devices.createInstance('devProduction')
+        devProduction.setLocation("/test1")
+        devProduction.setProdState(1000)
+
+        # this device should never be returned
+        devOther = self.dmd.Devices.createInstance('devOther')
+        devOther.setLocation("/test2")
+        devOther.setProdState(300)
+
+        # test sort in ascending order with location filter
+        results = self.facade.getDeviceBrains(uid="/zport/dmd/Devices", sort='productionState', params=dict(location="test1"))
+        resultIter = iter(results)
+        self.assertEquals(2, results.total)
+        device = resultIter.next()
+        self.assertEquals(device.getProductionState(), 400)
+        self.assertEquals(device.location.getRelatedId(), 'test1')
+        device = resultIter.next()
+        self.assertEquals(device.getProductionState(), 1000)
+        self.assertEquals(device.location.getRelatedId(), 'test1')
+
+        # test sort in descending order with location filter
+        results = self.facade.getDeviceBrains(uid="/zport/dmd/Devices", sort='productionState', dir='DESC', params=dict(location="test1"))
+        resultIter = iter(results)
+        self.assertEquals(2, results.total)
+        device = resultIter.next()
+        self.assertEquals(device.getProductionState(), 1000)
+        self.assertEquals(device.location.getRelatedId(), 'test1')
+        device = resultIter.next()
+        self.assertEquals(device.getProductionState(), 400)
+        self.assertEquals(device.location.getRelatedId(), 'test1')
+
     def test_setProductionState(self):
         dev = self.dmd.Devices.createInstance('dev')
         dev2 = self.dmd.Devices.createInstance('dev2')