natspec_test.go 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. package natspec
  2. import (
  3. "testing"
  4. )
  5. func makeUserdoc(desc string) []byte {
  6. return []byte(`
  7. {
  8. "source": "...",
  9. "language": "Solidity",
  10. "languageVersion": 1,
  11. "methods": {
  12. "multiply(uint256)": {
  13. "notice": "` + desc + `"
  14. },
  15. "balance(address)": {
  16. "notice": "` + "`(balanceInmGAV / 1000).fixed(0,3)`" + ` GAV is the total funds available to ` + "`who.address()`." + `"
  17. }
  18. },
  19. "invariants": [
  20. { "notice": "The sum total amount of GAV in the system is 1 million." }
  21. ],
  22. "construction": [
  23. { "notice": "Endows ` + "`message.caller.address()`" + ` with 1m GAV." }
  24. ]
  25. }
  26. `)
  27. }
  28. var data = "0xc6888fa1000000000000000000000000000000000000000000000000000000000000007a"
  29. var tx = `
  30. {
  31. "jsonrpc": "2.0",
  32. "method": "eth_call",
  33. "params": [{
  34. "to": "0x8521742d3f456bd237e312d6e30724960f72517a",
  35. "data": "0xc6888fa1000000000000000000000000000000000000000000000000000000000000007a"
  36. }],
  37. "id": 6
  38. }
  39. `
  40. var abi = []byte(`
  41. [{
  42. "name": "multiply",
  43. "constant": false,
  44. "type": "function",
  45. "inputs": [{
  46. "name": "a",
  47. "type": "uint256"
  48. }],
  49. "outputs": [{
  50. "name": "d",
  51. "type": "uint256"
  52. }]
  53. }]
  54. `)
  55. func TestNotice(t *testing.T) {
  56. desc := "Will multiply `a` by 7 and return `a * 7`."
  57. expected := "Will multiply 122 by 7 and return 854."
  58. userdoc := makeUserdoc(desc)
  59. ns, err := NewWithDocs(abi, userdoc, tx)
  60. if err != nil {
  61. t.Errorf("New: error: %v", err)
  62. }
  63. notice, err := ns.Notice()
  64. if err != nil {
  65. t.Errorf("expected no error, got %v", err)
  66. }
  67. if notice != expected {
  68. t.Errorf("incorrect notice. expected %v, got %v", expected, notice)
  69. } else {
  70. t.Logf("returned notice \"%v\"", notice)
  71. }
  72. }
  73. // test missing method
  74. func TestMissingMethod(t *testing.T) {
  75. desc := "Will multiply `a` by 7 and return `a * 7`."
  76. userdoc := makeUserdoc(desc)
  77. expected := "natspec.js error evaluating expression: Natspec evaluation failed, method does not exist"
  78. ns, err := NewWithDocs(abi, userdoc, tx)
  79. if err != nil {
  80. t.Errorf("New: error: %v", err)
  81. }
  82. notice, err := ns.noticeForMethod(tx, "missing_method", "")
  83. if err == nil {
  84. t.Errorf("expected error, got nothing (notice: '%v')", notice)
  85. } else {
  86. if err.Error() != expected {
  87. t.Errorf("expected error '%s' got '%v' (notice: '%v')", expected, err, notice)
  88. }
  89. }
  90. }
  91. // test invalid desc
  92. /*
  93. func TestInvalidDesc(t *testing.T) {
  94. desc := "Will multiply 122 by \"7\" and return 854."
  95. expected := "natspec.js error setting expression: (anonymous): Line 1:41 Unexpected number"
  96. userdoc := makeUserdoc(desc)
  97. ns, err := NewWithDocs(abi, userdoc, tx)
  98. if err != nil {
  99. t.Errorf("New: error: %v", err)
  100. }
  101. notice, err := ns.Notice()
  102. if err == nil {
  103. t.Errorf("expected error, got nothing (notice: '%v')", err, notice)
  104. } else {
  105. if err.Error() != expected {
  106. t.Errorf("expected error '%s' got '%v' (notice: '%v')", expected, err, notice)
  107. }
  108. }
  109. }
  110. */
  111. // test wrong input params
  112. func TestWrongInputParams(t *testing.T) {
  113. desc := "Will multiply `e` by 7 and return `a * 7`."
  114. expected := "natspec.js error evaluating expression: Natspec evaluation failed, wrong input params"
  115. userdoc := makeUserdoc(desc)
  116. ns, err := NewWithDocs(abi, userdoc, tx)
  117. if err != nil {
  118. t.Errorf("New: error: %v", err)
  119. }
  120. notice, err := ns.Notice()
  121. if err == nil {
  122. t.Errorf("expected error, got nothing (notice: '%v')", notice)
  123. } else {
  124. if err.Error() != expected {
  125. t.Errorf("expected error '%s' got '%v' (notice: '%v')", expected, err, notice)
  126. }
  127. }
  128. }