Fixed TestClassAt at mock environment

AryanGodara · May 24, 2024 · e9fe044 · e9fe044
1 parent dd98a9b
commit e9fe044
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/rpc/contract_test.go b/rpc/contract_test.go
@@ -89,7 +89,6 @@ func TestClassAt(t *testing.T) {
 
 			require.Condition(func() bool {
 				for _, deprecatedCairoEntryPoint := range class.DeprecatedEntryPointsByType.External {
-					t.Log(deprecatedCairoEntryPoint)
 					if test.ExpectedOperation == deprecatedCairoEntryPoint.Selector.String() {
 						return true
 					}
@@ -104,7 +103,6 @@ func TestClassAt(t *testing.T) {
 
 			require.Condition(func() bool {
 				for _, entryPointsByType := range class.EntryPointsByType.External {
-					t.Log(entryPointsByType)
 					if test.ExpectedOperation == entryPointsByType.Selector.String() {
 						return true
 					}

diff --git a/rpc/mock_test.go b/rpc/mock_test.go
@@ -410,8 +410,22 @@ func mock_starknet_getClassAt(result interface{}, method string, args ...interfa
 	if len(args) != 2 {
 		return errWrongArgs
 	}
+	fakeSelector, err := utils.HexToFelt("0xdeadbeef")
+	if err != nil {
+		return err
+	}
 	var class = DeprecatedContractClass{
 		Program: "H4sIAAAAAAAE/+Vde3PbOJL/Kj5VXW1mVqsC36Sr9g8n0c6mzonnbM",
+		DeprecatedEntryPointsByType: DeprecatedEntryPointsByType{
+			Constructor: []DeprecatedCairoEntryPoint{},
+			External: []DeprecatedCairoEntryPoint{
+				{
+					Offset:   "0x0xdeadbeef",
+					Selector: fakeSelector,
+				},
+			},
+			L1Handler: []DeprecatedCairoEntryPoint{},
+		},
 	}
 	outputContent, err := json.Marshal(class)
 	if err != nil {