io.rocketbase.commons.test.BaseIntegrationTest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons-auth-test Show documentation

There is a newer version: 4.4.1

package io.rocketbase.commons.test;


import io.rocketbase.commons.model.AppUserEntity;
import io.rocketbase.commons.security.JwtTokenService;
import io.rocketbase.commons.test.adapters.AuthRestTestTemplate;
import lombok.Getter;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.junit4.SpringRunner;

import javax.annotation.Resource;

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles(profiles = "test")
public abstract class BaseIntegrationTest {

    @Getter
    @Value("http://localhost:${local.server.port}")
    protected String baseUrl;

    @Resource
    private AppUserPersistenceTestService appUserPersistenceTestService;

    @Resource
    private JwtTokenService jwtTokenService;

    protected AppUserEntity getAppUser() {
        return getAppUser("user");
    }

    protected AppUserEntity getAppAdminUser() {
        return getAppUser("admin");
    }

    protected AppUserEntity getAppUser(String username) {
        return appUserPersistenceTestService.findByUsername(username).get();
    }

    protected AuthRestTestTemplate getAuthRestTemplate() {
        return getAuthRestTemplate("user");
    }

    protected AuthRestTestTemplate getAuthRestAdminTemplate() {
        return getAuthRestTemplate("admin");
    }

    protected AuthRestTestTemplate getAuthRestTemplate(String username) {
        return new AuthRestTestTemplate(getAppUser(username), jwtTokenService);
    }

    @Before
    public void beforeEachTest() {
        appUserPersistenceTestService.resetData();
    }

}