/*
 * Copyright 2000-2016 JetBrains s.r.o.
 *
 * 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 org.jetbrains.plugins.groovy.intentions;

/**
 * @author Max Medvedev
 */
public class ReplaceIfWithTernaryTest extends GrIntentionTestCase {
  public ReplaceIfWithTernaryTest() {
    super("Replace with ?:");
  }

  public void testReturn() throws Exception {
    doTextTest("""
                 def foo() {
                     i<caret>f (cond()) {
                         return 'a'
                     }
                     else return 'b'
                 }
                 """, """
                 def foo() {
                     return cond() ? 'a' : 'b'
                 }
                 """);
  }

  public void testAssign() {
    doTextTest("""
                 def a
                 
                 i<caret>f (cond()) {
                     a = 'a'
                 }
                 else {
                     a = 'b'
                 }
                 """, """
                 def a
                 
                 a = cond() ? 'a' : 'b'
                 """);
  }

  public void testAntiAssign() {
    doAntiTest("""
                 def a
                 def b
                 
                 i<caret>f (cond) {
                     a = 'a'
                 }
                 else {
                     b = 'b'
                 }
                 """);
  }
}
