def test_kvp_strict(self): """ Add a testsuite. """ parser = TestsuiteTestCase.parser_create() args = parser.parse_args("kvp add key1 value1.1".split()) self.assertTrue(args.func(args)) args = parser.parse_args("kvp type key1 STRICT".split()) self.assertTrue(args.func(args)) with self.assertRaises(KVP.DoesNotExist): KVP.get_or_create("key1", "value1.2")
def get_or_create(productname, branchname, order=-1): """ Get or create a testplan in a certain order. @param product_name Is the name of the product @param order order is the location of the testplan in the list of testplans. The order effects the location the testplan appears on web pages. """ from testdb.models import Product from testdb.models import KVP logging.info("adding product %s %s", productname, branchname) ## # Make sure to track branch and product against KVP. This is done # incase user decides to make these keys strict. KVP.get_or_create("branch", branchname) KVP.get_or_create("product", productname) if order == -1: find = Product.objects.filter(context__name=CONTEXT) try: order = find.order_by("-order")[0].order except IndexError: order = 0 order += 1 logging.info("using order %d", order) ## # Order is specified so now we have to move something. products = Product.objects.filter(context__name=CONTEXT, order__gte=order).order_by("order") current_order = order for product in products: if product.order == current_order: product.order += 1 logging.debug("updating order %s to %d", productname, product.order) product.save() current_order = product.order (product, created) = Product.get_or_create(CONTEXT, productname, branchname) product.order = order product.save() return (product, created)
def add_key(context, order, key, value): """ Add a key into the testplan. """ from testdb.models import Testplan from testdb.models import Testsuite from testdb.models import Key from testdb.models import KVP logging.info("add %s=%s to testsuite %s %s", key, value, context, order) context = Testplan.context_get(context) testplan = Testplan.objects.get(context=context) planorder = testplan.testplanorder_set.get(order=order) testsuite = Testsuite.objects.get(context=context, testplanorder=planorder) (testkey, _) = KVP.get_or_create(key=key, value=value) testsuite.key_change(testkey) ## # (key, _) = Key.objects.get_or_create(value=key) testplan.testplankvp_set.get_or_create(key=key) # ## return (testsuite, testkey)