Check next block start by case, not colon

wsl.go

@@ -815,9 +815,9 @@ func (p *Processor) findLeadingAndTrailingWhitespaces(stmt, nextStatement ast.No
 
 		switch n := nextStatement.(type) {
 		case *ast.CaseClause:
-			blockEndPos = n.Colon
+			blockEndPos = n.Case
 		case *ast.CommClause:
-			blockEndPos = n.Colon
+			blockEndPos = n.Case
 		default:
 			// We're not at the end of the case?
 			return

wsl_test.go

@@ -1014,6 +1014,47 @@ func TestShouldAddEmptyLines(t *testing.T) {
 				"if statements should only be cuddled with assignments used in the if statement itself",
 			},
 		},
+		{
+			description: "multiline case statements",
+			code: []byte(`package main
+
+			func main() {
+				switch {
+				case true,
+					false:
+					fmt.Println("ok")
+				case true ||
+					false:
+					fmt.Println("ok")
+				case true, false:
+					fmt.Println("ok")
+				case true || false:
+					fmt.Println("ok")
+				case true,
+					false:
+
+					fmt.Println("starting whitespace multiline case")
+				case true ||
+					false:
+
+					fmt.Println("starting whitespace multiline case")
+				case true,
+					false:
+					fmt.Println("ending whitespace multiline case")
+
+				case true,
+					false:
+					fmt.Println("last case should also fail")
+
+				}
+			}`),
+			expectedErrorStrings: []string{
+				"block should not end with a whitespace (or comment)",
+				"block should not start with a whitespace",
+				"block should not start with a whitespace",
+				"block should not end with a whitespace (or comment)",
+			},
+		},
 	}
 
 	for _, tc := range cases {