github.com/polygon-io/client-go@v1.16.4/rest/models/aggs_test.go (about)

     1  package models_test
     2  
     3  import (
     4  	"reflect"
     5  	"testing"
     6  
     7  	"github.com/polygon-io/client-go/rest/models"
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestListAggsParams(t *testing.T) {
    12  	adjusted := true
    13  	order := models.Asc
    14  	limit := 50
    15  
    16  	expect := models.ListAggsParams{
    17  		Adjusted: &adjusted,
    18  		Order:    &order,
    19  		Limit:    &limit,
    20  	}
    21  	actual := models.ListAggsParams{}.WithAdjusted(adjusted).WithOrder(order).WithLimit(limit)
    22  	checkParams(t, expect, *actual)
    23  }
    24  
    25  func TestGetAggsParamsMethods(t *testing.T) {
    26  	adjusted := true
    27  	order := models.Asc
    28  	limit := 50
    29  
    30  	expect := models.GetAggsParams{
    31  		Adjusted: &adjusted,
    32  		Order:    &order,
    33  		Limit:    &limit,
    34  	}
    35  	actual := models.GetAggsParams{}.WithAdjusted(adjusted).WithOrder(order).WithLimit(limit)
    36  	checkParams(t, expect, *actual)
    37  }
    38  
    39  func TestGetGroupedDailyAggsParamsMethods(t *testing.T) {
    40  	adjusted := true
    41  	includeOTC := true
    42  
    43  	expect := models.GetGroupedDailyAggsParams{
    44  		Adjusted:   &adjusted,
    45  		IncludeOTC: &includeOTC,
    46  	}
    47  	actual := models.GetGroupedDailyAggsParams{}.WithAdjusted(adjusted).WithIncludeOTC(includeOTC)
    48  	checkParams(t, expect, *actual)
    49  }
    50  
    51  func TestGetDailyOpenCloseAggParamsMethods(t *testing.T) {
    52  	adjusted := true
    53  
    54  	expect := models.GetDailyOpenCloseAggParams{
    55  		Adjusted: &adjusted,
    56  	}
    57  	actual := models.GetDailyOpenCloseAggParams{}.WithAdjusted(adjusted)
    58  	checkParams(t, expect, *actual)
    59  }
    60  
    61  func TestGetPreviousCloseAggParamsMethods(t *testing.T) {
    62  	adjusted := true
    63  
    64  	expect := models.GetPreviousCloseAggParams{
    65  		Adjusted: &adjusted,
    66  	}
    67  	actual := models.GetPreviousCloseAggParams{}.WithAdjusted(adjusted)
    68  	checkParams(t, expect, *actual)
    69  }
    70  
    71  func checkParams(t *testing.T, expect, actual interface{}) {
    72  	for _, field := range reflect.VisibleFields(reflect.TypeOf(actual)) {
    73  		assert.NotNil(t, field)
    74  	}
    75  	assert.Equal(t, expect, actual)
    76  }