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)
