def testGetClientsWithActionRequest(backendManager, clients, param): for client in clients: backendManager.host_insertObject(client) testclient1 = OpsiClient(id='testclient1.test.invalid') backendManager.host_insertObject(testclient1) product1 = LocalbootProduct('product1', '1.0', '1') backendManager.product_insertObject(product1) product2 = LocalbootProduct('product2', '2.0', '1') backendManager.product_insertObject(product2) fillerProducts = [ LocalbootProduct("filler1", '1', '1'), LocalbootProduct("filler2", '2', '2'), LocalbootProduct("filler3", '3', '3'), ] for poc in fillerProducts: backendManager.product_insertObject(poc) # Skipping update because we search for this actionRequests = ['setup', 'uninstall', 'always', 'once', 'custom', 'none'] fillerPocs = [ ProductOnClient(productId=product.getId(), productType=product.getType(), clientId=client.getId(), installationStatus=random.choice( ['installed', 'not_installed', 'unknown']), actionRequest=random.choice(actionRequests), productVersion=product.getProductVersion(), packageVersion=product.getPackageVersion()) for client, product in itertools.product(clients, fillerProducts) ] relevantPocs = [ ProductOnClient(productId=product1.getId(), productType=product1.getType(), clientId=testclient1.getId(), installationStatus='installed', actionRequest='update', productVersion=product1.getProductVersion(), packageVersion=product1.getPackageVersion()), ProductOnClient(productId=product2.getId(), productType=product2.getType(), clientId=testclient1.getId(), installationStatus='unknown', actionRequest='update', productVersion=product2.getProductVersion(), packageVersion=product2.getPackageVersion()), ] for poc in fillerPocs + relevantPocs: backendManager.productOnClient_insertObject(poc) clientsToCheck = backendManager.getClientsWithActionRequest(param) assert len(clientsToCheck) == 1 assert clientsToCheck[0] == testclient1.id
def testGetClientsWithProductsWithSpecificStatus(backendManager, clients, desiredStatus, expectedClient): for client in clients: backendManager.host_insertObject(client) testclient1 = OpsiClient(id='testclient1.test.invalid') backendManager.host_insertObject(testclient1) testclient2 = OpsiClient(id='testclient2.test.invalid') backendManager.host_insertObject(testclient2) product1 = LocalbootProduct('product1', '1.0', '1') backendManager.product_insertObject(product1) product2 = LocalbootProduct('product2', '2.0', '1') backendManager.product_insertObject(product2) fillerProducts = [ LocalbootProduct("filler1", '1', '1'), LocalbootProduct("filler2", '2', '2'), LocalbootProduct("filler3", '3', '3'), ] for poc in fillerProducts: backendManager.product_insertObject(poc) fillerPocs = [ ProductOnClient(productId=product.getId(), productType=product.getType(), clientId=client.getId(), installationStatus=random.choice( ['installed', 'not_installed', 'unknown']), productVersion=product.getProductVersion(), packageVersion=product.getPackageVersion()) for client, product in itertools.product(clients, fillerProducts) ] relevantPocs = [ ProductOnClient(productId=product1.getId(), productType=product1.getType(), clientId=testclient2.getId(), installationStatus='installed', productVersion=product1.getProductVersion(), packageVersion=product1.getPackageVersion()), ProductOnClient(productId=product2.getId(), productType=product2.getType(), clientId=testclient1.getId(), installationStatus='unknown', productVersion=product2.getProductVersion(), packageVersion=product2.getPackageVersion()), ] for poc in fillerPocs + relevantPocs: backendManager.productOnClient_insertObject(poc) clientsToCheck = backendManager.getClientsWithProducts( [product1.id, product2.id], desiredStatus) assert len(clientsToCheck) == 1 assert clientsToCheck[0] == expectedClient
def testSetProductActionRequestForHostGroup(backendManager): testGroup = HostGroup(id='host_group_1') client1 = OpsiClient(id='client1.test.invalid') client2 = OpsiClient(id='client2.test.invalid') client1ToGroup = ObjectToGroup(testGroup.getType(), testGroup.id, client1.id) client2ToGroup = ObjectToGroup(testGroup.getType(), testGroup.id, client2.id) depot = OpsiDepotserver(id='depotserver1.test.invalid') product2 = LocalbootProduct( id='product2', name=u'Product 2', productVersion='2.0', packageVersion='test', setupScript="setup.ins", ) prodOnDepot = ProductOnDepot(productId=product2.getId(), productType=product2.getType(), productVersion=product2.getProductVersion(), packageVersion=product2.getPackageVersion(), depotId=depot.getId()) backendManager.host_insertObject(client1) backendManager.host_insertObject(client2) backendManager.host_insertObject(depot) backendManager.group_insertObject(testGroup) backendManager.objectToGroup_createObjects( [client1ToGroup, client2ToGroup]) backendManager.config_create(u'clientconfig.depot.id') backendManager.configState_create(u'clientconfig.depot.id', client1.getId(), values=[depot.getId()]) backendManager.configState_create(u'clientconfig.depot.id', client2.getId(), values=[depot.getId()]) backendManager.product_insertObject(product2) backendManager.productOnDepot_insertObject(prodOnDepot) backendManager.setProductActionRequestForHostGroup('host_group_1', 'product2', 'setup') pocs = backendManager.productOnClient_getObjects() assert pocs assert len(pocs) == 2 for poc in backendManager.productOnClient_getObjects(): assert poc.productId == product2.getId() assert poc.clientId in (client1.id, client2.id)
def createClientAndDepot(backend): client = OpsiClient( id='backend-test-1.vmnat.local', description='Unittest Test client.' ) depot = OpsiDepotserver( id='depotserver1.some.test', description='Test Depot', ) backend.host_createObjects([client, depot]) clientConfigDepotId = UnicodeConfig( id=u'clientconfig.depot.id', description=u'Depotserver to use', possibleValues=[], defaultValues=[depot.id] ) backend.config_createObjects(clientConfigDepotId) clientDepotMappingConfigState = ConfigState( configId=clientConfigDepotId.getId(), objectId=client.getId(), values=depot.getId() ) backend.configState_createObjects(clientDepotMappingConfigState) return client, depot