Function TestBatchCreateLogs

coderd/agentapi/logs_test.go:26–482

(t *testing.T)

)

func TestBatchCreateLogs(t *testing.T) {
	t.Parallel()

	var (
		agent = database.WorkspaceAgent{
			ID: uuid.New(),
		}
		logSource = database.WorkspaceAgentLogSource{
			WorkspaceAgentID: agent.ID,
			CreatedAt:        dbtime.Now(),
			ID:               uuid.New(),
		}
	)

	t.Run("OK", func(t *testing.T) {
		t.Parallel()

		dbM := dbmock.NewMockStore(gomock.NewController(t))

		publishWorkspaceUpdateCalled := false
		publishWorkspaceAgentLogsUpdateCalled := false
		now := dbtime.Now()
		api := &agentapi.LogsAPI{
			AgentFn: func(context.Context) (database.WorkspaceAgent, error) {
				return agent, nil
			},
			Database: dbM,
			Log:      testutil.Logger(t),
			PublishWorkspaceUpdateFn: func(ctx context.Context, _ uuid.UUID, kind wspubsub.WorkspaceEventKind) error {
				publishWorkspaceUpdateCalled = true
				return nil
			},
			PublishWorkspaceAgentLogsUpdateFn: func(ctx context.Context, workspaceAgentID uuid.UUID, msg agentsdk.LogsNotifyMessage) {
				publishWorkspaceAgentLogsUpdateCalled = true

				// Check the message content, should be for -1 since the lowest
				// log we inserted was 0.
				assert.Equal(t, agentsdk.LogsNotifyMessage{CreatedAfter: -1}, msg)
			},
			TimeNowFn: func() time.Time { return now },
		}

		req := &agentproto.BatchCreateLogsRequest{
			LogSourceId: logSource.ID[:],
			Logs: []*agentproto.Log{
				{
					CreatedAt: timestamppb.New(now),
					Level:     agentproto.Log_TRACE,
					Output:    "log line 1",
				},
				{
					CreatedAt: timestamppb.New(now.Add(time.Hour)),
					Level:     agentproto.Log_DEBUG,
					Output:    "log line 2",
				},
				{
					CreatedAt: timestamppb.New(now.Add(2 * time.Hour)),
					Level:     agentproto.Log_INFO,