@Component
class AuthorizationFilter(
    val authApi: authApi
) : OncePerRequestFilter() {
    override fun doFilterInternal(request: HttpServletRequest, response: HttpServletResponse, filterChain: FilterChain) {
        if (request.method.equals("OPTIONS")) {
            filterChain.doFilter(request, response)
            return
        val token = request.getHeader("authorization")
        if (token == null) {
            response.sendError(401)
            return
        runCatching {
            authApi.authorize(token.replace("Bearer ", ""))
        }.onSuccess {
            AuthorizationContext.set(it)
            filterChain.doFilter(request, response)
        }.onFailure {
            it.printStackTrace()
            response.sendError(401)
    }

@TestInstance(TestInstance.Lifecycle.PER_CLASS)
@ExtendWith(SpringExtension::class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@ActiveProfiles("test")
class SocketIOServerTest {
    @MockBean
    lateinit var mockedApiComponent: AuthApi
    @Autowired
    lateinit var boardRepository: BoardRepository
    @Autowired
    private lateinit var servletRegistrationBean: ServletRegistrationBean<SocketIOServer>
    private lateinit var socketIOServer: SocketIOServer
    @LocalServerPort
    private val serverPort: String? = null
    lateinit var clientSocket: Socket
    private val userId = 1
    private val groupId = 123
    private val admin = false
    private val auth = Authorization("token", userId, groupId, admin)
    private val objectMapper = ObjectMapper()
    @BeforeAll
    fun connect() {
        AuthorizationContext.set(auth)
        Mockito.`when`(mockedApiComponent.authorize(anyOrNull())).thenReturn(auth)
        socketIOServer = servletRegistrationBean.servlet
        clientSocket = IO.socket("http://localhost:${serverPort}", IO.Options.builder().setExtraHeaders(mutableMapOf(Pair("Authorization", listOf("Bearer token")))).build())
        clientSocket.on(Socket.EVENT_CONNECT) {
            println("client connected")
        clientSocket.on(Socket.EVENT_DISCONNECT) {
            println("client disconnected")
        clientSocket.connect()
    @Test
    fun testPingPong() {
        var finished = false
        clientSocket.on("pong") {
            println("event: ${it[0]}")
            val pongTime = (it[0] as String).substring(18, 31).toLong()
            assertTrue(System.currentTimeMillis() - pongTime < 1000)
            finished = true
        clientSocket.emit("ping")
        while (!finished) Thread.yield()
    @Test
    fun testBasicNotification(){
        clientSocket.on("basic_notification"){
            println(Arrays.toString(it))
        socketIOServer.send(SocketIOEvent("${groupId}","basic_notification","data"))
        Thread.sleep(1000)
    @Test
    fun testBoardNotification() {
        clientSocket.on("entity_create") {
            val event = it[0] as String
            println("event: $event")
            val eventValue = objectMapper.readValue(event, Map::class.java)
            val entityValue = eventValue["entity"] as Map<*, *>
            assertEquals("BOARD", eventValue["entity_type"])
            assertEquals("board name", entityValue["name"])
            assertEquals(groupId, entityValue["groupId"])
            assertEquals(userId, entityValue["created_by"])
            assertEquals(userId, entityValue["last_modified_by"])
        val board = boardRepository.save(Board(groupId, "board name"))