github.com/mysteriumnetwork/node@v0.0.0-20240516044423-365054f76801/core/discovery/reducer/proposal_test.go (about)

     1  /*
     2   * Copyright (C) 2019 The "MysteriumNetwork/node" Authors.
     3   *
     4   * This program is free software: you can redistribute it and/or modify
     5   * it under the terms of the GNU General Public License as published by
     6   * the Free Software Foundation, either version 3 of the License, or
     7   * (at your option) any later version.
     8   *
     9   * This program is distributed in the hope that it will be useful,
    10   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12   * GNU General Public License for more details.
    13   *
    14   * You should have received a copy of the GNU General Public License
    15   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    16   */
    17  
    18  package reducer
    19  
    20  import (
    21  	"testing"
    22  
    23  	"github.com/stretchr/testify/assert"
    24  )
    25  
    26  func Test_ProviderID(t *testing.T) {
    27  	match := EqualString(ProviderID, provider1)
    28  
    29  	assert.False(t, match(proposalEmpty))
    30  	assert.True(t, match(proposalProvider1Streaming))
    31  	assert.True(t, match(proposalProvider1Noop))
    32  	assert.False(t, match(proposalProvider2Streaming))
    33  }
    34  
    35  func Test_ServiceType(t *testing.T) {
    36  	match := EqualString(ServiceType, serviceTypeStreaming)
    37  
    38  	assert.False(t, match(proposalEmpty))
    39  	assert.True(t, match(proposalProvider1Streaming))
    40  	assert.False(t, match(proposalProvider1Noop))
    41  	assert.True(t, match(proposalProvider2Streaming))
    42  }
    43  
    44  func Test_Location_FiltersByCountry(t *testing.T) {
    45  	match := EqualString(LocationCountry, "DE")
    46  
    47  	assert.False(t, match(proposalEmpty))
    48  	assert.True(t, match(proposalProvider1Streaming))
    49  	assert.False(t, match(proposalProvider1Noop))
    50  	assert.False(t, match(proposalProvider2Streaming))
    51  }
    52  
    53  func Test_Location_FiltersByIPType(t *testing.T) {
    54  	match := EqualString(LocationType, "residential")
    55  
    56  	assert.False(t, match(proposalEmpty))
    57  	assert.False(t, match(proposalProvider1Streaming))
    58  	assert.False(t, match(proposalProvider1Noop))
    59  	assert.True(t, match(proposalProvider2Streaming))
    60  }
    61  
    62  func Test_AccessPolicy_FiltersByID(t *testing.T) {
    63  	match := AccessPolicy(accessRuleWhitelist.ID, "")
    64  
    65  	assert.False(t, match(proposalEmpty))
    66  	assert.True(t, match(proposalProvider1Streaming))
    67  	assert.False(t, match(proposalProvider1Noop))
    68  	assert.True(t, match(proposalProvider2Streaming))
    69  }
    70  
    71  func Test_AccessPolicy_FiltersBySource(t *testing.T) {
    72  	match := AccessPolicy("", accessRuleBlacklist.Source)
    73  
    74  	assert.False(t, match(proposalEmpty))
    75  	assert.False(t, match(proposalProvider1Streaming))
    76  	assert.False(t, match(proposalProvider1Noop))
    77  	assert.True(t, match(proposalProvider2Streaming))
    78  }
    79  
    80  func Test_AccessPolicy_FiltersByIDAndSource(t *testing.T) {
    81  	match := AccessPolicy(accessRuleWhitelist.ID, accessRuleWhitelist.Source)
    82  
    83  	assert.False(t, match(proposalEmpty))
    84  	assert.True(t, match(proposalProvider1Streaming))
    85  	assert.False(t, match(proposalProvider1Noop))
    86  	assert.True(t, match(proposalProvider2Streaming))
    87  }