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
|
// Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.jetbrains.python
import com.intellij.openapi.fileEditor.FileDocumentManager
import com.intellij.openapi.util.TextRange
import com.jetbrains.python.codeInsight.typing.PyTypeShed
import com.jetbrains.python.codeInsight.userSkeletons.PyUserSkeletonsUtil.getUserSkeletonsDirectory
import com.jetbrains.python.debugger.PyLineBreakpointType
import com.jetbrains.python.fixtures.PyTestCase
class PyLineBreakpointTypeTest : PyTestCase() {
// PY-16932
fun testPutAtUserSkeleton() {
val skeletonsDir = getUserSkeletonsDirectory()
val pythonFile = skeletonsDir!!.findFileByRelativePath("nose/tools/__init__.py")
val line = 13
val document = FileDocumentManager.getInstance().getDocument(pythonFile!!)
val range = TextRange.create(document!!.getLineStartOffset(line), document.getLineEndOffset(line))
assertEquals(" pass", document.getText(range))
assertFalse(PyLineBreakpointType().canPutAt(pythonFile, line, myFixture.project))
}
// PY-16932
fun testPutAtSkeleton() {
runWithAdditionalFileInSkeletonDir("my_mod.py", "class A:\n def method(self):\n print(\"ok\")") { pythonFile ->
val line = 2
val document = FileDocumentManager.getInstance().getDocument(pythonFile)
val range = TextRange.create(document!!.getLineStartOffset(line), document.getLineEndOffset(line))
assertEquals(" print(\"ok\")", document.getText(range))
assertFalse(PyLineBreakpointType().canPutAt(pythonFile, line, myFixture.project))
}
}
// PY-16932
fun testPutAtPythonStub() {
val pythonFile = PyTypeShed.directory!!.findFileByRelativePath("stdlib/@python2/__builtin__.pyi")
val line = 46
val document = FileDocumentManager.getInstance().getDocument(pythonFile!!)
val range = TextRange.create(document!!.getLineStartOffset(line), document.getLineEndOffset(line))
assertEquals("_T = TypeVar(\"_T\")", document.getText(range))
assertFalse(PyLineBreakpointType().canPutAt(pythonFile, line, myFixture.project))
}
fun testCorrect() {
val pythonFile = myFixture.configureByText(PythonFileType.INSTANCE, "def foo():\n" +
" print(1)").virtualFile
val line = 1
val document = FileDocumentManager.getInstance().getDocument(pythonFile!!)
val range = TextRange.create(document!!.getLineStartOffset(line), document.getLineEndOffset(line))
assertEquals(" print(1)", document.getText(range))
assertTrue(PyLineBreakpointType().canPutAt(pythonFile, line, myFixture.project))
}
}
|