def test_create_batch_op_no_collection(self): json_obj = { "resource": "foo", "query": "status=FAIL&job=mainline" } op = create_batch_operation(json_obj, {}) self.assertIsNone(op)
def test_create_batch_op_generic(self): json_obj = { "resource": "boot", "query": "status=FAIL&job=mainline", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchOperation)
def test_create_batch_op_no_collection(self): json_obj = { "document_id": "defconfig", "query": "status=FAIL&job=mainline", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsNone(op)
def test_create_batch_fake_op(self, mocked_mongocl): json_obj = { "collection": "foo", "query": "status=PASS&job=foo", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchOperation)
def test_create_batch_distinct(self): json_obj = { "resource": "boot", "distinct": "board", "operation_id": "distinct-board" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchDistinctOperation) self.assertEqual("board", op.distinct)
def test_create_batch_test_group_op(self): json_obj = { "resource": "test_group", "query": "status=PASS&job=foo", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchTestGroupOperation) self.assertEqual("test_group", op.resource)
def test_create_batch_build_op(self): json_obj = { "resource": "build", "query": "status=PASS&job=foo", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchBuildOperation) self.assertEqual("build", op.resource)
def test_create_batch_op_count(self): json_obj = { "collection": "count", "document_id": "defconfig", "query": "status=FAIL&job=mainline", "operation_id": "foo" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchCountOperation)
def test_create_batch_op_count(self): json_obj = { "resource": "count", "document": "build", "query": "status=FAIL&job=mainline", "operation_id": "op_id" } op = create_batch_operation(json_obj, {}) self.assertIsInstance(op, BatchCountOperation) self.assertEqual("build", op.document) self.assertEqual("count", op.resource) self.assertEqual("op_id", op.operation_id)
def test_create_batch_op_none(self): op = create_batch_operation(None, None) self.assertIsNone(op)