import de.hybris.platform.b2ctelcoservices.model.DeviceModel
import de.hybris.platform.catalog.daos.CatalogVersionDao
import de.hybris.platform.catalog.model.CatalogVersionModel
import de.hybris.platform.category.daos.CategoryDao
import de.hybris.platform.category.model.CategoryModel
import de.hybris.platform.servicelayer.model.ModelService
import si.telekom.eshop.core.enums.TestProductStatusEnum

prodCatalogId = "tseshopProductCatalog"
prodCatalogVersion = "Staged"
testCatalogId = "tseshopTestCatalog"
testCatalogVersion = "Test"

ModelService modelService = spring.getBean("modelService")
CatalogVersionDao catalogVersionDao = spring.getBean("catalogVersionDao")
CategoryDao categoryDao = spring.getBean("categoryDao")

CatalogVersionModel prodCatalog = catalogVersionDao.findCatalogVersions(prodCatalogId, prodCatalogVersion).get(0)
CatalogVersionModel testCatalog = catalogVersionDao.findCatalogVersions(testCatalogId, testCatalogVersion).get(0)

CategoryModel productCategory1 = categoryDao.findCategoriesByCode(prodCatalog, "mobilephones").get(0)

code = "testCode" + new Random().nextInt(10000).toString()

println("Code is : " + code)

DeviceModel testProduct = modelService.create(DeviceModel.class)
testProduct.setCode(code)
testProduct.setPlmSourceCategory(productCategory1)
testProduct.setSupercategories([productCategory1])
testProduct.setCatalogVersion(testCatalog)
testProduct.setFriendlyUrlIdentifier(code)

modelService.save(testProduct)

DeviceModel stagedProduct = modelService.create(DeviceModel.class)
stagedProduct.setCode(code)
stagedProduct.setPlmSourceCategory(productCategory1)
stagedProduct.setSupercategories([productCategory1])
stagedProduct.setCatalogVersion(prodCatalog)
stagedProduct.setFriendlyUrlIdentifier(code)

modelService.save(stagedProduct)

testProduct.setTestProductStatus(TestProductStatusEnum.TESTED)

modelService.save(testProduct)