/*
 * Copyright 2003-2010 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package gls.scope

import gls.CompilableTestSupport

public class MultipleDefinitionOfSameVariableTest extends CompilableTestSupport {

    public void testInSameBlock() {
        shouldNotCompile """
            def foo = 1
            def foo = 2
        """

        shouldNotCompile """
            class Foo {
                def foo() {
                    def bar=1
                    def bar=2
                }
            }
        """
    }

    public void testInSubBlocks() {
        shouldNotCompile """
             def foo = 1
             5.times { def foo=2 }
        """

        shouldNotCompile """
            def foo = 1
            label1: { def foo=2 }
        """

        shouldNotCompile """
            def foo = 1
            for (i in []) { def foo=2 }
        """

        shouldNotCompile """
            def foo = 1
            while (true) { def foo=2 }
        """
    }

    public void testInNestedClosure() {
        shouldNotCompile """
            def foo = 1
            5.times { 6.times {def foo=2 }
        """

        assertScript """
            def foo = 1
            5.times { 6.times {foo=2 } }
            assert foo == 2
        """
    }

    public void testBindingHiding() {
        assertScript """
            foo = 1
            def foo = 3
            assert foo==3
            assert this.foo == 1
            assert binding.foo == 1
        """
    }

    public void testBindingAccessInMethod() {
        assertScript """
            def methodUsingBinding() {
                try {
                    s = "  bbb  ";
                } finally {
                    s = s.trim();
                }
                assert s == "bbb"
            }
            methodUsingBinding()
            assert s == "bbb"
        """
    }

    public void testMultipleOfSameName() {
        shouldNotCompile """
            class DoubleField {
                def zero = 0
                def zero = 0
            }
        """
    }

    void testPropertyField() {
        shouldCompile """
            class A {
                def foo
                private foo
            }
        """
    }

    void testFieldProperty() {
        shouldCompile """
            class A {
                private foo
                def foo
            }
        """
    }

    void testFieldPropertyProperty() {
        shouldNotCompile """
            class A {
                private foo
                def foo
                def foo
            }
        """
    }

    void testPropertyFieldField() {
        shouldNotCompile """
            class A {
                def foo
                private foo
                private foo
            }
        """
    }

}