Merge pull request #1389 from onflow/chasefleming/format-access-mods

Format boilerplat access modifiers
onflow · Feb 7, 2024 · 49d80e4 · 49d80e4
2 parents b05ee6f + d6116c1
commit 49d80e4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/internal/super/generate.go b/internal/super/generate.go
@@ -157,11 +157,13 @@ func generateNew(
 	switch templateType {
 	case "contract":
 		fileToWrite = fmt.Sprintf(`
-access(all) contract %s {
+access(all)
+contract %s {
     init() {}
 }`, name)
 	case "script":
-		fileToWrite = `access(all) fun main() {
+		fileToWrite = `access(all)
+fun main() {
     // Script details here
 }`
 	case "transaction":

diff --git a/internal/super/generate_test.go b/internal/super/generate_test.go
@@ -51,7 +51,8 @@ func TestGenerateNewContract(t *testing.T) {
 
 	// Check content is correct
 	expectedContent := `
-access(all) contract TestContract {
+access(all)
+contract TestContract {
     init() {}
 }`
 	assert.Equal(t, expectedContent, string(fileContent))
@@ -129,7 +130,8 @@ func TestGenerateNewScript(t *testing.T) {
 	assert.NoError(t, err, "Failed to read generated file")
 	assert.NotNil(t, content)
 
-	expectedContent := `access(all) fun main() {
+	expectedContent := `access(all)
+fun main() {
     // Script details here
 }`
 	assert.Equal(t, expectedContent, string(content))
@@ -185,7 +187,8 @@ func TestGenerateNewWithDirFlag(t *testing.T) {
 	assert.NotNil(t, content)
 
 	expectedContent := `
-access(all) contract TestContract {
+access(all)
+contract TestContract {
     init() {}
 }`
 	assert.Equal(t, expectedContent, string(content))