Skip to content

Commit

Permalink
Fix mocking of ThreadLocal (#2908)
Browse files Browse the repository at this point in the history
Fixes #2905

Co-authored-by: Roberto Trunfio <Roberto.Trunfio@appway.com>
  • Loading branch information
robertotru and Roberto Trunfio committed Feb 11, 2023
1 parent d6e39b8 commit 5153526
Show file tree
Hide file tree
Showing 2 changed files with 67 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -168,7 +168,7 @@ public boolean isMock(Object instance) {

@Override
public boolean isMocked(Object instance) {
return selfCallInfo.checkSelfCall(instance) && isMock(instance);
return isMock(instance) && selfCallInfo.checkSelfCall(instance);
}

@Override
Expand Down
66 changes: 66 additions & 0 deletions src/test/java/org/mockitousage/bugs/ThreadLocalTest.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockitousage.bugs;

import static org.mockito.Mockito.RETURNS_MOCKS;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;

import org.assertj.core.api.Assertions;
import org.junit.Test;
import org.mockitoutil.TestBase;

/**
* This was an issue reported in #2905. Mocking {@link ThreadLocal} or classes extending {@link ThreadLocal} was
* throwing a {@link StackOverflowError}.
*/
public class ThreadLocalTest extends TestBase {

@Test
public void mock_ThreadLocal_does_not_raise_StackOverflowError() {
StackOverflowError stackOverflowError =
Assertions.catchThrowableOfType(
() -> {
mock(ThreadLocal.class, RETURNS_MOCKS);
},
StackOverflowError.class);
Assertions.assertThat(stackOverflowError).isNull();
}

@Test
public void mock_class_extending_ThreadLocal_does_not_raise_StackOverflowError() {
StackOverflowError stackOverflowError =
Assertions.catchThrowableOfType(
() -> {
mock(SomeThreadLocal.class, RETURNS_MOCKS);
},
StackOverflowError.class);
Assertions.assertThat(stackOverflowError).isNull();
}

@Test
public void spy_ThreadLocal_does_not_raise_StackOverflowError() {
StackOverflowError stackOverflowError =
Assertions.catchThrowableOfType(
() -> {
spy(ThreadLocal.class);
},
StackOverflowError.class);
Assertions.assertThat(stackOverflowError).isNull();
}

@Test
public void spy_class_extending_ThreadLocal_does_not_raise_StackOverflowError() {
StackOverflowError stackOverflowError =
Assertions.catchThrowableOfType(
() -> {
spy(SomeThreadLocal.class);
},
StackOverflowError.class);
Assertions.assertThat(stackOverflowError).isNull();
}

static class SomeThreadLocal<T> extends ThreadLocal<T> {}
}

0 comments on commit 5153526

Please sign in to comment.