123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501 |
- package sarama
- import (
- "errors"
- "testing"
- )
- func TestClusterAdmin(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminInvalidController(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- _, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err == nil {
- t.Fatal(errors.New("controller not set still cluster admin was created"))
- }
- if err != ErrControllerNotAvailable {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreateTopic(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreateTopicsRequest": NewMockCreateTopicsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.CreateTopic("my_topic", &TopicDetail{NumPartitions: 1, ReplicationFactor: 1}, false)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreateTopicWithInvalidTopicDetail(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreateTopicsRequest": NewMockCreateTopicsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.CreateTopic("my_topic", nil, false)
- if err.Error() != "You must specify topic details" {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreateTopicWithDiffVersion(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreateTopicsRequest": NewMockCreateTopicsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_11_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.CreateTopic("my_topic", &TopicDetail{NumPartitions: 1, ReplicationFactor: 1}, false)
- if err != ErrInsufficientData {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDeleteTopic(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DeleteTopicsRequest": NewMockDeleteTopicsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.DeleteTopic("my_topic")
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDeleteEmptyTopic(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DeleteTopicsRequest": NewMockDeleteTopicsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.DeleteTopic("")
- if err != ErrInvalidTopic {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreatePartitions(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreatePartitionsRequest": NewMockCreatePartitionsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.CreatePartitions("my_topic", 3, nil, false)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreatePartitionsWithDiffVersion(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreatePartitionsRequest": NewMockCreatePartitionsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.CreatePartitions("my_topic", 3, nil, false)
- if err != ErrUnsupportedVersion {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDeleteRecords(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DeleteRecordsRequest": NewMockDeleteRecordsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- partitionOffset := make(map[int32]int64)
- partitionOffset[1] = 1000
- partitionOffset[2] = 1000
- partitionOffset[3] = 1000
- err = admin.DeleteRecords("my_topic", partitionOffset)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDeleteRecordsWithDiffVersion(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DeleteRecordsRequest": NewMockDeleteRecordsResponse(t),
- })
- config := NewConfig()
- config.Version = V0_10_2_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- partitionOffset := make(map[int32]int64)
- partitionOffset[1] = 1000
- partitionOffset[2] = 1000
- partitionOffset[3] = 1000
- err = admin.DeleteRecords("my_topic", partitionOffset)
- if err != ErrUnsupportedVersion {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDescribeConfig(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DescribeConfigsRequest": NewMockDescribeConfigsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- resource := ConfigResource{Name: "r1", Type: TopicResource, ConfigNames: []string{"my_topic"}}
- entries, err := admin.DescribeConfig(resource)
- if err != nil {
- t.Fatal(err)
- }
- if len(entries) <= 0 {
- t.Fatal(errors.New("no resource present"))
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminAlterConfig(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "AlterConfigsRequest": NewMockAlterConfigsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- var value string
- entries := make(map[string]*string)
- value = "3"
- entries["ReplicationFactor"] = &value
- err = admin.AlterConfig(TopicResource, "my_topic", entries, false)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminCreateAcl(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "CreateAclsRequest": NewMockCreateAclsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- r := Resource{ResourceType: AclResourceTopic, ResourceName: "my_topic"}
- a := Acl{Host: "localhost", Operation: AclOperationAlter, PermissionType: AclPermissionAny}
- err = admin.CreateACL(r, a)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminListAcls(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DescribeAclsRequest": NewMockListAclsResponse(t),
- "CreateAclsRequest": NewMockCreateAclsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- r := Resource{ResourceType: AclResourceTopic, ResourceName: "my_topic"}
- a := Acl{Host: "localhost", Operation: AclOperationAlter, PermissionType: AclPermissionAny}
- err = admin.CreateACL(r, a)
- if err != nil {
- t.Fatal(err)
- }
- resourceName := "my_topic"
- filter := AclFilter{
- ResourceType: AclResourceTopic,
- Operation: AclOperationRead,
- ResourceName: &resourceName,
- }
- rAcls, err := admin.ListAcls(filter)
- if err != nil {
- t.Fatal(err)
- }
- if len(rAcls) <= 0 {
- t.Fatal("no acls present")
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestClusterAdminDeleteAcl(t *testing.T) {
- seedBroker := NewMockBroker(t, 1)
- defer seedBroker.Close()
- seedBroker.SetHandlerByMap(map[string]MockResponse{
- "MetadataRequest": NewMockMetadataResponse(t).
- SetController(seedBroker.BrokerID()).
- SetBroker(seedBroker.Addr(), seedBroker.BrokerID()),
- "DeleteAclsRequest": NewMockDeleteAclsResponse(t),
- })
- config := NewConfig()
- config.Version = V1_0_0_0
- admin, err := NewClusterAdmin([]string{seedBroker.Addr()}, config)
- if err != nil {
- t.Fatal(err)
- }
- resourceName := "my_topic"
- filter := AclFilter{
- ResourceType: AclResourceTopic,
- Operation: AclOperationAlter,
- ResourceName: &resourceName,
- }
- _, err = admin.DeleteACL(filter, false)
- if err != nil {
- t.Fatal(err)
- }
- err = admin.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
|