-
Notifications
You must be signed in to change notification settings - Fork 774
Expand file tree
/
Copy pathTestInterrupt.cs
More file actions
96 lines (80 loc) · 2.98 KB
/
TestInterrupt.cs
File metadata and controls
96 lines (80 loc) · 2.98 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
using System;
using System.Diagnostics;
using System.Threading;
using System.Threading.Tasks;
using NUnit.Framework;
using Python.Runtime;
namespace Python.EmbeddingTest
{
public class TestInterrupt
{
PyObject threading;
[OneTimeSetUp]
public void SetUp()
{
// workaround for assert tlock.locked() warning
threading = Py.Import("threading");
}
[OneTimeTearDown]
public void Dispose()
{
threading.Dispose();
}
[Test]
public void PythonThreadIDStable()
{
long pythonThreadID = 0;
long pythonThreadID2 = 0;
var asyncCall = Task.Factory.StartNew(() =>
{
using (Py.GIL())
{
Interlocked.Exchange(ref pythonThreadID, (long)PythonEngine.GetPythonThreadID());
Interlocked.Exchange(ref pythonThreadID2, (long)PythonEngine.GetPythonThreadID());
}
});
var timeout = Stopwatch.StartNew();
IntPtr threadState = PythonEngine.BeginAllowThreads();
while (Interlocked.Read(ref pythonThreadID) == 0 || Interlocked.Read(ref pythonThreadID2) == 0)
{
Assert.Less(timeout.Elapsed, TimeSpan.FromSeconds(5), "thread IDs were not assigned in time");
}
PythonEngine.EndAllowThreads(threadState);
Assert.That(asyncCall.Wait(TimeSpan.FromSeconds(5)), Is.True, "Async thread has not finished in time");
Assert.That(pythonThreadID2, Is.EqualTo(pythonThreadID));
Assert.NotZero(pythonThreadID);
}
[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(@"
try:
import time
while True:
time.sleep(0.2)
except KeyboardInterrupt:
pass");
}
});
var timeout = Stopwatch.StartNew();
IntPtr threadState = PythonEngine.BeginAllowThreads();
while (Interlocked.Read(ref pythonThreadID) == 0)
{
Assert.Less(timeout.Elapsed, TimeSpan.FromSeconds(5), "thread ID was not assigned in time");
}
PythonEngine.EndAllowThreads(threadState);
int interruptReturnValue = PythonEngine.Interrupt((ulong)Interlocked.Read(ref pythonThreadID));
Assert.That(interruptReturnValue, Is.EqualTo(1));
threadState = PythonEngine.BeginAllowThreads();
Assert.That(asyncCall.Wait(TimeSpan.FromSeconds(5)), Is.True, "Async thread was not interrupted in time");
PythonEngine.EndAllowThreads(threadState);
Assert.That(asyncCall.Result, Is.EqualTo(0));
}
}
}