forked from pythonnet/pythonnet
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathTestInterrupt.cs
More file actions
64 lines (53 loc) · 1.67 KB
/
TestInterrupt.cs
File metadata and controls
64 lines (53 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
using System;
using System.Diagnostics;
using System.Threading;
using System.Threading.Tasks;
using NUnit.Framework;
using Python.Runtime;
namespace Python.EmbeddingTest
{
public class TestInterrupt
{
private IntPtr _threadState;
[OneTimeSetUp]
public void SetUp()
{
PythonEngine.Initialize();
_threadState = PythonEngine.BeginAllowThreads();
}
[OneTimeTearDown]
public void Dispose()
{
PythonEngine.EndAllowThreads(_threadState);
PythonEngine.Shutdown();
}
[Test]
public void InterruptTest()
{
long pythonThreadID = 0;
var asyncCall = Task.Factory.StartNew(() =>
{
using (Py.GIL())
{
Interlocked.Exchange(ref pythonThreadID, (long)PythonEngine.GetPythonThreadID());
return PythonEngine.RunSimpleString(@"
import time
while True:
time.sleep(0.2)");
}
});
var timeout = Stopwatch.StartNew();
while (Interlocked.Read(ref pythonThreadID) == 0)
{
Assert.Less(timeout.Elapsed, TimeSpan.FromSeconds(5), "thread ID was not assigned in time");
}
using (Py.GIL())
{
int interruptReturnValue = PythonEngine.Interrupt((ulong)Interlocked.Read(ref pythonThreadID));
Assert.AreEqual(1, interruptReturnValue);
}
Assert.IsTrue(asyncCall.Wait(TimeSpan.FromSeconds(5)), "Async thread was not interrupted in time");
Assert.AreEqual(-1, asyncCall.Result);
}
}
}