import WebIDL

def WebIDLTest(parser, harness):
    parser.parse("""
        interface TestMethods {
          void basic();
          static void basicStatic();
          void basicWithSimpleArgs(boolean arg1, byte arg2, unsigned long arg3);
          boolean basicBoolean();
          static boolean basicStaticBoolean();
          boolean basicBooleanWithSimpleArgs(boolean arg1, byte arg2, unsigned long arg3);
          void optionalArg(optional byte? arg1, optional sequence<byte> arg2);
          void variadicArg(byte?... arg1);
          object getObject();
          void setObject(object arg1);
          void setAny(any arg1);
          float doFloats(float arg1);
        };
    """)

    results = parser.finish()

    harness.ok(True, "TestMethods interface parsed without error.")
    harness.check(len(results), 1, "Should be one production.")
    iface = results[0]
    harness.ok(isinstance(iface, WebIDL.IDLInterface),
               "Should be an IDLInterface")
    harness.check(iface.identifier.QName(), "::TestMethods", "Interface has the right QName")
    harness.check(iface.identifier.name, "TestMethods", "Interface has the right name")
    harness.check(len(iface.members), 12, "Expect 12 members")

    methods = iface.members

    def checkArgument(argument, QName, name, type, optional, variadic):
        harness.ok(isinstance(argument, WebIDL.IDLArgument),
                   "Should be an IDLArgument")
        harness.check(argument.identifier.QName(), QName, "Argument has the right QName")
        harness.check(argument.identifier.name, name, "Argument has the right name")
        harness.check(str(argument.type), type, "Argument has the right return type")
        harness.check(argument.optional, optional, "Argument has the right optional value")
        harness.check(argument.variadic, variadic, "Argument has the right variadic value")

    def checkMethod(method, QName, name, signatures,
                    static=False, getter=False, setter=False, creator=False,
                    deleter=False, legacycaller=False, stringifier=False):
        harness.ok(isinstance(method, WebIDL.IDLMethod),
                   "Should be an IDLMethod")
        harness.ok(method.isMethod(), "Method is a method")
        harness.ok(not method.isAttr(), "Method is not an attr")
        harness.ok(not method.isConst(), "Method is not a const")
        harness.check(method.identifier.QName(), QName, "Method has the right QName")
        harness.check(method.identifier.name, name, "Method has the right name")
        harness.check(method.isStatic(), static, "Method has the correct static value")
        harness.check(method.isGetter(), getter, "Method has the correct getter value")
        harness.check(method.isSetter(), setter, "Method has the correct setter value")
        harness.check(method.isCreator(), creator, "Method has the correct creator value")
        harness.check(method.isDeleter(), deleter, "Method has the correct deleter value")
        harness.check(method.isLegacycaller(), legacycaller, "Method has the correct legacycaller value")
        harness.check(method.isStringifier(), stringifier, "Method has the correct stringifier value")
        harness.check(len(method.signatures()), len(signatures), "Method has the correct number of signatures")

        sigpairs = zip(method.signatures(), signatures)
        for (gotSignature, expectedSignature) in sigpairs:
            (gotRetType, gotArgs) = gotSignature
            (expectedRetType, expectedArgs) = expectedSignature

            harness.check(str(gotRetType), expectedRetType,
                          "Method has the expected return type.")

            for i in range(0, len(gotArgs)):
                (QName, name, type, optional, variadic) = expectedArgs[i]
                checkArgument(gotArgs[i], QName, name, type, optional, variadic)

    checkMethod(methods[0], "::TestMethods::basic", "basic", [("Void", [])])
    checkMethod(methods[1], "::TestMethods::basicStatic", "basicStatic",
                [("Void", [])], static=True)
    checkMethod(methods[2], "::TestMethods::basicWithSimpleArgs",
                "basicWithSimpleArgs",
       [("Void",
        [("::TestMethods::basicWithSimpleArgs::arg1", "arg1", "Boolean", False, False),
         ("::TestMethods::basicWithSimpleArgs::arg2", "arg2", "Byte", False, False),
         ("::TestMethods::basicWithSimpleArgs::arg3", "arg3", "UnsignedLong", False, False)])])
    checkMethod(methods[3], "::TestMethods::basicBoolean", "basicBoolean", [("Boolean", [])])
    checkMethod(methods[4], "::TestMethods::basicStaticBoolean", "basicStaticBoolean", [("Boolean", [])], static=True)
    checkMethod(methods[5], "::TestMethods::basicBooleanWithSimpleArgs",
                "basicBooleanWithSimpleArgs",
       [("Boolean",
        [("::TestMethods::basicBooleanWithSimpleArgs::arg1", "arg1", "Boolean", False, False),
         ("::TestMethods::basicBooleanWithSimpleArgs::arg2", "arg2", "Byte", False, False),
         ("::TestMethods::basicBooleanWithSimpleArgs::arg3", "arg3", "UnsignedLong", False, False)])])
    checkMethod(methods[6], "::TestMethods::optionalArg",
                "optionalArg",
       [("Void",
        [("::TestMethods::optionalArg::arg1", "arg1", "ByteOrNull", True, False),
         ("::TestMethods::optionalArg::arg2", "arg2", "ByteSequence", True, False)])])
    checkMethod(methods[7], "::TestMethods::variadicArg",
                "variadicArg",
       [("Void",
        [("::TestMethods::variadicArg::arg1", "arg1", "ByteOrNull", True, True)])])
    checkMethod(methods[8], "::TestMethods::getObject",
                "getObject", [("Object", [])])
    checkMethod(methods[9], "::TestMethods::setObject",
                "setObject",
       [("Void",
        [("::TestMethods::setObject::arg1", "arg1", "Object", False, False)])])
    checkMethod(methods[10], "::TestMethods::setAny",
                "setAny",
       [("Void",
        [("::TestMethods::setAny::arg1", "arg1", "Any", False, False)])])
    checkMethod(methods[11], "::TestMethods::doFloats",
                "doFloats",
       [("Float",
        [("::TestMethods::doFloats::arg1", "arg1", "Float", False, False)])])

    parser = parser.reset()
    threw = False
    try:
        parser.parse("""
          interface A {
            void foo(optional float bar = 1);
          };
        """)
        results = parser.finish()
    except Exception, x:
        threw = True
    harness.ok(not threw, "Should allow integer to float type corecion")

    parser = parser.reset()
    threw = False
    try:
        parser.parse("""
          interface A {
            [GetterThrows] void foo();
          };
        """)
        results = parser.finish()
    except Exception, x:
        threw = True
    harness.ok(threw, "Should not allow [GetterThrows] on methods")

    parser = parser.reset()
    threw = False
    try:
        parser.parse("""
          interface A {
            [SetterThrows] void foo();
          };
        """)
        results = parser.finish()
    except Exception, x:
        threw = True
    harness.ok(threw, "Should not allow [SetterThrows] on methods")

    parser = parser.reset()
    threw = False
    try:
        parser.parse("""
          interface A {
            [Throw] void foo();
          };
        """)
        results = parser.finish()
    except Exception, x:
        threw = True
    harness.ok(threw, "Should spell [Throws] correctly on methods")

    parser = parser.reset()
    threw = False
    try:
        parser.parse("""
          interface A {
            void __noSuchMethod__();
          };
        """)
        results = parser.finish()
    except Exception, x:
        threw = True
    harness.ok(threw, "Should not allow __noSuchMethod__ methods")