missed a few bits in the refactor of bytes up/down

6 years ago · da1a0a03b4
parent ca59864c36
commit da1a0a03b4
2 changed files with 1 additions and 3 deletions
--- a/internal/session/repository_session.go
+++ b/internal/session/repository_session.go
@ -197,8 +197,6 @@ func (r *Repository) UpdateSession(ctx context.Context, session *Session, versio
 	}
 	for _, f := range fieldMaskPaths {
 		switch {
-		case strings.EqualFold("BytesUp", f):
-		case strings.EqualFold("BytesDown", f):
 		case strings.EqualFold("TerminationReason", f):
 		case strings.EqualFold("ServerId", f):
 		case strings.EqualFold("ServerType", f):
--- a/internal/session/repository_session_test.go
+++ b/internal/session/repository_session_test.go
@ -447,7 +447,7 @@ func TestRepository_UpdateSession(t *testing.T) {
 				terminationReason: Terminated,
 				serverId:          newServerFunc(),
 				serverType:        servers.ServerTypeWorker.String(),
-				fieldMaskPaths:    []string{"BytesUp", "BytesDown", "TerminationReason", "ServerId", "ServerType"},
+				fieldMaskPaths:    []string{"TerminationReason", "ServerId", "ServerType"},
 			},
 			wantErr:        false,
 			wantRowsUpdate: 1,