FilesExpand file tree

main

/

embedded_python_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

84 lines (68 loc) · 1.99 KB

main

/

embedded_python_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

84 lines (68 loc) · 1.99 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package python

import (

"bytes"

"fmt"

"io"

"math/rand"

"testing"

"github.com/divinerapier/go-embed-python/internal"

"github.com/stretchr/testify/assert"

)

func TestEmbeddedPython(t *testing.T) {

rndName := fmt.Sprintf("test-%d", rand.Uint32())

ep, err := NewEmbeddedPython(rndName)

assert.NoError(t, err)

defer ep.Cleanup()

path := ep.GetExtractedPath()

assert.NotEqual(t, path, "")

pexe, _ := ep.GetExePath()

assert.True(t, internal.Exists(pexe))

cmd, _ := ep.PythonCmd("-c", "print('test test')")

stdout, err := cmd.StdoutPipe()

assert.NoError(t, err)

defer stdout.Close()

err = cmd.Start()

assert.NoError(t, err)

stdoutStr, err := io.ReadAll(stdout)

assert.NoError(t, err)

err = cmd.Wait()

assert.NoError(t, err)

stdoutStr = bytes.TrimSpace(stdoutStr)

assert.Equal(t, "test test", string(stdoutStr))

}

func TestPrintSystemInfo(t *testing.T) {

getSystemInfo := `

import platform, sys

print("system info:")

print("sys.version=" + sys.version)

print("platform.python_version=" + platform.python_version())

print("platform.machine=" + platform.machine())

print("platform.version=" + platform.version())

print("platform.platform=" + platform.platform())

print("platform.release=" + platform.release())

print("platform.uname=" + str(platform.uname()))

print("platform.system=" + platform.system())

print("platform.processor=" + platform.processor())

`

rndName := fmt.Sprintf("test-%d", rand.Uint32())

ep, err := NewEmbeddedPython(rndName)

assert.NoError(t, err)

defer ep.Cleanup()

path := ep.GetExtractedPath()

assert.NotEqual(t, path, "")

pexe, _ := ep.GetExePath()

assert.True(t, internal.Exists(pexe))

cmd, _ := ep.PythonCmd("-c", getSystemInfo)

stdout, _ := cmd.StdoutPipe()

stderr, _ := cmd.StderrPipe()

defer stdout.Close()

defer stderr.Close()

err = cmd.Start()

assert.NoError(t, err)

stdoutStr, _ := io.ReadAll(stdout)

stderrStr, _ := io.ReadAll(stderr)

t.Log("stdout=" + string(stdoutStr))

t.Log("stderr=" + string(stderrStr))

err = cmd.Wait()

assert.NoError(t, err)

}