test/bdd/steps/results.py

   1 """ Steps that check results.
   2 """
   3
   4 from nose.tools import * # for assert functions
   5
   6 def compare(operator, op1, op2):
   7     if operator == 'less than':
   8         return op1 < op2
   9     elif operator == 'more than':
  10         return op1 > op2
  11     elif operator == 'exactly':
  12         return op1 == op2
  13     elif operator == 'at least':
  14         return op1 >= op2
  15     elif operator == 'at most':
  16         return op1 <= op2
  17     else:
  18         raise Exception("unknown operator '%s'" % operator)
  19
  20 @step(u'(?P<operator>less than|more than|exactly|at least|at most) (?P<number>\d+) results? (?:is|are) returned')
  21 def validate_result_number(context, operator, number):
  22     numres = len(context.response.result)
  23     ok_(compare(operator, numres, int(number)),
  24         "Bad number of results: expected %s %s, got %d." % (operator, number, numres))
  25
  26
  27 @then(u'results contain')
  28 def step_impl(context):
  29     context.execute_steps("then at least 1 result is returned")
  30
  31     for line in context.table:
  32         context.response.match_row(line)
  33