import unittest

import sys
import os.path
import mozunit

from Expression import Expression, Context

class TestContext(unittest.TestCase):
  """
  Unit tests for the Context class
  """

  def setUp(self):
    self.c = Context()
    self.c['FAIL'] = 'PASS'

  def test_string_literal(self):
    """test string literal, fall-through for undefined var in a Context"""
    self.assertEqual(self.c['PASS'], 'PASS')

  def test_variable(self):
    """test value for defined var in the Context class"""
    self.assertEqual(self.c['FAIL'], 'PASS')

  def test_in(self):
    """test 'var in context' to not fall for fallback"""
    self.assert_('FAIL' in self.c)
    self.assert_('PASS' not in self.c)

class TestExpression(unittest.TestCase):
  """
  Unit tests for the Expression class
  evaluate() is called with a context {FAIL: 'PASS'}
  """

  def setUp(self):
    self.c = Context()
    self.c['FAIL'] = 'PASS'

  def test_string_literal(self):
    """Test for a string literal in an Expression"""
    self.assertEqual(Expression('PASS').evaluate(self.c), 'PASS')

  def test_variable(self):
    """Test for variable value in an Expression"""
    self.assertEqual(Expression('FAIL').evaluate(self.c), 'PASS')

  def test_not(self):
    """Test for the ! operator"""
    self.assert_(Expression('!0').evaluate(self.c))
    self.assert_(not Expression('!1').evaluate(self.c))

  def test_equals(self):
    """ Test for the == operator"""
    self.assert_(Expression('FAIL == PASS').evaluate(self.c))

  def test_notequals(self):
    """ Test for the != operator"""
    self.assert_(Expression('FAIL != 1').evaluate(self.c))

  def test_logical_and(self):
    """ Test for the && operator"""
    self.assertTrue(Expression('PASS == PASS && PASS != NOTPASS').evaluate(self.c))

  def test_logical_or(self):
    """ Test for the || operator"""
    self.assertTrue(Expression('PASS == NOTPASS || PASS != NOTPASS').evaluate(self.c))

  def test_logical_ops(self):
    """ Test for the && and || operators precedence"""
    # Would evaluate to false if precedence was wrong
    self.assertTrue(Expression('PASS == PASS || PASS != NOTPASS && PASS == NOTPASS').evaluate(self.c))

  def test_defined(self):
    """ Test for the defined() value"""
    self.assertTrue(Expression('defined(FAIL)').evaluate(self.c))
    self.assertTrue(Expression('!defined(PASS)').evaluate(self.c))


if __name__ == '__main__':
  mozunit.main()
