github.com/qiniu/go-sdk/v7@v7.27.0/pili/stream_test.go (about)

     1  //go:build integration
     2  // +build integration
     3  
     4  package pili
     5  
     6  import (
     7  	"context"
     8  	"testing"
     9  
    10  	"github.com/stretchr/testify/assert"
    11  )
    12  
    13  func TestManager_GetStreamsList(t *testing.T) {
    14  	if SkipTest() {
    15  		t.SkipNow()
    16  	}
    17  	ast := assert.New(t)
    18  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    19  	_, err := manager.GetStreamsList(context.Background(), GetStreamListRequest{Hub: TestHub, LiveOnly: false, Limit: 10})
    20  	ast.Nil(err)
    21  
    22  	_, err = manager.GetStreamsList(context.Background(), GetStreamListRequest{Hub: TestHub, LiveOnly: false, Limit: 99999})
    23  	ast.NotNil(err)
    24  }
    25  
    26  func TestManager_GetStreamBaseInfo(t *testing.T) {
    27  	if SkipTest() || TestStream == "" {
    28  		t.SkipNow()
    29  	}
    30  	ast := assert.New(t)
    31  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    32  	_, err := manager.GetStreamBaseInfo(context.Background(), GetStreamBaseInfoRequest{Hub: TestHub, Stream: TestStream})
    33  	ast.Nil(err)
    34  }
    35  
    36  func TestManager_StreamDisable(t *testing.T) {
    37  	if SkipTest() || TestStream == "" {
    38  		t.SkipNow()
    39  	}
    40  	ast := assert.New(t)
    41  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    42  	err := manager.StreamDisable(context.Background(), StreamDisabledRequest{
    43  		Hub:                 TestHub,
    44  		Stream:              TestStream,
    45  		DisablePeriodSecond: 60,
    46  	})
    47  	ast.Nil(err)
    48  
    49  	err = manager.StreamDisable(context.Background(), StreamDisabledRequest{
    50  		Hub:          TestHub,
    51  		Stream:       TestStream,
    52  		DisabledTill: 0,
    53  	})
    54  	ast.Nil(err)
    55  }
    56  
    57  func TestManager_GetStreamLiveStatus(t *testing.T) {
    58  	if SkipTest() || TestStream == "" {
    59  		t.SkipNow()
    60  	}
    61  	ast := assert.New(t)
    62  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    63  	_, err := manager.GetStreamLiveStatus(context.Background(), GetStreamLiveStatusRequest{Hub: TestHub, Stream: TestStream})
    64  	if err != nil {
    65  		ast.EqualError(err, "no live")
    66  	}
    67  }
    68  
    69  func TestManager_BatchGetStreamLiveStatus(t *testing.T) {
    70  	if SkipTest() {
    71  		t.SkipNow()
    72  	}
    73  	ast := assert.New(t)
    74  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    75  	_, err := manager.BatchGetStreamLiveStatus(context.Background(), BatchGetStreamLiveStatusRequest{
    76  		Hub:   TestHub,
    77  		Items: []string{TestStream},
    78  	})
    79  	ast.Nil(err)
    80  }
    81  
    82  func TestManager_GetStreamHistory(t *testing.T) {
    83  	if SkipTest() || TestStream == "" {
    84  		t.SkipNow()
    85  	}
    86  	ast := assert.New(t)
    87  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
    88  	_, err := manager.GetStreamHistory(context.Background(), GetStreamHistoryRequest{
    89  		Hub:    TestHub,
    90  		Stream: TestStream,
    91  		Start:  0,
    92  		End:    0,
    93  	})
    94  	ast.Nil(err)
    95  }
    96  
    97  func TestManager_StreamSaveas(t *testing.T) {
    98  	if SkipTest() || TestStream == "" {
    99  		t.SkipNow()
   100  	}
   101  	ast := assert.New(t)
   102  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
   103  	_, err := manager.StreamSaveas(context.Background(), StreamSaveasRequest{
   104  		Hub:        TestHub,
   105  		Stream:     TestStream,
   106  		Start:      1632796442,
   107  		End:        1632810842,
   108  		Fname:      "test",
   109  		ExpireDays: -1,
   110  	})
   111  	if err != nil {
   112  		ast.EqualError(err, "no data")
   113  	}
   114  }
   115  
   116  func TestManager_StreamSnapshot(t *testing.T) {
   117  	if SkipTest() || TestStream == "" {
   118  		t.SkipNow()
   119  	}
   120  	ast := assert.New(t)
   121  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
   122  	ret, err := manager.StreamSnapshot(context.Background(), StreamSnapshotRequest{
   123  		Hub:             TestHub,
   124  		Stream:          TestStream,
   125  		Time:            1632810081,
   126  		Fname:           "test",
   127  		DeleteAfterDays: 1,
   128  	})
   129  	if err == nil {
   130  		ast.Equal("test", ret.Fname)
   131  	}
   132  }
   133  
   134  func TestManager_StreamConverts(t *testing.T) {
   135  	if SkipTest() || TestStream == "" {
   136  		t.SkipNow()
   137  	}
   138  	ast := assert.New(t)
   139  	manager := NewManager(ManagerConfig{AccessKey: AccessKey, SecretKey: SecretKey})
   140  	err := manager.StreamConverts(context.Background(), StreamConvertsRequest{
   141  		Hub:      TestHub,
   142  		Stream:   TestStream,
   143  		Converts: []string{"480p", "720p"},
   144  	})
   145  	ast.Nil(err)
   146  }
   147  
   148  func TestEncodeStream(t *testing.T) {
   149  	ast := assert.New(t)
   150  	str := encodeStream("test0")
   151  	ast.Equal(str, "dGVzdDA=")
   152  }