openide/java/java-tests/testSrc/com/intellij/java/codeInsight/codeVision/JavaCodeVisionProviderTest.kt

// Copyright 2000-2023 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package com.intellij.java.codeInsight.codeVision

import com.intellij.codeInsight.daemon.impl.JavaInheritorsCodeVisionProvider
import com.intellij.codeInsight.daemon.impl.JavaReferencesCodeVisionProvider
import com.intellij.testFramework.utils.codeVision.CodeVisionTestCase
import org.intellij.lang.annotations.Language

class JavaCodeVisionProviderTest : CodeVisionTestCase() {
  fun testMethodUsages() = doTest("""
    class A { /*<# [no usages] #>*/
    
      void foo() { /*<# [1 usage] #>*/
      }
      
      void bar() { /*<# [no usages] #>*/
        foo();
      }
    }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId)

  fun testClassInheritors() = doTest("""
      class A { /*<# [no usages] #>*/
      
        class B {} /*<# [2 usages   2 inheritors] #>*/
        
        class B1 extends B {} /*<# [no usages] #>*/
      
        class B2 extends B {} /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId, JavaInheritorsCodeVisionProvider().groupId)

  fun testInterfaceInheritors() = doTest("""
      class A { /*<# [no usages] #>*/
        interface B {} /*<# [1 usage   1 implementation] #>*/
        
        class B1 implements B {} /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId, JavaInheritorsCodeVisionProvider().groupId)

  fun testMethodOverrides() = doTest("""
      class A { /*<# [1 usage   1 inheritor] #>*/
        void foo(){} /*<# [no usages   1 override] #>*/
      }
      
      class B extends A { /*<# [no usages] #>*/
        void foo(){} /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId, JavaInheritorsCodeVisionProvider().groupId)

  fun testDefaultMethodOverrides() = doTest("""
      interface A { /*<# [2 usages   2 implementations] #>*/
        default void foo(){} /*<# [no usages   2 overrides] #>*/
      }
      
      class B implements A { /*<# [no usages] #>*/
        public void foo(){} /*<# [no usages] #>*/
      }
      
      class B2 implements A { /*<# [no usages] #>*/
        public void foo(){} /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId, JavaInheritorsCodeVisionProvider().groupId)

  fun testAbstractMethodImplementations() = doTest("""
      interface A {/*<# [2 usages   2 implementations] #>*/
        void foo();/*<# [no usages   2 implementations] #>*/
      }
      
      class B implements A { /*<# [no usages] #>*/
        public void foo(){} /*<# [no usages] #>*/
      }
      
      class B2 implements A { /*<# [no usages] #>*/
        public void foo(){} /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId, JavaInheritorsCodeVisionProvider().groupId)

  fun testEnumMembers() = doTest("""
      class A { /*<# [no usages] #>*/
      
        enum E { /*<# [6 usages] #>*/
          E1, E2, E3, E4 /*<# [1 usage] #>*/
        }
      
        E foo() { /*<# [no usages] #>*/
          bar(E.E1, E.E2, E.E3, E.E4);
        }
      
        void bar(E... e) {} /*<# [1 usage] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId)

  fun testClassAtZeroOffset() = doTest("""      
      class A { /*<# [no usages] #>*/
        enum E { /*<# [6 usages] #>*/
          E1, E2, E3, E4 /*<# [1 usage] #>*/
        }
      
        E foo() { /*<# [no usages] #>*/
          bar(E.E1, E.E2, E.E3, E.E4);
        }
      
        void bar(E... e) {} /*<# [1 usage] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId)

  fun testClassAfterPackageStatement() = doTest("""
      package com.company;
      
      class A{} /*<# [1 usage] #>*/
      
      class B { /*<# [no usages] #>*/
      
       void use() { /*<# [no usages] #>*/
         new A();
       }
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId)

  fun testFieldOnFirstLineOfInterfaceHasLenses() = doTest("""
      package codeLenses;
      
      public interface Interface { /*<# [no usages] #>*/
          String s = "asd"; /*<# [no usages] #>*/
      
          String asd = "asd"; /*<# [no usages] #>*/
      }
    """.trimIndent(), JavaReferencesCodeVisionProvider().groupId)

  private fun doTest(@Language("JAVA") text: String, vararg enabledProviderGroupIds: String) {
    testProviders(text, "test.java", *enabledProviderGroupIds)
  }
}