Synnefo » snf-ganeti

    ("rapi", qa_rapi.TestRapiQuery),

import random

from ganeti import query

from ganeti import compat

from ganeti import qlang

def TestRapiQuery():

  """Testing resource queries via remote API.

  """

  master_name = qa_utils.ResolveNodeName(qa_config.GetMasterNode())

  rnd = random.Random(7818)

  for what in constants.QR_VIA_RAPI:

    all_fields = query.ALL_FIELDS[what].keys()

    rnd.shuffle(all_fields)

    # No fields, should return everything

    result = _rapi_client.QueryFields(what)

    qresult = objects.QueryFieldsResponse.FromDict(result)

    AssertEqual(len(qresult.fields), len(all_fields))

    # One field

    result = _rapi_client.QueryFields(what, fields=["name"])

    qresult = objects.QueryFieldsResponse.FromDict(result)

    AssertEqual(len(qresult.fields), 1)

    # Specify all fields, order must be correct

    result = _rapi_client.QueryFields(what, fields=all_fields)

    qresult = objects.QueryFieldsResponse.FromDict(result)

    AssertEqual(len(qresult.fields), len(all_fields))

    AssertEqual([fdef.name for fdef in qresult.fields], all_fields)

    # Unknown field

    result = _rapi_client.QueryFields(what, fields=["_unknown!"])

    qresult = objects.QueryFieldsResponse.FromDict(result)

    AssertEqual(len(qresult.fields), 1)

    AssertEqual(qresult.fields[0].name, "_unknown!")

    AssertEqual(qresult.fields[0].kind, constants.QFT_UNKNOWN)

    # Try once more, this time without the client

    _DoTests([

      ("/2/query/%s/fields" % what, None, "GET", None),

      ("/2/query/%s/fields?fields=name,name,%s" % (what, all_fields[0]),

       None, "GET", None),

      ])

    # Try missing query argument

    try:

      _DoTests([

        ("/2/query/%s" % what, None, "GET", None),

        ])

    except rapi.client.GanetiApiError, err:

      AssertEqual(err.code, 400)

    else:

      raise qa_error.Error("Request missing 'fields' parameter didn't fail")

    def _Check(exp_fields, data):

      qresult = objects.QueryResponse.FromDict(data)

      AssertEqual([fdef.name for fdef in qresult.fields], exp_fields)

      if not isinstance(qresult.data, list):

        raise qa_error.Error("Query did not return a list")

    _DoTests([

      # Specify fields in query

      ("/2/query/%s?fields=%s" % (what, ",".join(all_fields)),

       compat.partial(_Check, all_fields), "GET", None),

      ("/2/query/%s?fields=name" % what,

       compat.partial(_Check, ["name"]), "GET", None),

      # Note the spaces

      ("/2/query/%s?fields=name,%%20name%%09,name%%20" % what,

       compat.partial(_Check, ["name"] * 3), "GET", None),

      # PUT with fields in query

      ("/2/query/%s?fields=name" % what,

       compat.partial(_Check, ["name"]), "PUT", {}),

      # Fields in body

      ("/2/query/%s" % what, compat.partial(_Check, all_fields), "PUT", {

         "fields": all_fields,

         }),

      ("/2/query/%s" % what, compat.partial(_Check, ["name"] * 4), "PUT", {

         "fields": ["name"] * 4,

         }),

      ])

    def _CheckFilter():

      _DoTests([

        # With filter

        ("/2/query/%s" % what, compat.partial(_Check, all_fields), "PUT", {

           "fields": all_fields,

           "filter": [qlang.OP_TRUE, "name"],

           }),

        ])

    if what == constants.QR_LOCK:

      # Locks can't be filtered

      try:

        _CheckFilter()

      except rapi.client.GanetiApiError, err:

        AssertEqual(err.code, 500)

      else:

        raise qa_error.Error("Filtering locks didn't fail")

    else:

      _CheckFilter()

    if what == constants.QR_NODE:

      # Test with filter

      (nodes, ) = _DoTests([("/2/query/%s" % what,

        compat.partial(_Check, ["name", "master"]), "PUT", {

        "fields": ["name", "master"],

        "filter": [qlang.OP_TRUE, "master"],

        })])

      qresult = objects.QueryResponse.FromDict(nodes)

      AssertEqual(qresult.data, [

        [[constants.RS_NORMAL, master_name], [constants.RS_NORMAL, True]],

        ])

  queryargs = "&".join("tag=%s" % i for i in tags)

    ("%s?%s" % (uri, queryargs), _VerifyReturnsJob, "PUT", None),

    ("%s?%s" % (uri, queryargs), _VerifyReturnsJob, "DELETE", None),