/*
 * Copyright 2018, TeamDev. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package io.spine.tools.compiler.annotation.check;

import org.checkerframework.checker.nullness.qual.Nullable;
import org.jboss.forge.roaster.model.impl.AbstractJavaSource;
import org.jboss.forge.roaster.model.source.JavaClassSource;
import org.jboss.forge.roaster.model.source.JavaSource;

import java.util.Optional;

import static com.google.common.base.Preconditions.checkNotNull;
import static io.spine.tools.compiler.annotation.check.Annotations.findInternalAnnotation;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;

public class NestedTypesAnnotationCheck implements SourceCheck {

    private final boolean shouldBeAnnotated;

    public NestedTypesAnnotationCheck(boolean shouldBeAnnotated) {
        this.shouldBeAnnotated = shouldBeAnnotated;
    }

    @Override
    public void accept(@Nullable AbstractJavaSource<JavaClassSource> outerClass) {
        checkNotNull(outerClass);
        for (JavaSource<?> nestedType : outerClass.getNestedTypes()) {
            Optional<?> annotation = findInternalAnnotation(nestedType);
            if (shouldBeAnnotated) {
                assertNotNull(annotation);
            } else {
                assertNull(annotation);
            }
        }
    }
}

@@ -1534,23 +1534,21 @@
 s.assert
-NotNull
+False
 ;%0Aimport
@@ -1594,20 +1594,20 @@
 s.assert
-Null
+True
 ;%0A%0Apubli
@@ -2173,23 +2173,20 @@
   assert
-NotNull
+True
 (annotat
@@ -2180,32 +2180,44 @@
 tTrue(annotation
+.isPresent()
 );%0A            %7D
@@ -2246,20 +2246,21 @@
   assert
-Null
+False
 (annotat
@@ -2262,16 +2262,28 @@
 notation
+.isPresent()
 );%0A     

Fix a test check.