import org.checkerframework.checker.nullness.qual.*;
import org.checkerframework.checker.nullness.qual.EnsuresNonNull;

public class AssertAfter {

    protected @Nullable String value;

    @EnsuresNonNull("value")
    public boolean setRepNonNull() {
        value = "";
        return true;
    }

    public void plain() {
        // :: error: (dereference.of.nullable)
        value.toString();
    }

    public void testAfter() {
        setRepNonNull();
        value.toString();
    }

    public void testBefore() {
        // :: error: (dereference.of.nullable)
        value.toString();
        setRepNonNull();
    }

    public void withCondition(@Nullable String t) {
        if (t == null) {
            setRepNonNull();
        }
        // :: error: (dereference.of.nullable)
        value.toString();
    }

    public void inConditionInTrue() {
        if (setRepNonNull()) {
            value.toString();
        } else {
            // nothing to do
        }
    }

    // skip-test: Come back when working on improved flow
    public void asCondition() {
        if (setRepNonNull()) {
        } else {
            value.toString(); // valid!
        }
    }
}

// Test that private fields can be mentioned in pre- and post-conditions.

class A {
    private @Nullable String privateField = null;

    @EnsuresNonNull("privateField")
    public void m1() {
        privateField = "hello";
    }

    @RequiresNonNull("privateField")
    public void m2() {}
}

class B {

    void f() {
        A a = new A();
        a.m1();
        a.m2();
    }
}
